Skip to content

Commit 11369e4

Browse files
committed
Feat: adds KSaunders media items data loader
1 parent 4fb3601 commit 11369e4

File tree

1 file changed

+57
-0
lines changed

1 file changed

+57
-0
lines changed
Lines changed: 57 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,57 @@
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 KhaylaSaundersLoader implements Loader {
7+
getLoaderName(): string {
8+
return 'khaylasaunders';
9+
}
10+
//TA Help/Chatgpt to complete Extra Credit
11+
async loadData(): Promise<MediaItem[]> {
12+
const mediaItems = await this.loadMediaItems();
13+
const credits = await this.loadCredits();
14+
15+
for (const credit of credits) {
16+
const mediaItem = mediaItems.find(
17+
(item) => item.getId() === credit.getMediaItemId(),
18+
);
19+
20+
if (mediaItem) {
21+
mediaItem.addCredit(credit);
22+
}
23+
}
24+
25+
console.log(
26+
`Loaded ${credits.length} credits and ${mediaItems.length} media items`,
27+
);
28+
29+
return mediaItems;
30+
}
31+
32+
async loadMediaItems(): Promise<MediaItem[]> {
33+
// TODO: Implement this method.
34+
const mediaItems = [];
35+
const readable = fs
36+
.createReadStream('data/media_items.csv', 'utf-8')
37+
.pipe(csv());
38+
for await (const row of readable) {
39+
const { id, title, type, year } = row;
40+
mediaItems.push(new MediaItem(id, title, type, year, []));
41+
}
42+
return mediaItems;
43+
}
44+
45+
async loadCredits(): Promise<Credit[]> {
46+
const credits = [];
47+
const readable = fs
48+
.createReadStream('data/credits.csv', 'utf-8')
49+
.pipe(csv());
50+
for await (const row of readable) {
51+
const { media_item_id, role, name } = row;
52+
credits.push(new Credit(media_item_id, name, role));
53+
}
54+
return credits;
55+
}
56+
}
57+

0 commit comments

Comments
 (0)