fastapi-todolist/main.py

from fastapi import FastAPI, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
from sqlalchemy import select
from sqlalchemy.exc import SQLAlchemyError

from dependencies import SessionDep
from models import Todo

app = FastAPI()
app.add_middleware(
        CORSMiddleware,
        allow_origins=["*"],
        allow_credentials=True,
        allow_methods=["*"],
        allow_headers=["*"],
)

class todocreate(BaseModel):
    task: str

class todoupdate(BaseModel):
    completion: bool  | None = None
    task: str | None = None

class todoresponse(BaseModel):
    id: int
    task: str
    completion: bool
@app.get("/health")
def health():
    return {"status":"ok"}

@app.get("/todos", response_model=list[todoresponse])
def todos_get(session: SessionDep):
    statement = select(Todo)
    todos = session.scalars(statement).all()
    return todos

@app.post("/todos", response_model=todoresponse)
def todos_create(
    todo_data: todocreate,
    session: SessionDep,
):
    try:
        new_todo = Todo(task=todo_data.task)
        session.add(new_todo)
        session.commit()
        return new_todo
    except SQLAlchemyError:
        session.rollback()
        raise


@app.patch("/todos/{todo_id}", response_model=todoresponse)
def todos_patch(
    todo_id: int,
    update_data: todoupdate,
    session: SessionDep,
):
    try:
        todo = session.get(Todo, todo_id)
        if todo is None:
            raise HTTPException(status_code = 404, detail = "Todo not found")
        if update_data.task is not None:
            todo.task = update_data.task
        if update_data.completion is not None:
            todo.completion = update_data.completion
        session.commit()
        return todo
    except SQLAlchemyError:
        session.rollback()
        raise


@app.delete("/todos/{todo_id}", response_model=todoresponse)
def todos_delete(
    todo_id: int,
    session: SessionDep,
):
    try:
        todo = session.get(Todo, todo_id)
        if todo is None:
            raise HTTPException(status_code = 404, detail = "Todo not found")
        session.delete(todo)
        session.commit()
        return todo
    except SQLAlchemyError:
        session.rollback()
        raise