Merge pull request #357 from python-discord/joseph/change-roles-field-to-postgres-array

Change roles field on user model to Postgres array

1 files changed, 24 insertions, 5 deletions

diff --git a/pydis_site/apps/api/models/bot/user.py b/pydis_site/apps/api/models/bot/user.py
index 65e8751e..bff4d642 100644
--- a/pydis_site/apps/api/models/bot/user.py
+++ b/pydis_site/apps/api/models/bot/user.py
@@ -1,3 +1,5 @@
+from django.contrib.postgres.fields import ArrayField
+from django.core.exceptions import ValidationError
 from django.core.validators import MaxValueValidator, MinValueValidator
 from django.db import models
 
@@ -5,6 +7,14 @@ from pydis_site.apps.api.models.bot.role import Role
 from pydis_site.apps.api.models.utils import ModelReprMixin
 
 
+def _validate_existing_role(value: int) -> None:
+    """Validate that a role exists when given in to the user model."""
+    role = Role.objects.filter(id=value)
+
+    if not role:
+        raise ValidationError(f"Role with ID {value} does not exist")
+
+
 class User(ModelReprMixin, models.Model):
     """A Discord user."""
 
@@ -31,9 +41,18 @@ class User(ModelReprMixin, models.Model):
         ),
         help_text="The discriminator of this user, taken from Discord."
     )
-    roles = models.ManyToManyField(
-        Role,
-        help_text="Any roles this user has on our server."
+    roles = ArrayField(
+        models.BigIntegerField(
+            validators=(
+                MinValueValidator(
+                    limit_value=0,
+                    message="Role IDs cannot be negative."
+                ),
+                _validate_existing_role
+            )
+        ),
+        default=list,
+        help_text="IDs of roles the user has on the server"
     )
     in_guild = models.BooleanField(
         default=True,
@@ -51,7 +70,7 @@ class User(ModelReprMixin, models.Model):
 
         This will fall back to the Developers role if the user does not have any roles.
         """
-        roles = self.roles.all()
+        roles = Role.objects.filter(id__in=self.roles)
         if not roles:
             return Role.objects.get(name="Developers")
-        return max(self.roles.all())
+        return max(roles)