✨ create exhibition from basket

1 files changed, 14 insertions, 0 deletions

diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index 7fdc83197..35e61ddef 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -665,6 +665,12 @@ class ExhibitionFormMixin(IshtarMixin, LoginRequiredMixin):
 class ExhibitionCreateView(ExhibitionFormMixin, CreateView):
     page_name = _("Exhibition creation")
 
+    def get_form_kwargs(self):
+        kwargs = super().get_form_kwargs()
+        if "basket_id" in self.kwargs:
+            kwargs["initial"] = {"associated_basket_id": self.kwargs["basket_id"]}
+        return kwargs
+
     def get_context_data(self, **kwargs):
         data = super().get_context_data(**kwargs)
         # data["extra_form_modals"] = self.form_class.extra_form_modals
@@ -889,6 +895,14 @@ def findbasket_treatmentfile_add(request, pk, current_right=None):
     return treatmentfile_add(request, basket_pk=basket.pk)
 
 
+def findbasket_exhibition_add(request, pk, current_right=None):
+    try:
+        basket = models.FindBasket.objects.get(pk=pk)
+    except models.FindBasket.DoesNotExist:
+        raise Http404()
+    return ExhibitionCreateView.as_view()(request, basket_id=basket.pk)
+
+
 def container_treatment_add(request, pk, current_right=None):
     try:
         basket = models.FindBasket.objects.get(pk=pk)