Serialization - Import/Export: manage put an release of locks

1 files changed, 12 insertions, 1 deletions

diff --git a/archaeological_operations/serializers.py b/archaeological_operations/serializers.py
index 6fd0f10d6..91b0b1b0f 100644
--- a/archaeological_operations/serializers.py
+++ b/archaeological_operations/serializers.py
@@ -57,7 +57,8 @@ def operation_serialization(archive=False, return_empty_types=False,
                             archive_name=None, operation_queryset=None,
                             site_queryset=None, cr_queryset=None,
                             find_queryset=None, warehouse_queryset=None,
-                            get_queryset=False, no_geo=True):
+                            get_queryset=False, no_geo=True,
+                            put_locks=False, lock_user=None):
     result_queryset = {}
     if operation_queryset:
         operation_ids = operation_queryset.values_list("id", flat=True)
@@ -124,6 +125,16 @@ def operation_serialization(archive=False, return_empty_types=False,
 
     result = generic_get_results(OPERATION_MODEL_LIST, "operations",
                                  result_queryset=result_queryset, no_geo=no_geo)
+
+    if put_locks:
+        for model in OPERATION_MODEL_LIST:
+            if not hasattr(model, "locked"):
+                continue
+            q = model.objects
+            if result_queryset and model.__name__ in result_queryset:
+                q = result_queryset[model.__name__]
+            q.update(locked=True, lock_user=lock_user)
+
     full_archive = archive_serialization(
         result, archive_dir="operations", archive=archive,
         return_empty_types=return_empty_types, archive_name=archive_name,