Muliple person deletion wizard

1 files changed, 106 insertions, 3 deletions

diff --git a/ishtar_common/wizards.py b/ishtar_common/wizards.py
index 3cdcbc63d..65a37a860 100644
--- a/ishtar_common/wizards.py
+++ b/ishtar_common/wizards.py
@@ -38,7 +38,7 @@ from django.db.models.fields import NOT_PROVIDED
 
 from django.http import HttpResponseRedirect, Http404
 from django.forms import ValidationError
-from django.shortcuts import redirect, render
+from django.shortcuts import redirect, render, reverse
 from django.template import loader
 from django.utils.translation import ugettext_lazy as _
 from django.utils.safestring import mark_safe
@@ -112,6 +112,7 @@ class Wizard(IshtarWizard):
     storage_name = 'formtools.wizard.storage.session.SessionStorage'
     wizard_done_template = 'ishtar/wizard/wizard_done.html'
     wizard_done_window = ''
+    redirect_url = None
     wizard_confirm = 'ishtar/wizard/confirm_wizard.html'
     wizard_templates = {}
     filter_owns = {}
@@ -1508,7 +1509,7 @@ class DocumentSearch(SearchWizard):
 class DeletionWizard(Wizard):
     def __init__(self, *args, **kwargs):
         if (not hasattr(self, 'fields') or not self.fields) and \
-           (hasattr(self, 'model') and hasattr(self.model, 'TABLE_COLS')):
+                (hasattr(self, 'model') and hasattr(self.model, 'TABLE_COLS')):
             self.fields = self.model.TABLE_COLS
         assert self.model
         return super(DeletionWizard, self).__init__(*args, **kwargs)
@@ -1563,6 +1564,107 @@ class DeletionWizard(Wizard):
             self.request, 'ishtar/wizard/wizard_delete_done.html', {})
 
 
+class MultipleItemWizard(Wizard):
+    main_item_select_keys = ('selec-',)
+    current_object_key = "pks"
+
+    def get_current_objects(self):
+        current_objs = []
+        for key in self.main_item_select_keys:
+            main_form_key = key + self.url_name
+            try:
+                pks = self.session_get_value(main_form_key,
+                                             self.current_object_key)
+                if pks:
+                    for pk in pks.split(","):
+                        current_objs.append(self.model.objects.get(pk=int(pk)))
+            except(TypeError, ValueError, ObjectDoesNotExist):
+                pass
+        return current_objs
+
+
+class MultipleDeletionWizard(MultipleItemWizard):
+    def __init__(self, *args, **kwargs):
+        if (not hasattr(self, 'fields') or not self.fields) and \
+           (hasattr(self, 'model') and hasattr(self.model, 'TABLE_COLS')):
+            self.fields = self.model.TABLE_COLS
+        assert self.model
+        return super(MultipleDeletionWizard, self).__init__(*args, **kwargs)
+
+    def get_formated_datas(self, forms):
+        datas = super(MultipleDeletionWizard, self).get_formated_datas(forms)
+        self.current_objs = []
+        for form in forms:
+            if not hasattr(form, "cleaned_data"):
+                continue
+            for key in form.cleaned_data:
+                if key == 'pks':
+                    model = form.associated_models['pks']
+                    pks = form.cleaned_data["pks"].split(",")
+                    for pk in pks:
+                        try:
+                            self.current_objs.append(model.objects.get(pk=pk))
+                        except model.DoesNotExist:
+                            continue
+        if not self.current_objs:
+            return datas
+        full_res = []
+        for current_obj in self.current_objs:
+            res = {}
+            for field in self.model._meta.get_fields():
+                if field.name not in self.fields:
+                    continue
+                value = getattr(current_obj, field.name)
+                if not value:
+                    continue
+                label = ""
+                if hasattr(field, 'verbose_name'):
+                    label = field.verbose_name
+                if hasattr(value, 'all'):
+                    if not label and hasattr(field, 'related_model'):
+                        label = field.related_model._meta.verbose_name_plural
+                    value = ", ".join([str(item) for item in value.all()])
+                    if not value:
+                        continue
+                else:
+                    value = str(value)
+                res[field.name] = (label, value, '')
+            full_res.append(res)
+        if not datas and self.fields:
+            datas = [['', []]]
+        datas = []
+        for idx, res in enumerate(full_res):
+            data = []
+            for field in self.fields:
+                if field in res:
+                    data.append(res[field])
+            datas.append((str(self.current_objs[idx]), data))
+        return datas
+
+    def get_context_data(self, form, **kwargs):
+        data = super(MultipleDeletionWizard, self).get_context_data(form,
+                                                                    **kwargs)
+        data["current_objs"] = self.get_current_objects()
+        return data
+
+    def done(self, form_list, **kwargs):
+        objs = self.get_current_objects()
+        lbls = []
+        for obj in objs:
+            lbls.append(str(obj))
+            try:
+                obj.delete()
+            except ObjectDoesNotExist:
+                pass
+        msg = '<i class="fa fa-trash" aria-hidden="true"></i>&nbsp;&nbsp;'
+        msg += str(_("{} deleted.")).format(" ; ".join(lbls))
+        messages.add_message(self.request, messages.INFO, msg)
+        if self.redirect_url:
+            return HttpResponseRedirect(reverse(self.redirect_url))
+        return render(
+            self.request, 'ishtar/wizard/wizard_delete_done.html', {})
+
+
 class ClosingWizard(Wizard):
     # "close" an item
     # to be define in the overloaded class
@@ -1628,9 +1730,10 @@ class PersonModifWizard(PersonWizard):
         'identity-person_modification': "ishtar/wizard/wizard_person.html"}
 
 
-class PersonDeletionWizard(DeletionWizard):
+class PersonDeletionWizard(MultipleDeletionWizard):
     model = models.Person
     fields = model.TABLE_COLS
+    redirect_url = "person_search"
     wizard_templates = {
         'final-person_deletion': 'ishtar/wizard/wizard_person_deletion.html'}