blob: 9fffddeb83c6e6c5a550e7d417dad7255e812678 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
import typing as t
from pydantic import BaseModel, Field, root_validator
from backend.constants import FormFeatures
from .antispam import AntiSpam
from .discord_user import DiscordUser
class FormResponse(BaseModel):
"""Schema model for form response."""
id: str = Field(alias="_id")
user: t.Optional[DiscordUser]
antispam: t.Optional[AntiSpam]
response: t.Dict[str, t.Any]
form_id: str
class Config:
allow_population_by_field_name = True
@root_validator
def validate_data(cls, values: t.Dict[str, t.Any]) -> t.Dict[str, t.Any]:
"""Validates is all required (based on flags) is provided."""
flags = values.get("flags", [])
if FormFeatures.DISABLE_ANTISPAM not in flags and values.get("antispam") is None: # noqa
raise ValueError("Antispam information required.")
if FormFeatures.REQUIRES_LOGIN in flags:
if values.get("user") is None:
raise ValueError("User information required.")
values["user"]["require_email"] = FormFeatures.COLLECT_EMAIL in flags
return values
|
