Skip to content

Commit d571ce7

Browse files
feat: adds Trinitie's library data loader (#401)
* feat: adds lesson_10 homework and extra credit * chore: fixing indentations --------- Co-authored-by: Anthony D. Mays <[email protected]>
1 parent b37947e commit d571ce7

File tree

2 files changed

+62
-0
lines changed

2 files changed

+62
-0
lines changed

lesson_10/libraries/src/loaders/loaders.module.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ import { JoneemckellarLoader } from './joneemckellar_loader.js';
1111
import { LindaQuinoaLoader } from './linda_quinoa_loader.js';
1212
import { MattieWeathersbyLoader } from './mattie_weathersby_loader.js';
1313
import { NicoleJacksonLoader } from './nicole_jackson_loader.js';
14+
import { TrinitieJacksonLoader } from './trinitie_jackson_loader.js';
1415
import { TyranRiceLoader } from './tyranricejr_loader.js';
1516

1617
export const Loaders = Symbol.for('Loaders');
@@ -28,6 +29,7 @@ const LOADER_PROVIDERS = [
2829
LindaQuinoaLoader,
2930
MattieWeathersbyLoader,
3031
NicoleJacksonLoader,
32+
TrinitieJacksonLoader,
3133
TyranRiceLoader,
3234
];
3335

Lines changed: 60 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,60 @@
1+
import csv from 'csv-parser';
2+
import fs from 'fs';
3+
import { Credit, MediaItem } from '../models/index.js';
4+
import { Loader } from './loader.js';
5+
6+
export class TrinitieJacksonLoader implements Loader {
7+
getLoaderName(): string {
8+
return 'trinitiejackson';
9+
}
10+
11+
async loadData(): Promise<MediaItem[]> {
12+
const credits = await this.loadCredits();
13+
const mediaItems = await this.loadMediaItems();
14+
const creditsByMediaId = new Map<string, Credit[]>();
15+
16+
for (const credit of credits) {
17+
const mediaItemId = credit.getMediaItemId();
18+
const mediaCredits = creditsByMediaId.get(mediaItemId) || [];
19+
mediaCredits.push(credit);
20+
creditsByMediaId.set(mediaItemId, mediaCredits);
21+
}
22+
23+
for (const mediaItem of mediaItems) {
24+
const itemCredits = creditsByMediaId.get(mediaItem.getId()) || [];
25+
for (const credit of itemCredits) {
26+
mediaItem.addCredit(credit);
27+
}
28+
}
29+
30+
console.log(
31+
`Loaded ${credits.length} credits and ${mediaItems.length} media items`,
32+
);
33+
34+
return [...mediaItems.values()];
35+
}
36+
37+
async loadMediaItems(): Promise<MediaItem[]> {
38+
const mediaItems = [];
39+
const readable = fs
40+
.createReadStream('data/media_items.csv', 'utf-8')
41+
.pipe(csv());
42+
for await (const row of readable) {
43+
const { id, type, title, year } = row;
44+
mediaItems.push(new MediaItem(id, title, type, parseInt(year, 10), []));
45+
}
46+
return mediaItems;
47+
}
48+
49+
async loadCredits(): Promise<Credit[]> {
50+
const credits = [];
51+
const readable = fs
52+
.createReadStream('data/credits.csv', 'utf-8')
53+
.pipe(csv());
54+
for await (const row of readable) {
55+
const { media_item_id, role, name } = row;
56+
credits.push(new Credit(media_item_id, name, role));
57+
}
58+
return credits;
59+
}
60+
}

0 commit comments

Comments
 (0)