Imports - Current - Edit : adding edit action on C or A step with display malfunction #5287

Signed-off-by: Étienne Loks <etienne.loks@iggdrasil.net>

1 files changed, 26 insertions, 0 deletions

diff --git a/ishtar_common/views.py b/ishtar_common/views.py
index 9fe3ce15d..ce27ce0e7 100644
--- a/ishtar_common/views.py
+++ b/ishtar_common/views.py
@@ -1394,6 +1394,28 @@ class NewImportView(IshtarMixin, LoginRequiredMixin, CreateView):
         return HttpResponseRedirect(self.get_success_url())
 
 
+class EditImportView(IshtarMixin, LoginRequiredMixin, UpdateView):
+    template_name = "ishtar/form.html"
+    model = models.Import
+    form_class = forms.NewImportForm
+    page_name = _("Edit import")
+
+    def get_success_url(self):
+        return reverse("current_imports")
+
+    def get_form_kwargs(self):
+        kwargs = super(EditImportView, self).get_form_kwargs()
+        kwargs["user"] = self.request.user
+
+        return kwargs
+
+    def form_valid(self, form):
+        user = models.IshtarUser.objects.get(pk=self.request.user.pk)
+        self.object = form.save(user=user)
+
+        return HttpResponseRedirect(self.get_success_url())
+
+
 class ImportListView(IshtarMixin, LoginRequiredMixin, ListView):
     template_name = "ishtar/import_list.html"
     model = models.Import
@@ -1426,6 +1448,10 @@ class ImportListView(IshtarMixin, LoginRequiredMixin, ListView):
                 return HttpResponseRedirect(
                     reverse("import_delete", kwargs={"pk": imprt.pk})
                 )
+            elif action == "ED":
+                return HttpResponseRedirect(
+                    reverse("edit_import", kwargs={"pk": imprt.pk})
+                )
             elif action == "A":
                 imprt.initialize(
                     user=self.request.user.ishtaruser,