Manage link between import and imported object - Auto-delete imported object when deleting import

1 files changed, 11 insertions, 2 deletions

diff --git a/ishtar_common/views.py b/ishtar_common/views.py
index b005659db..1b27bcf43 100644
--- a/ishtar_common/views.py
+++ b/ishtar_common/views.py
@@ -45,7 +45,7 @@ from django.template import RequestContext, loader
 from django.utils.decorators import method_decorator
 from django.utils.translation import ugettext, ugettext_lazy as _
 from django.views.generic import ListView
-from django.views.generic.edit import CreateView
+from django.views.generic.edit import CreateView, DeleteView
 
 from xhtml2odt import xhtml2odt
 
@@ -905,7 +905,8 @@ class ImportListView(IshtarMixin, LoginRequiredMixin, ListView):
                 continue
             action = request.POST[field]
             if action == 'D':
-                imprt.delete()
+                return HttpResponseRedirect(reverse('import_delete',
+                                                    kwargs={'pk':imprt.pk}))
             elif action == 'A':
                 imprt.initialize()
             elif action == 'I':
@@ -934,3 +935,11 @@ class ImportLinkView(IshtarMixin, LoginRequiredMixin, ModelFormSetView):
 
     def get_success_url(self):
         return reverse('current_imports')
+
+class ImportDeleteView(IshtarMixin, LoginRequiredMixin, DeleteView):
+    template_name = 'ishtar/import_delete.html'
+    model = models.Import
+    page_name = _(u"Delete import")
+
+    def get_success_url(self):
+        return reverse('current_imports')