blob: 99d6248f6d63c84d7eb176b3153dcd6ac0b28c0d (
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
|
import typing as t
from pydantic import BaseModel, root_validator
class DiscordUser(BaseModel):
"""Schema model of Discord user for form response."""
# Discord default fields
id: int # This is actually snowflake, but we simplify it here
username: str
discriminator: str
avatar: t.Optional[str]
bot: t.Optional[bool]
system: t.Optional[bool]
locale: t.Optional[str]
verified: t.Optional[bool]
email: t.Optional[str]
flags: t.Optional[int]
premium_type: t.Optional[int]
public_flags: t.Optional[int]
# Custom fields
admin: bool
@root_validator
def validate_data(cls, values: t.Dict[str, t.Any]) -> t.Dict[str, t.Any]:
"""Validates email data when email collection is required."""
if values.get("require_email", False) is True:
if values.get("email") is None or values.get("verified") is None:
raise ValueError("Email information about user is required.")
return values
|
