Adapt search form for sites

1 files changed, 33 insertions, 0 deletions

diff --git a/archaeological_operations/forms.py b/archaeological_operations/forms.py
index af4e21e31..dc389c81c 100644
--- a/archaeological_operations/forms.py
+++ b/archaeological_operations/forms.py
@@ -1231,6 +1231,8 @@ class OperationDeletionForm(FinalForm):
 
 
 class SiteSelect(TableSelect):
+    _model = models.ArchaeologicalSite
+
     search_vector = forms.CharField(
         label=_(u"Full text search"), widget=widgets.SearchWidget(
             'archaeological-operations', 'archaeologicalsite'))
@@ -1239,11 +1241,42 @@ class SiteSelect(TableSelect):
     name = forms.CharField(label=_(u"Name"), max_length=200, required=False)
     periods = forms.ChoiceField(label=_(u"Periods"), choices=[], required=False)
     remains = forms.ChoiceField(label=_(u"Remains"), choices=[], required=False)
+    towns = get_town_field()
+    comment = forms.CharField(label=_(u"Comment"), max_length=200,
+                              required=False)
+    locality_ngi = forms.CharField(
+        label=_(u"National Geographic Institute locality"), max_length=200,
+        required=False)
+    locality_cadastral = forms.CharField(
+        label=_(u"Cadastral locality"), max_length=200,
+        required=False)
+    shipwreck_name = forms.CharField(
+        label=_(u"Shipwreck name"), max_length=200,
+        required=False)
+    oceanographic_service_localisation = forms.CharField(
+        label=_(u"Oceanographic service localisation"), max_length=200,
+        required=False)
+    shipwreck_code = forms.CharField(
+        label=_(u"Shipwreck code"), max_length=200,
+        required=False)
+    sinking_date = DateField(label=_(u"Sinking date"), required=False)
+    discovery_area = forms.CharField(
+        label=_(u"Discovery area"), max_length=200,
+        required=False)
     TYPES = [
         FieldType('periods', models.Period),
         FieldType('remains', models.RemainType),
     ]
 
+    def __init__(self, *args, **kwargs):
+        super(SiteSelect, self).__init__(*args, **kwargs)
+        if not get_current_profile().underwater:
+            self.fields.pop('shipwreck_name')
+            self.fields.pop('oceanographic_service_localisation')
+            self.fields.pop('shipwreck_code')
+            self.fields.pop('sinking_date')
+            self.fields.pop('discovery_area')
+
 
 class SiteFormSelection(IshtarForm):
     SEARCH_AND_SELECT = True