From bc68d2f7363d81c3a36f4c5641377dd4d024a932 Mon Sep 17 00:00:00 2001
From: ks129 <45097959+ks129@users.noreply.github.com>
Date: Tue, 8 Dec 2020 10:41:04 +0200
Subject: Create DiscordUser Pydantic model for form response

---
 backend/models/discord_user.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 backend/models/discord_user.py

(limited to 'backend/models')

diff --git a/backend/models/discord_user.py b/backend/models/discord_user.py
new file mode 100644
index 0000000..99d6248
--- /dev/null
+++ b/backend/models/discord_user.py
@@ -0,0 +1,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
-- 
cgit v1.2.3