47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
from fastapi import APIRouter, Request, Depends
|
|
from fastapi.responses import HTMLResponse
|
|
from fastapi.templating import Jinja2Templates
|
|
from sqlalchemy import select, desc
|
|
|
|
from app.models.base import async_session
|
|
from app.models.user import User
|
|
from app.models.workout import Workout
|
|
from app.models.checkin import Checkin
|
|
from app.auth import get_current_user
|
|
|
|
router = APIRouter()
|
|
templates = Jinja2Templates(directory="app/templates")
|
|
|
|
|
|
@router.get("/", response_class=HTMLResponse)
|
|
async def root_redirect():
|
|
from fastapi.responses import RedirectResponse
|
|
return RedirectResponse(url="/dashboard")
|
|
|
|
|
|
@router.get("/dashboard", response_class=HTMLResponse)
|
|
async def dashboard(request: Request, user: User = Depends(get_current_user)):
|
|
async with async_session() as session:
|
|
result = await session.execute(
|
|
select(Workout)
|
|
.where(Workout.user_id == user.id)
|
|
.order_by(desc(Workout.date))
|
|
.limit(5)
|
|
)
|
|
recent_workouts = result.scalars().all()
|
|
|
|
result = await session.execute(
|
|
select(Checkin)
|
|
.where(Checkin.user_id == user.id)
|
|
.order_by(desc(Checkin.date))
|
|
.limit(1)
|
|
)
|
|
latest_checkin = result.scalar_one_or_none()
|
|
|
|
return templates.TemplateResponse("dashboard.html", {
|
|
"request": request,
|
|
"user": user,
|
|
"recent_workouts": recent_workouts,
|
|
"latest_checkin": latest_checkin,
|
|
})
|