1 files changed, 45 insertions, 4 deletions
diff --git a/ishtar_common/forms_common.py b/ishtar_common/forms_common.py
index 657b619e6..f2fc15c58 100644
--- a/ishtar_common/forms_common.py
+++ b/ishtar_common/forms_common.py
@@ -87,10 +87,12 @@ from ishtar_common.utils import (
     update_data,
 )
 
-from archaeological_operations.models import Operation
+from archaeological_operations.models import Operation, ArchaeologicalSite, \
+    AdministrativeAct
+from archaeological_files.models import File as ArchaeologicalFile
 from archaeological_context_records.models import ContextRecord
-from archaeological_finds.models import Find, FindBasket
-from archaeological_warehouse.models import Container
+from archaeological_finds.models import Find, FindBasket, Treatment, TreatmentFile
+from archaeological_warehouse.models import Container, Warehouse
 
 
 def get_town_field(label=_("Town"), required=True):
@@ -2599,6 +2601,12 @@ class DocumentSelect(HistorySelect):
         label=_("Operation - type"), choices=[]
     )
     operations__year = forms.IntegerField(label=_("Operation - year"))
+    site = forms.IntegerField(
+        label=_("Archaeological site"),
+        widget=widgets.JQueryAutoComplete(
+            reverse_lazy('autocomplete-archaeologicalsite'),
+            associated_model=ArchaeologicalSite),
+        validators=[models.valid_id(ArchaeologicalSite)])
     context_record = forms.IntegerField(
         label=_("Context record"),
         required=False,
@@ -2624,6 +2632,24 @@ class DocumentSelect(HistorySelect):
         validators=[models.valid_id(Find)],
     )
     find__denomination = forms.CharField(label=_("Find - denomination"), required=False)
+    treatment = forms.IntegerField(
+        label=_("Treatment"),
+        widget=widgets.JQueryAutoComplete(
+            reverse_lazy('autocomplete-treatment'),
+            associated_model=Treatment),
+    )
+    treatment_file = forms.IntegerField(
+        label=_("Treatment file"),
+        widget=widgets.JQueryAutoComplete(
+            reverse_lazy('autocomplete-treatmentfile'),
+            associated_model=TreatmentFile),
+    )
+    warehouse = forms.IntegerField(
+        label=_("Warehouse"),
+        widget=widgets.JQueryAutoComplete(
+            reverse_lazy('autocomplete-warehouse'),
+            associated_model=Warehouse),
+    )
     containers = forms.IntegerField(
         label=_("Container"),
         required=False,
@@ -2634,6 +2660,20 @@ class DocumentSelect(HistorySelect):
     )
     town = get_town_field()
     area = widgets.Select2SimpleField(label=_("Area"))
+    file = forms.IntegerField(
+        label=_("File"),
+        widget=widgets.JQueryAutoComplete(
+            reverse_lazy('autocomplete-file'),
+            associated_model=ArchaeologicalFile),
+        validators=[models.valid_id(ArchaeologicalFile)])
+    """  TODO: add cached label to administrative act
+    administrative_act = forms.IntegerField(
+        label=_("Administrative act"),
+        widget=widgets.JQueryAutoComplete(
+            reverse_lazy('autocomplete-administrativeact'),
+            associated_model=AdministrativeAct),
+        validators=[models.valid_id(AdministrativeAct)])
+    """
     receipt_date = DateField(label=_("Receipt date"))
     creation_date = DateField(label=_("Creation date"))
     receipt_date_in_documentation = DateField(label=_("Receipt date"))
@@ -2653,7 +2693,8 @@ class DocumentSelect(HistorySelect):
     PROFILE_FILTER = {
         "context_record": ["context_record"],
         "find": ["find"],
-        "warehouse": ["container"],
+        "warehouse": ["container", "warehouse"],
+        "files": ["file"]
     }