Manage relation types in searches

1 files changed, 23 insertions, 3 deletions

diff --git a/archaeological_operations/models.py b/archaeological_operations/models.py
index dd3302251..6f083fe7f 100644
--- a/archaeological_operations/models.py
+++ b/archaeological_operations/models.py
@@ -410,14 +410,34 @@ class Operation(ClosedItem, BaseHistorizedItem, OwnPerms, ValueGetter,
             pgettext_lazy(TXT_SEARCH_COMMENT, u"operator"),
             'operator__cached_label__iexact'
         ),
+        'remains': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"remain"),
+            'remains__pk'
+        ),
         'periods': (
             pgettext_lazy(TXT_SEARCH_COMMENT, u"period"),
             'periods__pk'
         ),
-        'remains': (
-            pgettext_lazy(TXT_SEARCH_COMMENT, u"remain"),
-            'remains__pk'
+        'start_before': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"start-before"),
+            'start_date__lte'
+        ),
+        'start_after': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"start-after"),
+            'start_date__gte'
+        ),
+        'end_before': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"end-before"),
+            'excavation_end_date__lte'
         ),
+        'end_after': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"end-after"),
+            'excavation_end_date__gte'
+        ),
+        'relation_types': (
+            pgettext_lazy(TXT_SEARCH_COMMENT, u"relation-types"),
+            'relation_types'
+        )
     }
     for v in ALT_NAMES.values():
         EXTRA_REQUEST_KEYS[v[0]] = v[1]