From d99e961414c0c15812bbc583253008550079ccc8 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Thu, 5 Sep 2019 17:14:06 +0200 Subject: Serializers: filter by finds --- archaeological_finds/serializers.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) (limited to 'archaeological_finds/serializers.py') diff --git a/archaeological_finds/serializers.py b/archaeological_finds/serializers.py index 5fc8d2224..56d113636 100644 --- a/archaeological_finds/serializers.py +++ b/archaeological_finds/serializers.py @@ -11,7 +11,8 @@ FIND_MODEL_LIST = [ def find_serialization(archive=False, return_empty_types=False, archive_name=None, operation_queryset=None, - site_queryset=None, cr_queryset=None): + site_queryset=None, cr_queryset=None, + find_queryset=None): result_queryset = {} if operation_queryset: operation_ids = operation_queryset.values_list("id", flat=True) @@ -46,6 +47,13 @@ def find_serialization(archive=False, return_empty_types=False, models.Find.objects.filter( base_finds__context_record__in=cr_ids), } + elif find_queryset: + find_ids = find_queryset.values_list("id", flat=True) + result_queryset = { + models.BaseFind.__name__: models.BaseFind.objects.filter( + find__id__in=find_ids), + models.Find.__name__: find_queryset, + } result = generic_get_results(FIND_MODEL_LIST, "finds", result_queryset=result_queryset) full_archive = archive_serialization( -- cgit v1.2.3