aboutsummaryrefslogtreecommitdiffstats
path: root/backend/models/form_response.py
blob: 933f5e4ea0af801e60039a02c3c8416f151ecdbd (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 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: t.Optional[DiscordUser]
    antispam: t.Optional[AntiSpam]
    response: dict[str, t.Any]
    form_id: str
    timestamp: str

    @validator("timestamp", pre=True)
    def set_timestamp(cls, iso_string: t.Optional[str]) -> t.Optional[str]:
        if iso_string is None:
            return datetime.datetime.now(tz=datetime.timezone.utc).isoformat()

        elif not isinstance(iso_string, str):
            raise ValueError("Submission timestamp must be a string.")

        # 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__: t.List[FormResponse]