2 files changed, 63 insertions, 2 deletions
diff --git a/backend/models/discord_role.py b/backend/models/discord_role.py
new file mode 100644
index 0000000..9f0b7dd
--- /dev/null
+++ b/backend/models/discord_role.py
@@ -0,0 +1,40 @@
+import typing
+
+from pydantic import BaseModel
+
+
+class RoleTags(BaseModel):
+    """Meta information about a discord role."""
+
+    bot_id: typing.Optional[str]
+    integration_id: typing.Optional[str]
+    premium_subscriber: bool
+
+    def __init__(self, **data: typing.Any):
+        """
+        Handle the terrible discord API.
+
+        Discord only returns the premium_subscriber field if it's true,
+        meaning the typical validation process wouldn't work.
+
+        We manually parse the raw data to determine if the field exists, and give it a useful
+        bool value.
+        """
+        data["premium_subscriber"] = "premium_subscriber" in data.keys()
+        super().__init__(**data)
+
+
+class DiscordRole(BaseModel):
+    """Schema model of Discord guild roles."""
+
+    id: str
+    name: str
+    color: int
+    hoist: bool
+    icon: typing.Optional[str]
+    unicode_emoji: typing.Optional[str]
+    position: int
+    permissions: str
+    managed: bool
+    mentionable: bool
+    tags: typing.Optional[RoleTags]
diff --git a/backend/models/discord_user.py b/backend/models/discord_user.py
index 9f246ba..3f4209d 100644
--- a/backend/models/discord_user.py
+++ b/backend/models/discord_user.py
@@ -1,10 +1,11 @@
+import datetime
 import typing as t
 
 from pydantic import BaseModel
 
 
-class DiscordUser(BaseModel):
-    """Schema model of Discord user for form response."""
+class _User(BaseModel):
+    """Base for discord users and members."""
 
     # Discord default fields.
     username: str
@@ -20,5 +21,25 @@ class DiscordUser(BaseModel):
     premium_type: t.Optional[int]
     public_flags: t.Optional[int]
 
+
+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: t.Optional[str]
+    avatar: t.Optional[str]
+    roles: list[str]
+    joined_at: datetime.datetime
+    premium_since: t.Optional[datetime.datetime]
+    deaf: bool
+    mute: bool
+    pending: t.Optional[bool]
+    permissions: t.Optional[str]
+    communication_disabled_until: t.Optional[datetime.datetime]