mirror of
https://github.com/grey-cat-1908/formaptix-server.git
synced 2024-11-11 18:57:27 +03:00
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
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)
|
|
)
|
|
]
|
|
}
|