From f2cb0542faac028dc3f90a61b46c0949bb35339f Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Wed, 4 Sep 2019 12:03:28 +0200 Subject: Serializer: manage queryset filter --- archaeological_operations/serializers.py | 27 ++++++++++++++++++++++++--- 1 file changed, 24 insertions(+), 3 deletions(-) (limited to 'archaeological_operations/serializers.py') diff --git a/archaeological_operations/serializers.py b/archaeological_operations/serializers.py index 7d4d9033d..22f0aedc7 100644 --- a/archaeological_operations/serializers.py +++ b/archaeological_operations/serializers.py @@ -9,9 +9,30 @@ OPERATION_MODEL_LIST = [ def operation_serialization(archive=False, return_empty_types=False, - archive_name=None): - result = generic_get_results(OPERATION_MODEL_LIST, "operations") + archive_name=None, operation_queryset=None, + site_queryset=None): + result_queryset = {} + if operation_queryset: + result_queryset = { + models.Operation.__name__: operation_queryset, + models.ArchaeologicalSite.__name__: + models.ArchaeologicalSite.objects.filter( + operations__id__in=operation_queryset.values_list( + "id", flat=True) + ) + } + elif site_queryset: + result_queryset = { + models.ArchaeologicalSite.__name__: site_queryset, + models.Operation.__name__: models.Operation.objects.filter( + archaeological_sites__id__in=site_queryset.values_list( + "id", flat=True)) + } + + result = generic_get_results(OPERATION_MODEL_LIST, "operations", + result_queryset=result_queryset) full_archive = archive_serialization( result, archive_dir="operations", archive=archive, - return_empty_types=return_empty_types, archive_name=archive_name) + return_empty_types=return_empty_types, archive_name=archive_name, + ) return full_archive -- cgit v1.2.3