fix: add site deletion to delete_user script

techmatters · Aug 31, 2023 · d5d744a · d5d744a
1 parent f54b9cc
commit d5d744a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/terraso_backend/apps/core/management/commands/delete_user.py b/terraso_backend/apps/core/management/commands/delete_user.py
@@ -19,6 +19,7 @@
 
 from apps.core.models import User
 from apps.project_management.models import Project
+from apps.project_management.models.sites import Site
 
 
 class Command(BaseCommand):
@@ -50,6 +51,12 @@ def handle(self, *args, **options):
         for project in projects:
             project.delete()
 
+        # sites created by the user
+        sites = Site.objects.filter(owner_id=user.id)
+
+        for site in sites:
+            site.delete()
+
         # NOTE: user deletion currently fails due to audit logs
         try:
             user.delete()