from typing import Optional from fastapi import APIRouter, Depends, HTTPException, status from sqlalchemy.orm import Session from app.db.database import get_db from app.models.todo import Todo from app.schemas.todo import Todo as TodoSchema from app.schemas.todo import TodoCreate, TodoUpdate router = APIRouter() @router.post("/", response_model=TodoSchema, status_code=status.HTTP_201_CREATED) def create_todo(todo: TodoCreate, db: Session = Depends(get_db)): """ Create a new todo item. """ db_todo = Todo( title=todo.title, description=todo.description, completed=todo.completed ) db.add(db_todo) db.commit() db.refresh(db_todo) return db_todo @router.get("/", response_model=list[TodoSchema]) def read_todos( skip: int = 0, limit: int = 100, completed: Optional[bool] = None, db: Session = Depends(get_db) ): """ Retrieve all todo items with optional filtering. """ query = db.query(Todo) # Filter by completion status if specified if completed is not None: query = query.filter(Todo.completed == completed) todos = query.offset(skip).limit(limit).all() return todos @router.get("/{todo_id}", response_model=TodoSchema) def read_todo(todo_id: int, db: Session = Depends(get_db)): """ Retrieve 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=TodoSchema) 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 fields that are provided todo_data = todo.model_dump(exclude_unset=True) for key, value in todo_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, response_model=None ) 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