format, sort categories

This commit is contained in:
matthias@matsewe.de
2024-05-24 18:21:09 +02:00
parent d1864052cc
commit ae3b3a4a4f
5 changed files with 22 additions and 12 deletions

View File

@@ -22,6 +22,7 @@ app.mount("/static", StaticFiles(directory="static"), name="static")
templates = Jinja2Templates(directory="templates")
@app.get("/")
async def root(request: Request) -> HTMLResponse:
return templates.TemplateResponse(
@@ -29,22 +30,29 @@ async def root(request: Request) -> HTMLResponse:
)
@app.get("/vote")
async def vote(request: Request, session_id : str, db: Annotated[Session, Depends(get_db)]) -> HTMLResponse:
songs = [Song(**s.__dict__, vote=v) for s, v in get_songs_and_vote_for_session(db, session_id)]
async def vote(request: Request, session_id: str, db: Annotated[Session, Depends(get_db)]) -> HTMLResponse:
songs = [Song(**s.__dict__, vote=v)
for s, v in get_songs_and_vote_for_session(db, session_id)]
songs_by_category = {}
all_categories = set()
for song in songs:
if song.main_category not in songs_by_category:
songs_by_category[song.main_category] = []
songs_by_category[song.main_category].append(song)
all_categories.update(song.categories.keys())
all_categories = list(all_categories)
all_categories.sort()
print(all_categories)
return templates.TemplateResponse(
request=request, name="voting.html", context={
"songs_by_category": songs_by_category,
"songs_by_category": songs_by_category,
"all_categories": {c: i+1 for i, c in enumerate(all_categories)},
"session_id": session_id
}
)
}
)