Sites: add locality and underwater fields (forms and sheet)

1 files changed, 5 insertions, 1 deletions

diff --git a/archaeological_operations/views.py b/archaeological_operations/views.py
index 55c688854..61c9e6c19 100644
--- a/archaeological_operations/views.py
+++ b/archaeological_operations/views.py
@@ -44,7 +44,7 @@ from archaeological_operations.forms import ArchaeologicalSiteForm, \
     AdministrativeActOpeFormSelection, AdministrativeActOpeForm, \
     AdministrativeActOpeModifForm, FinalAdministrativeActDeleteForm, \
     AdministrativeActRegisterFormSelection, DocumentGenerationAdminActForm, \
-    SiteForm, SiteTownFormset
+    SiteForm, SiteTownFormset, SiteUnderwaterForm, check_underwater_module
 
 from ishtar_common.views import get_item, show_item, revert_item, new_item
 
@@ -352,11 +352,13 @@ site_search_wizard = SiteSearch.as_view(
 site_creation_steps = [
     ('general-site_creation', SiteForm),
     ('towns-site_creation', SiteTownFormset),
+    ('underwater-site_creation', SiteUnderwaterForm),
     ('final-site_creation', FinalForm)
 ]
 
 site_creation_wizard = SiteWizard.as_view(
     site_creation_steps,
+    condition_dict={'underwater-site_creation': check_underwater_module},
     url_name='site_creation',
 )
 
@@ -364,11 +366,13 @@ site_modification_steps = [
     ('selec-site_modification', SiteFormSelection),
     ('general-site_modification', SiteForm),
     ('towns-site_modification', SiteTownFormset),
+    ('underwater-site_modification', SiteUnderwaterForm),
     ('final-site_modification', FinalForm)
 ]
 
 site_modification_wizard = SiteModificationWizard.as_view(
     site_modification_steps,
+    condition_dict={'underwater-site_modification': check_underwater_module},
     url_name='site_modification',
 )