Treatment searches

1 files changed, 40 insertions, 0 deletions

diff --git a/archaeological_operations/models.py b/archaeological_operations/models.py
index 082bbd2fe..34a3d2a01 100644
--- a/archaeological_operations/models.py
+++ b/archaeological_operations/models.py
@@ -1472,6 +1472,46 @@ class AdministrativeAct(BaseHistorizedItem, OwnPerms, ValueGetter):
             pgettext_lazy(TXT_SEARCH_COMMENT, u"file-permit-reference"),
             'associated_file__permit_reference__iexact'
         ),
+        'treatment__name': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"treatment-name"),
+            'treatment__label__icontains'
+        ),
+        'treatment__other_reference': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"treatment-reference"),
+            'treatment__other_reference__icontains'
+        ),
+        'treatment__year': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"treatment-year"),
+            'treatment__year'
+        ),
+        'treatment__index': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"treatment-index"),
+            'treatment__index'
+        ),
+        'treatment__treatment_types': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"treatment-type"),
+            'treatment__treatment_types__label__iexact'
+        ),
+        'treatment_file__name': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"treatment-file-name"),
+            'treatment_file__name__icontains'
+        ),
+        'treatment_file__internal_reference': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"treatment-file-reference"),
+            'treatment_file__internal_reference__icontains'
+        ),
+        'treatment_file__year': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"treatment-file-year"),
+            'treatment_file__year'
+        ),
+        'treatment_file__index': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"treatment-file-index"),
+            'treatment_file__index'
+        ),
+        'treatment_file__type': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"treatment-file-type"),
+            'treatment_file__type__label__iexact'
+        ),
     }
     for v in ALT_NAMES.values():
         EXTRA_REQUEST_KEYS[v[0]] = v[1]