api for geo_get_items for context records + tests

1 files changed, 24 insertions, 6 deletions

diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index f4c66b3bf..bba4efb13 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -1338,10 +1338,28 @@ class PublicFindAPI(APIView):
         return Response(serializer.data)
 
 
-def get_geo_items(request, pk, get_polygons, current_right=None):
-    try:
-        base_find = models.BaseFind.objects.get(pk=pk)
-    except models.BaseFind.DoesNotExist:
-        raise Http404()
-    dic = base_find.get_geo_items(get_polygons=get_polygons)
+def get_geo_items(
+    request,
+    get_polygons,
+    pk=None,
+    context_record_pk=None,
+    operation_pk=None,
+    current_right=None,
+):
+    if operation_pk:
+        try:
+            bf = Operation.objects.get(pk=operation_pk)
+        except Operation.DoesNotExist:
+            raise Http404()
+    elif context_record_pk:
+        try:
+            bf = ContextRecord.objects.get(pk=context_record_pk)
+        except ContextRecord.DoesNotExist:
+            raise Http404()
+    else:
+        try:
+            bf = models.BaseFind.objects.get(pk=pk)
+        except models.BaseFind.DoesNotExist:
+            raise Http404()
+    dic = bf.get_geo_items(get_polygons=get_polygons)
     return HttpResponse(json.dumps(dic).encode("utf-8"))