Skip to content

Commit b37947e

Browse files
feat: adds Tyrans's loadMediaItems() functionality and combines credits \w media items (#407)
Co-authored-by: Anthony D. Mays <[email protected]>
1 parent 46be0d6 commit b37947e

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 { TyranRiceLoader } from './tyranricejr_loader.js';
1415

1516
export const Loaders = Symbol.for('Loaders');
1617

@@ -27,6 +28,7 @@ const LOADER_PROVIDERS = [
2728
LindaQuinoaLoader,
2829
MattieWeathersbyLoader,
2930
NicoleJacksonLoader,
31+
TyranRiceLoader,
3032
];
3133

3234
@Module({
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 TyranRiceLoader implements Loader {
7+
getLoaderName(): string {
8+
return 'tyranricejr';
9+
}
10+
11+
async loadData(): Promise<MediaItem[]> {
12+
const credits = await this.loadCredits();
13+
const mediaItems = await this.loadMediaItems();
14+
15+
// Build a quick lookup by media item id
16+
const mediaMap = new Map<string, MediaItem>();
17+
for (const item of mediaItems) {
18+
mediaMap.set(item.getId(), item);
19+
}
20+
21+
// Attach credits to the right media item
22+
for (const credit of credits) {
23+
const item = mediaMap.get(credit.getMediaItemId());
24+
if (item) {
25+
item.addCredit(credit);
26+
}
27+
}
28+
29+
console.log(
30+
`Loaded ${credits.length} credits and ${mediaItems.length} media items`,
31+
);
32+
33+
return [...mediaItems.values()];
34+
}
35+
36+
async loadMediaItems(): Promise<MediaItem[]> {
37+
const mediaItems = [];
38+
const readable = fs
39+
.createReadStream('data/media_items.csv', 'utf-8')
40+
.pipe(csv());
41+
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)