diff options
Diffstat (limited to 'archaeological_finds/views.py')
| -rw-r--r-- | archaeological_finds/views.py | 14 |
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) |