Merge pull request #10837 from archesproject/9788_add_edit_log_manage…

…ment_command

Add clear edit log management command #9788

arches/management/commands/resources.py

@@ -44,11 +44,22 @@ def add_arguments(self, parser):
             help="A graphid of the Resource Model you would like to remove all instances from.",
         )
 
+        parser.add_argument(
+            "-e",
+            "--editlog",
+            action="store_true",
+            dest="editlog",
+            help="used to clear the edit log. If a graphid is provided, only the edit log for that graph will be cleared.",
+        )
+
     def handle(self, *args, **options):
         if options["operation"] == "remove_resources":
-            self.remove_resources(force=options["yes"], graphid=options["graph"])
+            self.remove_resources(force=options["yes"], graphid=options["graph"], clear_edit_log=options["editlog"])
+
+        if options["operation"] == "clear_edit_log":
+            self.clear_edit_log(graphid=options["graph"])
 
-    def remove_resources(self, load_id="", graphid=None, force=False):
+    def remove_resources(self, load_id="", graphid=None, force=False, clear_edit_log=False):
         """
         Runs the resource_remover command found in data_management.resources
         """
@@ -66,8 +77,22 @@ def remove_resources(self, load_id="", graphid=None, force=False):
 
         if graphid is None:
             resource_remover.clear_resources()
+            if clear_edit_log:
+                self.clear_edit_log()
         else:
             graph = Graph.objects.get(graphid=graphid)
             graph.delete_instances(verbose=True)
+            if clear_edit_log:
+                self.clear_edit_log(graphid)
 
         return
+
+    def clear_edit_log(self, graphid=None):
+        """
+        Clears the edit log
+        """
+        if graphid:
+            models.EditLog.objects.filter(resourceclassid=graphid).delete()
+        else:
+            models.EditLog.objects.all().delete()
+