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

1 files changed, 11 insertions, 1 deletions

diff --git a/archaeological_finds/serializers.py b/archaeological_finds/serializers.py
index 7e79123e6..e130a0ef8 100644
--- a/archaeological_finds/serializers.py
+++ b/archaeological_finds/serializers.py
@@ -43,7 +43,8 @@ def find_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)
@@ -94,6 +95,15 @@ def find_serialization(archive=False, return_empty_types=False,
 
     result = generic_get_results(FIND_MODEL_LIST, "finds",
                                  result_queryset=result_queryset, no_geo=no_geo)
+    if put_locks:
+        for model in FIND_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,