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,
title,
url,
source,
yt_id,
spfy_id,
thumbnail,
@@ -43,6 +44,7 @@ def create_song(db,
aca_artist=aca_artist,
title=title,
url=url,
source=source,
yt_id=yt_id,
spfy_id=spfy_id,
thumbnail=thumbnail,

1
app/models.py
View File

@@ -14,6 +14,7 @@ class Song(Base):
aca_artist: Mapped[Optional[str]]
title: Mapped[Optional[str]]
url: Mapped[Optional[str]]
source: Mapped[Optional[str]]
yt_id: Mapped[Optional[str]]
spfy_id: 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({"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():
row = np.array(row[1])
for i, row in song_list[1:].iterrows():
row = np.array(row)
yt_id = get_youtube_id(row[3])
spfy_id = get_spotify_id(row[3])
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())):
continue
@@ -94,12 +94,13 @@ async def create_upload_file(db: Session = Depends(get_db)):
aca_artist=row[1],
title=row[2],
url=row[3],
source=row[4],
yt_id=yt_id,
spfy_id=spfy_id,
thumbnail=get_thumbnail(row[3]),
is_aca=row[4] == "ja",
arng_url=row[5],
is_aca=row[5] == "ja",
arng_url=row[6],
categories=categories,
main_category=category_names[get_main_category(row[6:19])],
singable=row[19] != "nein"
main_category=category_names[get_main_category(row[7:16])],
singable=row[16] != "nein"
)