Bans

bans/models.py

@@ -1,6 +1,8 @@
+from collections.abc import Iterable
 from django.db import models
 from uuid import uuid4
-
+from users.models import UserProfile
+from django.db.models.signals import post_save, post_delete, pre_save
 # Create your models here.
 
 
@@ -25,7 +27,7 @@ class SSOBan(models.Model):
 
     id = models.UUIDField(primary_key=True, default=uuid4, blank=False)
     banned_user = models.ForeignKey(
-        to="users.UserProfile", related_name="banned_user", on_delete=models.CASCADE
+        to="users.UserProfile", related_name="banned_user", on_delete=models.CASCADE, null=True, blank=True
     )
     time_of_creation = models.DateTimeField(auto_now_add=True)
     reason = models.CharField(max_length=30, choices=BAN_REASON_CHOICHES)
@@ -38,3 +40,12 @@ class SSOBan(models.Model):
         null=True,
         blank=True,
     )
+    banned_user_ldapid = models.CharField(max_length=20, blank=True, null=True)
+
+
+    def save(self, *args, **kwargs) -> None:
+        if self.banned_user_ldapid:
+            self.banned_user = UserProfile.objects.get(ldap_id=self.banned_user_ldapid)
+        return super().save(*args, **kwargs)
+
+