diff options
Diffstat (limited to 'archaeological_warehouse/serializers.py')
| -rw-r--r-- | archaeological_warehouse/serializers.py | 134 |
1 files changed, 79 insertions, 55 deletions
diff --git a/archaeological_warehouse/serializers.py b/archaeological_warehouse/serializers.py index b0cf4d96f..19ae8ba99 100644 --- a/archaeological_warehouse/serializers.py +++ b/archaeological_warehouse/serializers.py @@ -1,13 +1,13 @@ from django.db.models import Q -from ishtar_common.serializers_utils import generic_get_results, \ - archive_serialization +from ishtar_common.serializers_utils import generic_get_results, archive_serialization from archaeological_warehouse import models WAREHOUSE_MODEL_LIST = [ - models.Warehouse, models.Container, + models.Warehouse, + models.Container, models.WarehouseDivisionLink, - models.ContainerLocalisation + models.ContainerLocalisation, ] # TODO: associated documents @@ -21,57 +21,76 @@ def generate_warehouse_queryset(base_query_key, ids): base_key = "{}__{}".format(find_key, base_query_key) q = Q(**{base_key: ids}) container_ids.update( - list(models.Container.objects.filter( - q).values_list("id", flat=True))) - q_loca = Q( - **{"container__{}__{}".format(find_key, base_query_key): ids}) + list(models.Container.objects.filter(q).values_list("id", flat=True)) + ) + q_loca = Q(**{"container__{}__{}".format(find_key, base_query_key): ids}) container_loca_ids.update( - list(models.ContainerLocalisation.objects.filter( - q_loca).values_list("id", flat=True))) + list( + models.ContainerLocalisation.objects.filter(q_loca).values_list( + "id", flat=True + ) + ) + ) for container_key in ("containers", "owned_containers"): - q = Q(** - {"{}__{}__{}".format(container_key, find_key, - base_query_key): ids}) - q_div = Q(** - {"warehouse__{}__{}__{}".format( - container_key, find_key, base_query_key): ids}) + q = Q(**{"{}__{}__{}".format(container_key, find_key, base_query_key): ids}) + q_div = Q( + **{ + "warehouse__{}__{}__{}".format( + container_key, find_key, base_query_key + ): ids + } + ) warehouse_ids.update( - list(models.Warehouse.objects.filter(q).values_list( - "id", flat=True))) + list(models.Warehouse.objects.filter(q).values_list("id", flat=True)) + ) warehouse_div_ids.update( - list(models.WarehouseDivisionLink.objects.filter( - q_div).values_list("id", flat=True))) + list( + models.WarehouseDivisionLink.objects.filter(q_div).values_list( + "id", flat=True + ) + ) + ) result_queryset = { - models.Warehouse.__name__: - models.Warehouse.objects.filter(id__in=warehouse_ids), - models.Container.__name__: - models.Container.objects.filter(id__in=container_ids), - models.WarehouseDivisionLink.__name__: - models.WarehouseDivisionLink.objects.filter( - id__in=warehouse_div_ids), - models.ContainerLocalisation.__name__: - models.ContainerLocalisation.objects.filter( - id__in=container_loca_ids) + models.Warehouse.__name__: models.Warehouse.objects.filter( + id__in=warehouse_ids + ), + models.Container.__name__: models.Container.objects.filter( + id__in=container_ids + ), + models.WarehouseDivisionLink.__name__: models.WarehouseDivisionLink.objects.filter( + id__in=warehouse_div_ids + ), + models.ContainerLocalisation.__name__: models.ContainerLocalisation.objects.filter( + id__in=container_loca_ids + ), } return result_queryset -def warehouse_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, - put_locks=False, lock_user=None): +def warehouse_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, + put_locks=False, + lock_user=None, +): result_queryset = {} if operation_queryset: operation_ids = operation_queryset.values_list("id", flat=True) base_query_key = "base_finds__context_record__operation_id__in" - result_queryset = generate_warehouse_queryset(base_query_key, - operation_ids) + result_queryset = generate_warehouse_queryset(base_query_key, operation_ids) elif site_queryset: site_ids = site_queryset.values_list("id", flat=True) - base_query_key = "base_finds__context_record__"\ - "operation__archaeological_sites__id__in" + base_query_key = ( + "base_finds__context_record__" "operation__archaeological_sites__id__in" + ) result_queryset = generate_warehouse_queryset(base_query_key, site_ids) elif cr_queryset: cr_ids = cr_queryset.values_list("id", flat=True) @@ -85,22 +104,24 @@ def warehouse_serialization(archive=False, return_empty_types=False, warehouse_ids = warehouse_queryset.values_list("id", flat=True) result_queryset = { models.Warehouse.__name__: warehouse_queryset, - models.Container.__name__: - models.Container.objects.filter( - Q(location__id__in=warehouse_ids) | - Q(responsible__id__in=warehouse_ids) - ), - models.WarehouseDivisionLink.__name__: - models.WarehouseDivisionLink.objects.filter( - warehouse_id__in=warehouse_ids), - models.ContainerLocalisation.__name__: - models.ContainerLocalisation.objects.filter( - division__warehouse_id__in=warehouse_ids) + models.Container.__name__: models.Container.objects.filter( + Q(location__id__in=warehouse_ids) | Q(responsible__id__in=warehouse_ids) + ), + models.WarehouseDivisionLink.__name__: models.WarehouseDivisionLink.objects.filter( + warehouse_id__in=warehouse_ids + ), + models.ContainerLocalisation.__name__: models.ContainerLocalisation.objects.filter( + division__warehouse_id__in=warehouse_ids + ), } if get_queryset: return result_queryset - result = generic_get_results(WAREHOUSE_MODEL_LIST, "warehouse", - result_queryset=result_queryset, no_geo=no_geo) + result = generic_get_results( + WAREHOUSE_MODEL_LIST, + "warehouse", + result_queryset=result_queryset, + no_geo=no_geo, + ) if put_locks: for model in WAREHOUSE_MODEL_LIST: if not hasattr(model, "locked"): @@ -111,7 +132,10 @@ def warehouse_serialization(archive=False, return_empty_types=False, q.update(locked=True, lock_user=lock_user) full_archive = archive_serialization( - result, archive_dir="warehouse", archive=archive, - return_empty_types=return_empty_types, archive_name=archive_name, + result, + archive_dir="warehouse", + archive=archive, + return_empty_types=return_empty_types, + archive_name=archive_name, ) return full_archive |