Merge pull request #4 from snakesonabrain/master

Stopping celery task
czue · Sep 10, 2019 · 9c60b3e · 9c60b3e
2 parents ca4359f + 4333c0a
commit 9c60b3e
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -99,3 +99,6 @@ ENV/
 
 # mypy
 .mypy_cache/
+
+# PyCharm files
+.idea/
diff --git a/celery_progress/backend.py b/celery_progress/backend.py
@@ -39,6 +39,18 @@ def set_progress(self, current, total):
             }
         )
 
+    def stop_task(self, current, total, exc):
+        self.task.update_state(
+            state='FAILURE',
+            meta={
+                'current': current,
+                'total': total,
+                'percent': 100.0,
+                'exc_message': str(exc),
+                'exc_type': str(type(exc))
+            }
+        )
+
 
 class Progress(object):