matthias/liederwahl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update for new table structure

Browse Source
This commit is contained in:
matthias@matsewe.de
2024-05-24 10:24:48 +02:00
parent f6016f5736
commit 0c63295939
3 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/crud.py
View File

@@ -30,6 +30,7 @@ def create_song(db,
aca_artist, aca_artist,
title, title,
url, url,
source,
yt_id, yt_id,
spfy_id, spfy_id,
thumbnail, thumbnail,
@@ -43,6 +44,7 @@ def create_song(db,
aca_artist=aca_artist, aca_artist=aca_artist,
title=title, title=title,
url=url, url=url,
source=source,
yt_id=yt_id, yt_id=yt_id,
spfy_id=spfy_id, spfy_id=spfy_id,
thumbnail=thumbnail, thumbnail=thumbnail,

1
app/models.py
View File

@@ -14,6 +14,7 @@ class Song(Base):
aca_artist: Mapped[Optional[str]] aca_artist: Mapped[Optional[str]]
title: Mapped[Optional[str]] title: Mapped[Optional[str]]
url: Mapped[Optional[str]] url: Mapped[Optional[str]]
source: Mapped[Optional[str]]
yt_id: Mapped[Optional[str]] yt_id: Mapped[Optional[str]]
spfy_id: Mapped[Optional[str]] spfy_id: Mapped[Optional[str]]
thumbnail: Mapped[Optional[str]] thumbnail: Mapped[Optional[str]]

17
app/routers/admin.py
View File

@@ -75,16 +75,16 @@ async def create_upload_file(db: Session = Depends(get_db)):
song_list = song_list.replace({np.nan: None}) song_list = song_list.replace({np.nan: None})
song_list = song_list.replace({"n/a": None}) song_list = song_list.replace({"n/a": None})
category_names = list(song_list.iloc[0][6:19]) category_names = list(song_list.iloc[0][7:16])
for row in song_list[1:].iterrows(): for i, row in song_list[1:].iterrows():
row = np.array(row[1]) row = np.array(row)
yt_id = get_youtube_id(row[3]) yt_id = get_youtube_id(row[3])
spfy_id = get_spotify_id(row[3]) spfy_id = get_spotify_id(row[3])
categories = {n: v for n, v in zip( categories = {n: v for n, v in zip(
category_names, row[6:19] != None)} category_names, row[7:16] != None)}
if not np.any(list(categories.values())): if not np.any(list(categories.values())):
continue continue
@@ -94,12 +94,13 @@ async def create_upload_file(db: Session = Depends(get_db)):
aca_artist=row[1], aca_artist=row[1],
title=row[2], title=row[2],
url=row[3], url=row[3],
source=row[4],
yt_id=yt_id, yt_id=yt_id,
spfy_id=spfy_id, spfy_id=spfy_id,
thumbnail=get_thumbnail(row[3]), thumbnail=get_thumbnail(row[3]),
is_aca=row[4] == "ja", is_aca=row[5] == "ja",
arng_url=row[5], arng_url=row[6],
categories=categories, categories=categories,
main_category=category_names[get_main_category(row[6:19])], main_category=category_names[get_main_category(row[7:16])],
singable=row[19] != "nein" singable=row[16] != "nein"
) )