59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
from fastapi import FastAPI, Request, Depends
|
|
from app.routers import admin, user, songs, session
|
|
from fastapi.staticfiles import StaticFiles
|
|
from fastapi.responses import HTMLResponse
|
|
from fastapi.templating import Jinja2Templates
|
|
from app.database import engine, Base, get_db
|
|
from app.crud import get_songs_and_vote_for_session
|
|
from sqlalchemy.orm import Session
|
|
from typing import Annotated
|
|
from app.schemas import Song
|
|
|
|
Base.metadata.create_all(engine)
|
|
|
|
app = FastAPI()
|
|
|
|
app.include_router(admin.router)
|
|
app.include_router(user.router)
|
|
app.include_router(songs.router)
|
|
app.include_router(session.router)
|
|
|
|
app.mount("/static", StaticFiles(directory="static"), name="static")
|
|
|
|
templates = Jinja2Templates(directory="templates")
|
|
|
|
|
|
@app.get("/")
|
|
async def root(request: Request) -> HTMLResponse:
|
|
return templates.TemplateResponse(
|
|
request=request, name="landing.html"
|
|
)
|
|
|
|
|
|
@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)]
|
|
|
|
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,
|
|
"all_categories": {c: i+1 for i, c in enumerate(all_categories)},
|
|
"session_id": session_id
|
|
}
|
|
)
|