Add task completed debugging

ministryofjustice · Nov 29, 2024 · c8a7052 · c8a7052
1 parent 184c6fc
commit c8a7052
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/controlpanel/api/tasks/handlers/base.py b/controlpanel/api/tasks/handlers/base.py
@@ -19,9 +19,12 @@ class BaseTaskHandler(CeleryTask):
     task_obj = None
 
     def complete(self):
-        if self.task_obj:
-            self.task_obj.completed = True
-            self.task_obj.save()
+        if not self.task_obj:
+            return log.warn("No Task object to mark as completed.")
+
+        self.task_obj.completed = True
+        self.task_obj.save()
+        log.info(f"Task object completed: {self.task_obj.task_id}")
 
     def get_task_obj(self):
         task_id = self.request.id