From 3728ddd681c428abf2a686b8b31f6cc95c5972cc Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Sun, 10 Jun 2018 12:29:37 +0200 Subject: Quick action to add a document from a sheet (refs #4107) --- ishtar_common/views.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'ishtar_common/views.py') diff --git a/ishtar_common/views.py b/ishtar_common/views.py index 74aa589c1..2baa5994a 100644 --- a/ishtar_common/views.py +++ b/ishtar_common/views.py @@ -1596,6 +1596,21 @@ class DocumentFormMixin(IshtarMixin, LoginRequiredMixin): class DocumentCreateView(DocumentFormMixin, CreateView): page_name = _(u"Document creation") + def get_form_kwargs(self): + kwargs = super(DocumentCreateView, self).get_form_kwargs() + initial = kwargs.get('initial', {}) + for related_key in models.Document.RELATED_MODELS_ALT: + model = models.Document._meta.get_field(related_key).related_model + if model.SLUG in self.request.GET: + try: + item = model.objects.get(pk=self.request.GET[model.SLUG]) + except model.DoesNotExist: + continue + initial[related_key] = str(item.pk) + if initial: + kwargs['initial'] = initial + return kwargs + class DocumentSelectView(IshtarMixin, LoginRequiredMixin, FormView): -- cgit v1.2.3