Serializers: filter by finds

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)