refactor: add cleanup manager function

quibble-dev · Dec 14, 2024 · d5a9cab · d5a9cab
1 parent a3cb20c
commit d5a9cab
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/backend/apps/comment/api/v1/viewsets.py b/backend/apps/comment/api/v1/viewsets.py
@@ -10,3 +10,5 @@ class CommentModelViewSet(viewsets.ModelViewSet):
 
     def perform_destroy(self, instance):
         CommentModel.objects.soft_delete(instance)  # pyright: ignore
+        # perform cleanup
+        CommentModel.objects.clean_up_soft_deleted()  # pyright: ignore
diff --git a/backend/apps/comment/managers.py b/backend/apps/comment/managers.py
@@ -12,3 +12,9 @@ def soft_delete(self, instance):
             instance.content = "[deleted]"
             instance.quibbler = None
             instance.save()
+
+    def clean_up_soft_deleted(self):
+        # cleanup all comment instances with no children
+        for comment_instance in self.filter(deleted=True):
+            if comment_instance.children_count == 0:
+                comment_instance.delete()