diff options
Diffstat (limited to 'archaeological_context_records/serializers.py')
| -rw-r--r-- | archaeological_context_records/serializers.py | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/archaeological_context_records/serializers.py b/archaeological_context_records/serializers.py index 70e93f10f..7bb9b6724 100644 --- a/archaeological_context_records/serializers.py +++ b/archaeological_context_records/serializers.py @@ -10,8 +10,8 @@ CR_MODEL_LIST = [ def cr_serialization(archive=False, return_empty_types=False, - archive_name=None, operation_queryset=None, - site_queryset=None, cr_queryset=None): + archive_name=None, operation_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) @@ -47,6 +47,18 @@ def cr_serialization(archive=False, return_empty_types=False, right_record_id__in=cr_ids, ) } + elif find_queryset: + find_ids = find_queryset.values_list("id", flat=True) + result_queryset = { + models.ContextRecord.__name__: models.ContextRecord.objects.filter( + base_finds__find__id__in=find_ids, + ), + models.RecordRelations.__name__: + models.RecordRelations.objects.filter( + left_record__base_finds__find__id__in=find_ids, + right_record__base_finds__find__id__in=find_ids, + ) + } result = generic_get_results(CR_MODEL_LIST, "context_records", result_queryset=result_queryset) |