From 6fb231acaabcb00817e571a06c0ca4d2f60f976f Mon Sep 17 00:00:00 2001 From: Joe Banks Date: Wed, 16 Dec 2020 23:26:30 +0000 Subject: Make model changes --- backend/models/__init__.py | 14 +++++++++++--- backend/models/form.py | 4 ++++ backend/models/form_response.py | 4 ++++ 3 files changed, 19 insertions(+), 3 deletions(-) (limited to 'backend/models') diff --git a/backend/models/__init__.py b/backend/models/__init__.py index 98fa619..29ccb24 100644 --- a/backend/models/__init__.py +++ b/backend/models/__init__.py @@ -1,7 +1,15 @@ from .antispam import AntiSpam from .discord_user import DiscordUser -from .form import Form -from .form_response import FormResponse +from .form import Form, FormList +from .form_response import FormResponse, ResponseList from .question import Question -__all__ = ["AntiSpam", "DiscordUser", "Form", "FormResponse", "Question"] +__all__ = [ + "AntiSpam", + "DiscordUser", + "Form", + "FormResponse", + "Question", + "FormList", + "ResponseList" +] diff --git a/backend/models/form.py b/backend/models/form.py index cb58065..9d8ffaa 100644 --- a/backend/models/form.py +++ b/backend/models/form.py @@ -52,3 +52,7 @@ class Form(BaseModel): returned_data = data return returned_data + + +class FormList(BaseModel): + __root__: t.List[Form] diff --git a/backend/models/form_response.py b/backend/models/form_response.py index f3296cd..8bc69e4 100644 --- a/backend/models/form_response.py +++ b/backend/models/form_response.py @@ -17,3 +17,7 @@ class FormResponse(BaseModel): class Config: allow_population_by_field_name = True + + +class ResponseList(BaseModel): + __root__: t.List[FormResponse] -- cgit v1.2.3