From b8a888a5328ac224e34323e13c83ffe699f43acc Mon Sep 17 00:00:00 2001
From: Remy van der Wereld <r.van.der.wereld@amsterdam.nl>
Date: Tue, 17 Sep 2024 20:59:13 +0200
Subject: [PATCH] Cases are floating when user is deleted

---
 .../users/user_federation_delete.html         |  4 ++--
 app/web/users/views.py                        | 20 +++++++++++++++++++
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/app/web/users/templates/users/user_federation_delete.html b/app/web/users/templates/users/user_federation_delete.html
index 928cb94a..8f8e0edf 100644
--- a/app/web/users/templates/users/user_federation_delete.html
+++ b/app/web/users/templates/users/user_federation_delete.html
@@ -14,9 +14,9 @@
 
 {% block main %}
     <h2>Gebruiker verwijderen</h2>
-    <p class="u-margin-bottom-2x">Weet je zeker dat de gebruiker '{{ object.username }}' wilt verwijderen?</p>
+    <p class="u-margin-bottom-2x">Weet je zeker dat je gebruiker <strong>'{{ object.username }}'</strong> wilt verwijderen?</p>
     {% if cases %}
-    <p>De onderstaande cliënten hebben na het verwijderen van deze gebruiker geen persoonlijk begeleider meer in de omslagroute.</p>
+    <p>De onderstaande cliënten hebben na het verwijderen van deze gebruiker geen persoonlijk begeleider meer in de omslagroute en zullen aan jou worden toegekend.</p>
     <ul>
     {% for case in cases %}
         <li>cliënt id: {{ case.id }}</li>
diff --git a/app/web/users/views.py b/app/web/users/views.py
index 9f92c0c3..8abe64db 100644
--- a/app/web/users/views.py
+++ b/app/web/users/views.py
@@ -379,6 +379,26 @@ def get_queryset(self):
 
     def test_func(self):
         return auth_test(self.request.user, [PB_FEDERATIE_BEHEERDER, FEDERATIE_BEHEERDER, WONEN])
+    
+    
+    def post(self, request, *args, **kwargs):
+        # Get user who will be deleted
+        deleted_user = self.get_object()
+        # Get all cases from the deleted user's profile
+        deleted_user_profile_cases = deleted_user.profile.cases.all()
+        
+        # Loop through all cases in profile
+        for profile_case in deleted_user_profile_cases:
+            # Check if case exists in other profiles
+            profiles_with_case_count = Profile.objects.filter(cases=profile_case).count()
+            if profiles_with_case_count <= 1:
+                # There is no other profile with this case
+                # Add this case to current user
+                self.request.user.profile.cases.add(profile_case)
+
+        # Delete user profile as well
+        deleted_user.profile.delete()
+        return super().delete(request, *args, **kwargs)
 
 
 class OIDCAuthenticationRequestView(DatapuntOIDCAuthenticationRequestView):