diff options
Diffstat (limited to 'ishtar_common/views.py')
| -rw-r--r-- | ishtar_common/views.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/ishtar_common/views.py b/ishtar_common/views.py index 1363684a7..485facd75 100644 --- a/ishtar_common/views.py +++ b/ishtar_common/views.py @@ -2100,6 +2100,13 @@ class DocumentCreateView(DocumentFormMixin, CreateView): def get_form_kwargs(self): kwargs = super(DocumentCreateView, self).get_form_kwargs() initial = kwargs.get("initial", {}) + + try: + document = models.Document.objects.get(pk=self.kwargs.get("pk")) + assert check_permission(self.request, "document/create", document.pk) + except (AssertionError, models.Document.DoesNotExist): + raise Http404() + 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: @@ -2108,9 +2115,19 @@ class DocumentCreateView(DocumentFormMixin, CreateView): except model.DoesNotExist: continue initial[related_key] = str(item.pk) + + value = getattr(document, related_key) + if hasattr(value, "all"): + value = ",".join([str(v.pk) for v in value.all()]) + if hasattr(value, "pk"): + value = value.pk + initial[related_key] = value + initial["source"] = document.pk + if initial: kwargs["initial"] = initial kwargs["user"] = self.request.user + return kwargs |