Quick action to add a document from a sheet (refs #4107)

1 files changed, 15 insertions, 0 deletions

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):