Add a condition for preventive form (refs #14)

1 files changed, 14 insertions, 1 deletions

diff --git a/ishtar/furnitures/forms.py b/ishtar/furnitures/forms.py
index 08fb912a7..3732bd528 100644
--- a/ishtar/furnitures/forms.py
+++ b/ishtar/furnitures/forms.py
@@ -127,6 +127,19 @@ class FileForm3(forms.Form):
                                   choices=models.SaisineType.get_types())
     reception_date = forms.DateField(label=_(u"Reception date"))
 
+def is_preventive(self, request, storage):
+    if storage.prefix not in request.session or \
+       'step_data' not in request.session[storage.prefix] or \
+       '0' not in request.session[storage.prefix]['step_data'] or\
+       '0-file_type' not in request.session[storage.prefix]['step_data']['0']:
+        return False
+    try:
+        file_type = int(request.session[storage.prefix]['step_data']['0']\
+                                       ['0-file_type'])
+        return file_type == models.PREVENTIVE
+    except ValueError:
+        return False
+
 file_creation_wizard = FileWizard([FileForm1, FileForm2, FileForm3],
-                                  url_name='file_creation')
+                   url_name='file_creation', condition_list={'2':is_preventive})