Operation wizard refactoring

1 files changed, 2 insertions, 101 deletions

diff --git a/archaeological_operations/wizards.py b/archaeological_operations/wizards.py
index 7afe73442..44b3988eb 100644
--- a/archaeological_operations/wizards.py
+++ b/archaeological_operations/wizards.py
@@ -48,8 +48,6 @@ class OperationSearch(SearchWizard):
 
 class OperationWizard(Wizard):
     model = models.Operation
-    object_parcel_type = "operation"
-    parcel_step_key = "parcels"
     relations_step_key = "relations"
     # step including the current(s) town(s)
     town_step_keys = ["towns-", "townsgeneral-"]
@@ -62,9 +60,7 @@ class OperationWizard(Wizard):
     def get_template_names(self):
         templates = super(OperationWizard, self).get_template_names()
         current_step = self.steps.current
-        if current_step.startswith(self.parcel_step_key):
-            templates = ["ishtar/wizard/parcels_wizard.html"] + templates
-        elif current_step.startswith(self.relations_step_key):
+        if current_step.startswith(self.relations_step_key):
             templates = ["ishtar/wizard/relations_wizard.html"] + templates
         return templates
 
@@ -94,13 +90,8 @@ class OperationWizard(Wizard):
         """
         context = super(OperationWizard, self).get_context_data(form, **kwargs)
         step = self.steps.current
-        if step.startswith("towns"):
+        if step.startswith("general"):
             context["TOWNS"] = self.get_towns()
-        elif step.startswith("parcels-") and self.get_current_file():
-            # if a file is associated to the operation add the button "Add all"
-            context["add_all"] = True
-        if step.startswith("parcels") and hasattr(self, "automatic_parcel_association"):
-            context["automatic_parcel_association"] = self.automatic_parcel_association
         # reminder of the current file
         reminder = self.get_reminder()
         if reminder:
@@ -121,31 +112,6 @@ class OperationWizard(Wizard):
             pass
         return sorted(towns, key=lambda x: x[1])
 
-    def get_available_parcels(self, file):
-        self.automatic_parcel_association = False
-        parcels = []
-        current_parcels = []
-        operation = self.get_current_object()
-        if operation:
-            for parcel in operation.parcels.all():
-                current_parcels.append(
-                    (parcel.town, parcel.section, parcel.parcel_number)
-                )
-                parcels.append((parcel.pk, parcel.short_label))
-        try:
-            for parcel in file.parcels.all():
-                key = (parcel.town, parcel.section, parcel.parcel_number)
-                if key in current_parcels:
-                    current_parcels.pop(current_parcels.index(key))
-                    continue
-                parcels.append((parcel.pk, parcel.short_label))
-            if current_parcels:
-                # not all operation parcel exist for the file
-                self.automatic_parcel_association = True
-        except (ValueError, ObjectDoesNotExist):
-            pass
-        return sorted(parcels, key=lambda x: x[1])
-
     def get_form(self, step=None, data=None, files=None):
         """
         Manage specifics fields
@@ -163,43 +129,6 @@ class OperationWizard(Wizard):
         # manage the dynamic choice of towns
         if step.startswith("towns") and hasattr(form, "management_form"):
             data["TOWNS"] = self.get_towns()
-        elif step.startswith(self.parcel_step_key) and hasattr(form, "management_form"):
-            file = self.get_current_file()
-            if file:
-                data["PARCELS"] = self.get_available_parcels(file)
-            else:
-                town_ids = []
-                for town_step_key in self.town_step_keys:
-                    town_form_key = town_step_key + self.url_name
-                    town_ids = (
-                        self.session_get_value(
-                            town_form_key,
-                            self.town_input_id,
-                            multi=self.towns_formset,
-                            multi_value=self.multi_towns,
-                        )
-                        or []
-                    )
-                    if town_ids:
-                        towns = []
-                        if type(town_ids) == str:
-                            town_ids = [town_ids]
-                        for ids in town_ids:
-                            for d in ids.split(","):
-                                if d:
-                                    towns.append(d)
-                        town_ids = towns
-                        break
-                if type(town_ids) not in (list, tuple):
-                    town_ids = [town_ids]
-                towns = []
-                for town_id in town_ids:
-                    try:
-                        town = models.Town.objects.get(pk=int(town_id))
-                        towns.append((town.pk, str(town)))
-                    except (ValueError, ObjectDoesNotExist):
-                        pass
-                data["TOWNS"] = sorted(towns, key=lambda x: x[1])
         data = data or None
         form = super(OperationWizard, self).get_form(step, data, files)
         return form
@@ -267,34 +196,6 @@ class OperationWizard(Wizard):
         initial.update(self.__copy_fields(file, keys))
         return initial
 
-    def post(self, *args, **kwargs):
-        request = self.request
-        post_data = request.POST.copy()
-
-        # add all parcel from available in the archaeological file
-        if not post_data.get("add_all_parcels", None):
-            return super(OperationWizard, self).post(*args, **kwargs)
-
-        file = self.get_current_file()
-        if not file:
-            return super(OperationWizard, self).post(*args, **kwargs)
-        parcel_form_key = self.steps.current
-        # parcel_form_key = "parcels-" + self.url_name
-        idx = -1
-        # remove non relevant deleted keys
-        for k in post_data.keys():
-            if k.startswith(parcel_form_key) and k.endswith("-DELETE"):
-                post_data.pop(k)
-        for idx, parcel in enumerate(self.get_available_parcels(file)):
-            parcel_pk, parcel_name = parcel
-            post_data["%s-%d-parcel" % (parcel_form_key, idx)] = parcel_pk
-        post_data[parcel_form_key + "-TOTAL_FORMS"] = idx + 2
-        request.POST = post_data
-        return super(OperationWizard, self).post(*args, **kwargs)
-
-    def post_save(self):
-        self.current_object.clean_parcel_duplicates()
-
 
 class OperationModificationWizard(OperationWizard):
     modification = True