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

1 files changed, 27 insertions, 0 deletions

diff --git a/archaeological_operations/forms.py b/archaeological_operations/forms.py
index ae9a18862..91d421541 100644
--- a/archaeological_operations/forms.py
+++ b/archaeological_operations/forms.py
@@ -1296,6 +1296,14 @@ class SiteForm(ManageOldType):
     remain = forms.MultipleChoiceField(
         label=_("Remains"), choices=[], widget=widgets.Select2Multiple,
         required=False)
+    locality_ngi = forms.CharField(
+        label=_(u"National Geographic Institute locality"),
+        widget=forms.Textarea, required=False
+    )
+    locality_cadastral = forms.CharField(
+        label=_(u"Cadastral locality"),
+        widget=forms.Textarea, required=False
+    )
 
     TYPES = [
         FieldType('period', models.Period, True),
@@ -1319,6 +1327,25 @@ SiteTownFormset.form_admin_name = _(u"Archaeological site - 020 - Towns")
 SiteTownFormset.form_slug = u"archaeological_site-020-towns"
 
 
+def check_underwater_module(self):
+    return get_current_profile().underwater
+
+
+class SiteUnderwaterForm(ManageOldType):
+    form_label = _(u"Underwater")
+    form_admin_name = _(u"Archaeological site - 030 - Underwater")
+    form_slug = u"archaeological_site-030-underwater"
+
+    shipwreck_code = forms.CharField(
+        label=_(u"Shipwreck code"), required=False)
+    sinking_date = DateField(
+        label=_(u"Sinking date"), required=False)
+    oceanographic_service_localisation = forms.CharField(
+        label=_("Oceanographic service localisation"),
+        widget=forms.Textarea, required=False
+    )
+
+
 ####################################
 # Source management for operations #
 ####################################