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

1 files changed, 11 insertions, 1 deletions

diff --git a/archaeological_context_records/serializers.py b/archaeological_context_records/serializers.py
index 9f454eb35..a41e2bd40 100644
--- a/archaeological_context_records/serializers.py
+++ b/archaeological_context_records/serializers.py
@@ -49,7 +49,8 @@ def generate_warehouse_queryset(ids):
 def cr_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):
+                     warehouse_queryset=None, get_queryset=False, no_geo=True,
+                     put_locks=False, lock_user=None):
     result_queryset = {}
     find_ids, cr_ids = None, None
     if operation_queryset:
@@ -134,6 +135,15 @@ def cr_serialization(archive=False, return_empty_types=False,
 
     result = generic_get_results(CR_MODEL_LIST, "context_records",
                                  result_queryset=result_queryset, no_geo=no_geo)
+    if put_locks:
+        for model in CR_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="context_records", archive=archive,
         return_empty_types=return_empty_types, archive_name=archive_name,