diff --git a/models/form.py b/models/form.py index acb630a..16ff761 100644 --- a/models/form.py +++ b/models/form.py @@ -34,7 +34,7 @@ class BaseQuestion(BaseModel): question_type: QuestionType label: str = Field(min_length=1) description: str | None = Field(None, min_length=1) - image_url: str | None + image_url: str | None = None required: bool = True @field_serializer("id") diff --git a/routes/form.py b/routes/form.py index 55e99ad..4c4dcf0 100644 --- a/routes/form.py +++ b/routes/form.py @@ -74,10 +74,11 @@ async def user_forms(user: User): @router.get("/get") -async def get_form(id: int) -> Form: +async def get_form(id: int) -> Form | None: 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() - return Form.model_validate(form) + if form is not None: + return Form.model_validate(form)