aboutsummaryrefslogtreecommitdiffstats
path: root/backend/models/discord_user.py
blob: be106722a2cd62c39d8fdf12cac825fcc5eed8f0 (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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import datetime
import typing as t

from pydantic import BaseModel


class _User(BaseModel):
    """Base for discord users and members."""

    # Discord default fields.
    username: str
    id: str
    discriminator: str
    avatar: str | None
    bot: bool | None
    system: bool | None
    locale: str | None
    verified: bool | None
    email: str | None
    flags: int | None
    premium_type: int | None
    public_flags: int | None


class DiscordUser(_User):
    """Schema model of Discord user for form response."""

    # Custom fields
    admin: bool


class DiscordMember(BaseModel):
    """A discord guild member."""

    user: _User
    nick: str | None
    avatar: str | None
    roles: list[str]
    joined_at: datetime.datetime
    premium_since: datetime.datetime | None
    deaf: bool
    mute: bool
    pending: bool | None
    permissions: str | None
    communication_disabled_until: datetime.datetime | None

    def dict(self, *args, **kwargs) -> dict[str, t.Any]:
        """Convert the model to a python dict, and encode timestamps in a serializable format."""
        data = super().dict(*args, **kwargs)
        for field, value in data.items():
            if isinstance(value, datetime.datetime):
                data[field] = value.isoformat()

        return data