Sheet find: link to baskets

1 files changed, 12 insertions, 1 deletions

diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index 620f33ef4..a04d7fe2f 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -112,7 +112,18 @@ def autocomplete_treatmentfile(request):
     return HttpResponse(data, content_type='text/plain')
 
 
-show_find = show_item(models.Find, 'find')
+def show_find_extra(request, find):
+    if not request.user or not request.user.ishtaruser:
+        return {}
+    user = request.user.ishtaruser
+    q = models.FindBasket.objects.filter(items__pk=find.pk).filter(
+        Q(user=user) | Q(shared_with__pk=user.pk) |
+        Q(shared_write_with__pk=user.pk)
+    )
+    return {"baskets": [(basket.pk, basket.full_label) for basket in q.all()]}
+
+
+show_find = show_item(models.Find, 'find', extra_dct=show_find_extra)
 display_find = display_item(models.Find)
 revert_find = revert_item(models.Find)