From 601bcd199a5078ad951f3bce77edc0f1f940a790 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@proxience.com>
Date: Tue, 18 Aug 2015 15:07:58 +0200
Subject: Archaeological files: associate many towns. PEP8 and formating.

---
 archaeological_files_pdl/wizards.py | 24 +++++++++++++-----------
 1 file changed, 13 insertions(+), 11 deletions(-)

(limited to 'archaeological_files_pdl/wizards.py')

diff --git a/archaeological_files_pdl/wizards.py b/archaeological_files_pdl/wizards.py
index 2d3491c8d..c081d13d1 100644
--- a/archaeological_files_pdl/wizards.py
+++ b/archaeological_files_pdl/wizards.py
@@ -19,21 +19,22 @@
 
 from archaeological_files.wizards import FileWizard as BaseFileWizard
 
+
 class FileWizard(BaseFileWizard):
     parcel_step_key = 'parcelspdl-'
     town_step_keys = ['preventiveplanning-', 'researchaddress-']
-    town_input_id = 'main_town'
+    town_input_id = 'town'
     multi_towns = False
     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',
-                    }
+        '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',
+    }
 
     def get_current_year(self):
         general_form_key = 'general-' + self.url_name
@@ -41,7 +42,7 @@ class FileWizard(BaseFileWizard):
 
     def get_form_kwargs(self, *args, **kwargs):
         returned = super(FileWizard, self).get_form_kwargs(*args, **kwargs)
-        if args and (args[0].startswith('generalcontractor-') or 
+        if args and (args[0].startswith('generalcontractor-') or
                      args[0].startswith('planningservice-')):
             if 'status' in self.request.GET:
                 returned['status'] = self.request.GET['status']
@@ -49,5 +50,6 @@ class FileWizard(BaseFileWizard):
             returned['year'] = self.get_current_year()
         return returned
 
+
 class FileModificationWizard(FileWizard):
     modification = True
-- 
cgit v1.2.3