Skip to content

Commit 598c030

Browse files
committed
feat: added evan_philakhong_loader.ts
1 parent b01aa8a commit 598c030

File tree

1 file changed

+49
-0
lines changed

1 file changed

+49
-0
lines changed
Lines changed: 49 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,49 @@
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 EvanPhilakhongLoader implements Loader {
7+
getLoaderName(): string {
8+
return 'anthonymays';
9+
}
10+
11+
async loadData(): Promise<MediaItem[]> {
12+
const credits = await this.loadCredits();
13+
const mediaItems = await this.loadMediaItems();
14+
15+
console.log(
16+
`Loaded ${credits.length} credits and ${mediaItems.length} media items`,
17+
);
18+
19+
return [...mediaItems.values()];
20+
}
21+
22+
async loadMediaItems(): Promise<MediaItem[]> {
23+
const mediaItems = [];
24+
const credits = this.loadCredits;
25+
const readable = fs
26+
.createReadStream('data/media_items.csv', 'utf-8')
27+
.pipe(csv());
28+
for await (const row of readable) {
29+
const { media_item_id, title, genre, year, credits } = row;
30+
mediaItems.push(
31+
new MediaItem(media_item_id, title, genre, year, credits),
32+
);
33+
}
34+
35+
return mediaItems;
36+
}
37+
38+
async loadCredits(): Promise<Credit[]> {
39+
const credits = [];
40+
const readable = fs
41+
.createReadStream('data/credits.csv', 'utf-8')
42+
.pipe(csv());
43+
for await (const row of readable) {
44+
const { media_item_id, role, name } = row;
45+
credits.push(new Credit(media_item_id, name, role));
46+
}
47+
return credits;
48+
}
49+
}

0 commit comments

Comments
 (0)