Serializers: operations/context records - filter vy context records

1 files changed, 15 insertions, 1 deletions

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)