diff options
Diffstat (limited to 'archaeological_operations/serializers.py')
| -rw-r--r-- | archaeological_operations/serializers.py | 13 |
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, |