diff options
Diffstat (limited to 'archaeological_files_pdl/wizards.py')
| -rw-r--r-- | archaeological_files_pdl/wizards.py | 107 |
1 files changed, 58 insertions, 49 deletions
diff --git a/archaeological_files_pdl/wizards.py b/archaeological_files_pdl/wizards.py index 988303937..211ea84f8 100644 --- a/archaeological_files_pdl/wizards.py +++ b/archaeological_files_pdl/wizards.py @@ -26,45 +26,44 @@ from ishtar_common.utils import ugettext_lazy as _ class FileWizard(BaseFileWizard): - parcel_step_key = 'parcelspdl-' - town_step_keys = ['preventiveplanning-', 'researchaddress-'] - town_input_id = 'town' + parcel_step_key = "parcelspdl-" + town_step_keys = ["preventiveplanning-", "researchaddress-"] + town_input_id = "town" towns_formset = False multi_towns = True wizard_templates = { #'generalcontractor-%(url_name)s': # 'ishtar/wizard/wizard_generalcontractor.html', - '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', + "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' + 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') + general_form_key = "general-" + self.url_name + return self.session_get_value(general_form_key, "year") 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') + 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')) + "preventivetype-" + self.url_name, "saisine_type" + ) + ) return models.SaisineType.objects.get(pk=idx) except (TypeError, ValueError, models.PermitType.DoesNotExist): pass @@ -78,53 +77,63 @@ class FileWizard(BaseFileWizard): try: idx = int( self.session_get_value( - 'preventivetype-' + self.url_name, 'permit_type')) + "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) + 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 + context["FILE_PREFIX"] = settings.ISHTAR_FILE_PREFIX if saisine_type: - context['saisine_type'] = str(saisine_type) - context['saisine_type_message'] = str(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) + context["saisine_type_message"] += str( + _(": delay of {} days") + ).format(saisine_type.delay) elif self.steps.current == formfinal: - if self.steps.current.endswith('creation'): # creation only + if self.steps.current.endswith("creation"): # creation only parcels = [] parcel_step_key = self.parcel_step_key + self.url_name - parcel_numbers = self.session_get_value( - parcel_step_key, 'parcel_number', multi=True) or [] - sections = self.session_get_value( - parcel_step_key, 'section', multi=True) or [] - towns = self.session_get_value( - parcel_step_key, 'town', multi=True) or [] + parcel_numbers = ( + self.session_get_value(parcel_step_key, "parcel_number", multi=True) + or [] + ) + sections = ( + self.session_get_value(parcel_step_key, "section", multi=True) or [] + ) + towns = ( + self.session_get_value(parcel_step_key, "town", multi=True) or [] + ) for idx, parcel_number in enumerate(parcel_numbers): - if not parcel_number or len(sections) <= idx \ - or len(towns) <= idx: + if not parcel_number or len(sections) <= idx or len(towns) <= idx: continue - parcels.append({ - 'town': towns[idx], - 'section': sections[idx], - 'parcel_number': parcel_number}) - context['similar_files'] = models.File.similar_files(parcels) + parcels.append( + { + "town": towns[idx], + "section": sections[idx], + "parcel_number": parcel_number, + } + ) + context["similar_files"] = models.File.similar_files(parcels) else: # edition only try: numeric_reference = int( self.session_get_value( - 'instruction-' + self.url_name, - 'numeric_reference')) + "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() + year=self.get_current_year(), + ).exclude(pk=self.get_current_object().pk) + context["numeric_reference_files"] = q.all() except ValueError: pass |