✨ create exhibition from basket

3 files changed, 39 insertions, 0 deletions

diff --git a/archaeological_finds/models_finds.py b/archaeological_finds/models_finds.py
index ec54570db..e1c4e2f71 100644
--- a/archaeological_finds/models_finds.py
+++ b/archaeological_finds/models_finds.py
@@ -1036,6 +1036,17 @@ class FindBasket(Basket, MainItem, ValueGetter):
                     False,
                 ),
             ]
+        if self.can_do(request, "archaeological_finds.add_exhibition"):
+            actions += [
+                (
+                    reverse("findbasket-add-exhibition", args=[self.pk]),
+                    _("Create exhibition"),
+                    "fa fa-calendar",
+                    "",
+                    "",
+                    False,
+                ),
+            ]
 
         if can_edit_find:
             duplicate = self.get_quick_action_by_url("findbasket-qa-duplicate")
diff --git a/archaeological_finds/urls.py b/archaeological_finds/urls.py
index b99140e00..e8935511f 100644
--- a/archaeological_finds/urls.py
+++ b/archaeological_finds/urls.py
@@ -183,6 +183,13 @@ urlpatterns = [
         name="findbasket-add-treatmentfile",
     ),
     url(
+        r"^findbasket-add-exhibition/(?P<pk>[0-9-]+)/$",
+        check_permissions(
+            ["archaeological_finds.add_exhibition"]
+        )(views.findbasket_exhibition_add),
+        name="findbasket-add-exhibition",
+    ),
+    url(
         r"^find-add-treatment/(?P<pk>[0-9-]+)/$",
         check_permissions(
             ["archaeological_finds.add_treatment"]
@@ -444,6 +451,13 @@ urlpatterns = [
         name="exhibition-search",
     ),
     path(
+        "exhibition/create/<int:basket_id>/",
+        check_permissions(
+            ["archaeological_finds.add_exhibition"]
+        )(views.ExhibitionCreateView.as_view()),
+        name="exhibition-create-from-basket",
+    ),
+    path(
         "exhibition/create/",
         check_permissions(
             ["archaeological_finds.add_exhibition"]
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)