matsewe
d4cf9a53dd
All checks were successful
release-tag / release-image (push) Successful in 5m49s
145 lines
4.2 KiB
Python
145 lines
4.2 KiB
Python
import pandas as pd
|
|
import numpy as np
|
|
import re
|
|
import requests
|
|
import os
|
|
from fastapi import APIRouter, Security, Depends, HTTPException, Response
|
|
from fastapi.responses import JSONResponse
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.database import get_db, engine, Base
|
|
from app.security import get_current_user
|
|
from app.crud import create_song, get_setting, set_setting, get_all_songs_and_votes
|
|
from typing import Any
|
|
|
|
router = APIRouter(
|
|
prefix="/admin",
|
|
dependencies=[Security(get_current_user, scopes=["admin"])],
|
|
responses={404: {"description": "Not found"}},
|
|
)
|
|
|
|
|
|
def get_main_category(categories) -> int:
|
|
if np.sum(categories != None) == 1:
|
|
return np.argmax(categories != None, axis=0)
|
|
elif "h" in categories:
|
|
return np.argmax(categories == "h", axis=0)
|
|
else:
|
|
return np.argmax(categories != None, axis=0)
|
|
|
|
|
|
def get_youtube_id(url):
|
|
if url is None:
|
|
return None
|
|
|
|
youtube_regex = (
|
|
r'(https?://)?(www\.)?'
|
|
'(youtube|youtu|youtube-nocookie)\.(com|be)/'
|
|
'(watch\?v=|embed/|shorts/|v/|.+\?v=)?([^&=%\?]{11})')
|
|
|
|
youtube_regex_match = re.match(youtube_regex, url)
|
|
if youtube_regex_match:
|
|
return youtube_regex_match.group(6)
|
|
|
|
return None
|
|
|
|
|
|
def get_thumbnail(url):
|
|
if url is None:
|
|
return "/static/cover.jpg"
|
|
|
|
m = get_youtube_id(url)
|
|
if m:
|
|
thumbnail_url = "https://img.youtube.com/vi/" + m + "/mqdefault.jpg"
|
|
return thumbnail_url
|
|
elif "spotify" in url:
|
|
return re.findall(r'(https?://i.scdn.co/image[^"]+)', requests.get(url).text)[0]
|
|
else:
|
|
return "/static/cover.jpg"
|
|
|
|
|
|
def get_spotify_id(url):
|
|
if url is None:
|
|
return None
|
|
if "spotify" in url:
|
|
return url.split("/track/")[1]
|
|
else:
|
|
return None
|
|
|
|
|
|
@router.post("/load_list")
|
|
async def create_upload_file(include_non_singable: bool = False, db: Session = Depends(get_db)):
|
|
|
|
Base.metadata.drop_all(engine)
|
|
Base.metadata.create_all(engine)
|
|
|
|
song_list = pd.read_excel(os.environ['LIST_URL'])
|
|
|
|
song_list = song_list.replace({np.nan: None})
|
|
song_list = song_list.replace({"n/a": None})
|
|
|
|
category_names = list(song_list.iloc[0][8:17])
|
|
|
|
for i, row in song_list[1:].iterrows():
|
|
|
|
row = np.array(row)
|
|
|
|
if (row[17] == "nein") and not include_non_singable:
|
|
continue
|
|
|
|
if not row[2]: # no title
|
|
continue
|
|
|
|
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[8:17] != None)}
|
|
|
|
if (not np.any(list(categories.values()))):
|
|
main_category = None
|
|
else:
|
|
main_category = category_names[get_main_category(row[8:17])]
|
|
|
|
create_song(db,
|
|
og_artist=row[0],
|
|
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_current=row[5] == "ja",
|
|
is_aca=row[6] == "ja",
|
|
arng_url=row[7],
|
|
categories=categories,
|
|
main_category=main_category,
|
|
singable=row[17] != "nein",
|
|
comment=row[18]
|
|
)
|
|
|
|
|
|
@router.post("/toggle_veto_mode")
|
|
async def toggle_veto_mode(db: Session = Depends(get_db)) -> bool:
|
|
veto_setting = get_setting(db, "veto_mode")
|
|
if veto_setting:
|
|
set_setting(db, "veto_mode", False)
|
|
return False
|
|
else:
|
|
set_setting(db, "veto_mode", True)
|
|
return True
|
|
|
|
|
|
@router.get("/results")
|
|
async def get_evaluation(format: str = "json", db = Depends(get_db)) -> Any:
|
|
res = get_all_songs_and_votes(db)
|
|
if format == "json":
|
|
return JSONResponse(content=res)
|
|
elif format == "csv":
|
|
df = pd.json_normalize(res)
|
|
df.columns = [c.split(".")[-1] for c in df.columns]
|
|
return Response(content=df.to_csv(), media_type="application/csv", headers={'Content-Disposition': 'attachment; filename="results.csv"'})
|
|
else:
|
|
raise HTTPException(status_code=404, detail="format must be json or csv")
|