WIP - correction suppression importdevelop-5.0

1 files changed, 13 insertions, 2 deletions

diff --git a/ishtar_common/views.py b/ishtar_common/views.py
index e346e625d..dbaf8626b 100644
--- a/ishtar_common/views.py
+++ b/ishtar_common/views.py
@@ -2639,18 +2639,29 @@ class ImportMatchView(ImportPermissionMixin, IshtarMixin, LoginRequiredMixin, Mo
         return reverse("import_link_unmatched", args=[self.kwargs["pk"]])
 
 
-class ImportDeleteView(ImportPermissionMixin, IshtarMixin, LoginRequiredMixin, DeleteView):
+class ImportDeleteView(ImportPermissionMixin, IshtarMixin, LoginRequiredMixin,
+                       DeleteView):
     template_name = "ishtar/import_delete.html"
     model = models.Import
     page_name = _("Delete import")
     permission_full = "ishtar_common.delete_import"
     permission_own = "delete_own_import"
 
+    def form_valid(self, form):
+        if not self.object.delete():
+            messages.add_message(
+                self.request, messages.WARNING,
+                _("Integrity error - contact the administrator"))
+            return HttpResponseRedirect(self.request.path)
+        success_url = self.get_success_url()
+        return HttpResponseRedirect(success_url)
+
     def get_success_url(self):
         return reverse("current_imports")
 
 
-class ImportGroupDeleteView(ImportPermissionMixin, IshtarMixin, LoginRequiredMixin, DeleteView):
+class ImportGroupDeleteView(ImportPermissionMixin, IshtarMixin, LoginRequiredMixin,
+                            DeleteView):
     template_name = "ishtar/import_delete.html"
     model = models.ImportGroup
     page_name = _("Delete import")