formaptix-server/routes/form.py

from fastapi import APIRouter, HTTPException
from sqlalchemy import select

import database
from models import FormData, Form, BaseModel
from .utils import User

router = APIRouter(prefix="/form")


class CreateForm(BaseModel):
    form_id: int


@router.post("/create")
async def create_form(user: User, form_data: FormData) -> CreateForm:
    async with database.sessions.begin() as session:
        form = database.Form(
            name=form_data.name, owner_id=user.id, data=form_data.model_dump()
        )

        session.add(form)
        await session.flush()
        await session.refresh(form)

        return CreateForm(form_id=form.id)


@router.delete("/delete")
async def delete_form(user: User, id: int):
    async with database.sessions.begin() as session:
        stmt = select(database.Form).where(
            database.Form.id == id
        )
        db_request = await session.execute(stmt)
        form = db_request.scalar_one_or_none()

        if form is None:
            raise HTTPException(404, "No form was found")
        if form.owner_id != user.id:
            raise HTTPException(403, "Forbidden")
        
        await session.delete(form)


@router.get("/my")
async def user_forms(user: User):
    async with database.sessions.begin() as session:
        return {
            "forms": [
                Form.model_validate(item)
                for item in await session.scalars(
                    select(database.Form).where(database.Form.owner_id == user.id)
                )
            ]
        }