blob: 3c8297b48eb07849b7b8b7392dfcfa1f80761637 (
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
37
|
import datetime
import typing as t
from pydantic import BaseModel, Field, validator
from .antispam import AntiSpam
from .discord_user import DiscordUser
class FormResponse(BaseModel):
"""Schema model for form response."""
id: str = Field(alias="_id")
user: DiscordUser | None
antispam: AntiSpam | None
response: dict[str, t.Any]
form_id: str
timestamp: str
@validator("timestamp", pre=True)
def set_timestamp(cls, iso_string: str | None) -> str:
if iso_string is None:
return datetime.datetime.now(tz=datetime.UTC).isoformat()
if not isinstance(iso_string, str):
msg = "Submission timestamp must be a string."
raise TypeError(msg)
# Convert to datetime and back to ensure string is valid
return datetime.datetime.fromisoformat(iso_string).isoformat()
class Config:
allow_population_by_field_name = True
class ResponseList(BaseModel):
__root__: list[FormResponse]
|