from typing import List from fastapi import APIRouter, Depends, HTTPException, status from sqlalchemy.orm import Session from app.db.database import get_db from app.models.task import Task as TaskModel from app.schemas.task import TaskCreate, TaskUpdate, Task router = APIRouter() @router.post("/", response_model=Task, status_code=status.HTTP_201_CREATED) def create_task(task: TaskCreate, db: Session = Depends(get_db)): """ Create a new task. """ db_task = TaskModel(**task.dict()) db.add(db_task) db.commit() db.refresh(db_task) return db_task @router.get("/", response_model=List[Task]) def read_tasks( skip: int = 0, limit: int = 100, completed: bool = None, priority: int = None, db: Session = Depends(get_db), ): """ Retrieve tasks with optional filtering. """ query = db.query(TaskModel) # Apply filters if provided if completed is not None: query = query.filter(TaskModel.completed == completed) if priority is not None: query = query.filter(TaskModel.priority == priority) # Apply pagination tasks = query.order_by(TaskModel.created_at.desc()).offset(skip).limit(limit).all() return tasks @router.get("/{task_id}", response_model=Task) def read_task(task_id: int, db: Session = Depends(get_db)): """ Get a specific task by ID. """ task = db.query(TaskModel).filter(TaskModel.id == task_id).first() if task is None: raise HTTPException(status_code=404, detail="Task not found") return task @router.put("/{task_id}", response_model=Task) def update_task(task_id: int, task_update: TaskUpdate, db: Session = Depends(get_db)): """ Update a task. """ db_task = db.query(TaskModel).filter(TaskModel.id == task_id).first() if db_task is None: raise HTTPException(status_code=404, detail="Task not found") # Update task with provided values, skip None values update_data = task_update.dict(exclude_unset=True) for key, value in update_data.items(): setattr(db_task, key, value) db.commit() db.refresh(db_task) return db_task @router.delete("/{task_id}", status_code=status.HTTP_204_NO_CONTENT, response_model=None) def delete_task(task_id: int, db: Session = Depends(get_db)): """ Delete a task. """ db_task = db.query(TaskModel).filter(TaskModel.id == task_id).first() if db_task is None: raise HTTPException(status_code=404, detail="Task not found") db.delete(db_task) db.commit() return None