From 69e21b1290d10aea42a99a90ec9387c9e6bdcf60 Mon Sep 17 00:00:00 2001
From: grey-cat-1908 <61203964+grey-cat-1908@users.noreply.github.com>
Date: Mon, 12 Aug 2024 14:02:12 +0000
Subject: [PATCH] createform response model

---
 routes/form.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/routes/form.py b/routes/form.py
index 56617cf..8c374c9 100644
--- a/routes/form.py
+++ b/routes/form.py
@@ -2,14 +2,18 @@ from fastapi import APIRouter
 from sqlalchemy import select
 
 import database
-from models import FormData, Form
+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):
+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()
@@ -19,7 +23,7 @@ async def create_form(user: User, form_data: FormData):
         await session.flush()
         await session.refresh(form)
 
-        return {"status": "success", "form_id": form.id}
+        return CreateForm(form_id=form.id)
 
 
 @router.get("/my")