From 35fb017eb816402140317dd9ba78c91f7b68d121 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Wed, 4 Sep 2019 16:10:56 +0200 Subject: Serializers: operations/context records - filter vy context records --- archaeological_operations/serializers.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) (limited to 'archaeological_operations/serializers.py') diff --git a/archaeological_operations/serializers.py b/archaeological_operations/serializers.py index 3500cdad2..a480f307a 100644 --- a/archaeological_operations/serializers.py +++ b/archaeological_operations/serializers.py @@ -12,7 +12,7 @@ OPERATION_MODEL_LIST = [ def operation_serialization(archive=False, return_empty_types=False, archive_name=None, operation_queryset=None, - site_queryset=None): + site_queryset=None, cr_queryset=None): result_queryset = {} if operation_queryset: operation_ids = operation_queryset.values_list("id", flat=True) @@ -40,6 +40,20 @@ def operation_serialization(archive=False, return_empty_types=False, right_record__archaeological_sites__id__in=site_ids, ), } + elif cr_queryset: + cr_ids = cr_queryset.values_list("id", flat=True) + result_queryset = { + models.ArchaeologicalSite.__name__: + models.ArchaeologicalSite.objects.filter( + operations__context_record__id__in=cr_ids), + models.Operation.__name__: models.Operation.objects.filter( + context_record__id__in=cr_ids), + models.RecordRelations.__name__: + models.RecordRelations.objects.filter( + left_record__context_record__id__in=cr_ids, + right_record__context_record__id__in=cr_ids, + ), + } result = generic_get_results(OPERATION_MODEL_LIST, "operations", result_queryset=result_queryset) -- cgit v1.2.3