From b103ce9f92012db9d0bc9164b76705633c3c453a Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Wed, 11 Sep 2019 14:34:25 +0200 Subject: Serialization - Import/Export: manage put an release of locks --- archaeological_context_records/serializers.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) (limited to 'archaeological_context_records') 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, -- cgit v1.2.3