From f093907393b4f479f62409286831b938ba65c558 Mon Sep 17 00:00:00 2001
From: D0rs4n <41237606+D0rs4n@users.noreply.github.com>
Date: Tue, 31 Aug 2021 16:27:18 +0200
Subject: Create a signal to unassign roles from user when deleted

Add a signal to the api app that automatically unassigns
deleted roles from users that have them
---
 pydis_site/apps/api/signals.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)
 create mode 100644 pydis_site/apps/api/signals.py

(limited to 'pydis_site/apps/api/signals.py')

diff --git a/pydis_site/apps/api/signals.py b/pydis_site/apps/api/signals.py
new file mode 100644
index 00000000..c69704b1
--- /dev/null
+++ b/pydis_site/apps/api/signals.py
@@ -0,0 +1,12 @@
+from django.db.models.signals import pre_delete
+from django.dispatch import receiver
+
+from pydis_site.apps.api.models.bot import Role, User
+
+
+@receiver(signal=pre_delete, sender=Role)
+def delete_role_from_user(sender: Role, instance: Role, **kwargs) -> None:
+    """Unassigns the Role (instance) that is being deleted from every user that has it."""
+    for user in User.objects.filter(roles__contains=[instance.id]):
+        del user.roles[user.roles.index(instance.id)]
+        user.save()
-- 
cgit v1.2.3