File module refactoring - more tests for files

1 files changed, 86 insertions, 36 deletions

diff --git a/archaeological_files/wizards.py b/archaeological_files/wizards.py
index 1538a984a..569b01a56 100644
--- a/archaeological_files/wizards.py
+++ b/archaeological_files/wizards.py
@@ -39,10 +39,23 @@ class FileSearch(SearchWizard):
 class FileWizard(OperationWizard):
     model = models.File
     object_parcel_type = "associated_file"
-    parcel_step_key = "parcels-"
-    town_step_keys = ["towns-"]
+    parcel_step_key = "parcelspdl-"
+    town_step_keys = ["preventiveplanning-", "researchaddress-"]
     wizard_done_window = reverse_lazy("show-file")
     redirect_url = "file_modification"
+    town_input_id = "town"
+    towns_formset = False
+    multi_towns = True
+    wizard_templates = {
+        "planningservice-%(url_name)s": "ishtar/wizard/wizard_planningservice.html",
+        "instruction-%(url_name)s": "ishtar/wizard/wizard_instruction.html",
+        "preventiveplanning-%(url_name)s": "ishtar/wizard/wizard_preventiveplanning.html",
+    }
+    wizard_confirm = "ishtar/wizard/file_confirm_wizard.html"
+
+    def get_current_year(self):
+        general_form_key = "general-" + self.url_name
+        return self.session_get_value(general_form_key, "year")
 
     def get_extra_model(self, dct, m2m, form_list):
         dct = super(FileWizard, self).get_extra_model(dct, m2m, form_list)
@@ -53,9 +66,79 @@ class FileWizard(OperationWizard):
             dct["numeric_reference"] = current_ref and current_ref + 1 or 1
         return dct
 
+    def get_form_kwargs(self, *args, **kwargs):
+        returned = super(FileWizard, self).get_form_kwargs(*args, **kwargs)
+        if args and args[0].startswith("generalcontractor-"):
+            if "status" in self.request.GET:
+                returned["status"] = self.request.GET["status"]
+        if args and args[0].startswith("instruction-"):
+            returned["year"] = self.get_current_year()
+            returned["saisine_type"] = self.get_saisine_type()
+            returned["reception_date"] = self.session_get_value(
+                "general-" + self.url_name, "reception_date"
+            )
+        return returned
+
+    def get_saisine_type(self):
+        try:
+            idx = int(
+                self.session_get_value(
+                    "preventivetype-" + self.url_name, "saisine_type"
+                )
+            )
+            return models.SaisineType.objects.get(pk=idx)
+        except (TypeError, ValueError, models.PermitType.DoesNotExist):
+            pass
+
+    def get_context_data(self, form, **kwargs):
+        context = super(FileWizard, self).get_context_data(form)
+        formplanning = "planningservice-" + self.url_name
+        forminstruction = "instruction-" + self.url_name
+        formfinal = "final-" + self.url_name
+        if self.steps.current == formplanning:
+            try:
+                idx = int(
+                    self.session_get_value(
+                        "preventivetype-" + self.url_name, "permit_type"
+                    )
+                )
+                permit_type = models.PermitType.objects.get(pk=idx)
+                context["permit_type"] = str(permit_type)
+                context["permit_type_code"] = str(permit_type.txt_idx)
+            except (TypeError, ValueError, models.PermitType.DoesNotExist):
+                pass
+        elif self.steps.current == forminstruction:
+            saisine_type = self.get_saisine_type()
+            context["FILE_PREFIX"] = settings.ISHTAR_FILE_PREFIX
+            if saisine_type:
+                context["saisine_type"] = str(saisine_type)
+                context["saisine_type_message"] = str(saisine_type)
+                if saisine_type.delay:
+                    context["saisine_type_message"] += str(
+                        _(": delay of {} days")
+                    ).format(saisine_type.delay)
+        elif self.steps.current == formfinal:
+            if not self.steps.current.endswith("creation"):  # modification only
+                try:
+                    numeric_reference = int(
+                        self.session_get_value(
+                            "instruction-" + self.url_name, "numeric_reference"
+                        )
+                    )
+
+                    q = models.File.objects.filter(
+                        numeric_reference=numeric_reference,
+                        year=self.get_current_year(),
+                    ).exclude(pk=self.get_current_object().pk)
+                    context["numeric_reference_files"] = q.all()
+                except (ValueError, TypeError):
+                    pass
+
+        return context
+
     def done(self, form_list, **kwargs):
         """
-        Save parcels and make numeric_reference unique
+        Make numeric_reference unique
         """
         r = super(FileWizard, self).done(form_list, return_object=True, **kwargs)
         if type(r) not in (list, tuple) or len(r) != 2:
@@ -77,39 +160,6 @@ class FileWizard(OperationWizard):
             if changed:
                 obj.numeric_reference = numeric_reference
                 obj.save()
-        obj.parcels.clear()
-        for form in form_list:
-            if (
-                not hasattr(form, "prefix")
-                or not form.prefix.startswith(self.parcel_step_key)
-                or not hasattr(form, "forms")
-            ):
-                continue
-            for frm in form.forms:
-                if not frm.is_valid():
-                    continue
-                dct = frm.cleaned_data.copy()
-                if "parcel" in dct:
-                    try:
-                        parcel = Parcel.objects.get(pk=dct["parcel"])
-                        setattr(parcel, self.object_parcel_type, obj)
-                        parcel.save()
-                    except (ValueError, ObjectDoesNotExist):
-                        continue
-                    continue
-                try:
-                    dct["town"] = models.Town.objects.get(pk=int(dct["town"]))
-                except (ValueError, ObjectDoesNotExist, KeyError):
-                    continue
-                dct["associated_file"], dct["operation"] = None, None
-                dct[self.object_parcel_type] = obj
-                if "DELETE" in dct:
-                    dct.pop("DELETE")
-                parcel = Parcel.objects.filter(**dct).count()
-                if not parcel:
-                    dct["history_modifier"] = self.request.user
-                    parcel = Parcel(**dct)
-                    parcel.save()
         return res