from fastapi import APIRouter, Depends, HTTPException, status, Query from sqlalchemy.orm import Session from typing import List, Optional from app.database.config import get_db from app.models.todo import Todo, PriorityLevel from app.api.schemas import TodoCreate, TodoResponse, TodoUpdate router = APIRouter() @router.post("/", response_model=TodoResponse, status_code=status.HTTP_201_CREATED) def create_todo(todo: TodoCreate, db: Session = Depends(get_db)): """Create a new todo item""" db_todo = Todo(**todo.dict()) db.add(db_todo) db.commit() db.refresh(db_todo) return db_todo @router.get("/", response_model=List[TodoResponse]) def read_todos( skip: int = 0, limit: int = 100, priority: Optional[PriorityLevel] = Query( None, description="Filter todos by priority level" ), db: Session = Depends(get_db), ): """ Get all todo items with optional filtering by priority - **skip**: Number of records to skip - **limit**: Maximum number of records to return - **priority**: Optional filter by priority level (low, medium, high) """ query = db.query(Todo) if priority: query = query.filter(Todo.priority == priority) todos = query.offset(skip).limit(limit).all() return todos @router.get("/{todo_id}", response_model=TodoResponse) def read_todo(todo_id: int, db: Session = Depends(get_db)): """Get a specific todo item by ID""" db_todo = db.query(Todo).filter(Todo.id == todo_id).first() if db_todo is None: raise HTTPException(status_code=404, detail="Todo not found") return db_todo @router.put("/{todo_id}", response_model=TodoResponse) def update_todo(todo_id: int, todo: TodoUpdate, db: Session = Depends(get_db)): """Update a todo item""" db_todo = db.query(Todo).filter(Todo.id == todo_id).first() if db_todo is None: raise HTTPException(status_code=404, detail="Todo not found") update_data = todo.dict(exclude_unset=True) for key, value in update_data.items(): setattr(db_todo, key, value) db.commit() db.refresh(db_todo) return db_todo @router.delete("/{todo_id}", status_code=status.HTTP_204_NO_CONTENT) def delete_todo(todo_id: int, db: Session = Depends(get_db)): """Delete a todo item""" db_todo = db.query(Todo).filter(Todo.id == todo_id).first() if db_todo is None: raise HTTPException(status_code=404, detail="Todo not found") db.delete(db_todo) db.commit() return None