Serializers: filter by finds

1 files changed, 9 insertions, 1 deletions

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(