'code'

src/routers/stats.py

@@ -2,22 +2,24 @@
 Task statistics router
 Provides endpoints for calculating and retrieving task statistics
 """
-from fastapi import APIRouter, Depends, HTTPException
-from sqlalchemy.orm import Session
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlmodel import Session, select, func
 from datetime import datetime, timedelta
-from src.database import get_db
-from src.models.task import Task
-from src.models.user import User
-from src.middleware.auth import get_current_user
+from uuid import UUID
+
+from ..database import get_session_dep
+from ..models.task import Task
+from ..models.user import User
+from ..utils.deps import get_current_user
 
 router = APIRouter(prefix="/api", tags=["stats"])
 
 
 @router.get("/{user_id}/tasks/stats")
 async def get_task_stats(
-    user_id: str,
+    user_id: UUID,
     current_user: User = Depends(get_current_user),
-    db: Session = Depends(get_db)
+    session: Session = Depends(get_session_dep)
 ):
     """
     Get task statistics for a user.
@@ -25,10 +27,10 @@ async def get_task_stats(
     """
     # Verify user is requesting their own stats
     if current_user.id != user_id:
-        raise HTTPException(status_code=403, detail="Unauthorized")
+        raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="Unauthorized")
     
     # Get all tasks for user
-    tasks = db.query(Task).filter(Task.user_id == user_id).all()
+    tasks = session.exec(select(Task).where(Task.user_id == user_id)).all()
     
     total = len(tasks)
     completed = len([t for t in tasks if t.completed])