diff options
| author | 2020-12-17 10:05:21 +0300 | |
|---|---|---|
| committer | 2020-12-17 10:05:21 +0300 | |
| commit | b7951972ef5a45c36c9372823cf6d2aa65a652d2 (patch) | |
| tree | 3c4ad6c7aeb7ec69c47f10c45fe10ca466c3d989 /backend/models | |
| parent | Adds Webhook Option (diff) | |
| parent | Merge pull request #38 from python-discord/submission-timestamp (diff) | |
Merge branch 'main' into discord-webhook
Diffstat (limited to 'backend/models')
| -rw-r--r-- | backend/models/form_response.py | 15 | 
1 files changed, 14 insertions, 1 deletions
| diff --git a/backend/models/form_response.py b/backend/models/form_response.py index f3296cd..0da7b15 100644 --- a/backend/models/form_response.py +++ b/backend/models/form_response.py @@ -1,6 +1,7 @@ +import datetime  import typing as t -from pydantic import BaseModel, Field +from pydantic import BaseModel, Field, validator  from .antispam import AntiSpam  from .discord_user import DiscordUser @@ -14,6 +15,18 @@ class FormResponse(BaseModel):      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 | 
