Make archive import/export tasks serializable for Celery tasks

1 files changed, 11 insertions, 2 deletions

diff --git a/ishtar_common/tasks.py b/ishtar_common/tasks.py
index 07286ad76..eec97d68a 100644
--- a/ishtar_common/tasks.py
+++ b/ishtar_common/tasks.py
@@ -31,6 +31,7 @@ from ishtar_common.models import Town, Department
 from ishtar_common.utils import task
 from ishtar_common.serializers import full_serialization, restore_serialized
 
+from ishtar_common.models import ImportTask, ExportTask
 from archaeological_operations.models import Operation, ArchaeologicalSite
 from archaeological_context_records.models import ContextRecord
 from archaeological_finds.models import Find
@@ -43,7 +44,11 @@ from archaeological_warehouse.views import get_warehouse
 
 
 @task()
-def launch_import(import_task):
+def launch_import(import_task_id):
+    try:
+        import_task = ImportTask.objects.get(pk=import_task_id)
+    except ImportTask.DoesNotExist:
+        return
     if import_task.state != 'S':
         return
     import_task.launch_date = datetime.datetime.now()
@@ -60,7 +65,11 @@ def launch_import(import_task):
 
 
 @task()
-def launch_export(export_task):
+def launch_export(export_task_id):
+    try:
+        export_task = ExportTask.objects.get(pk=export_task_id)
+    except ExportTask.DoesNotExist:
+        return
     if export_task.state != 'S':
         return
     export_task.launch_date = datetime.datetime.now()