Documents: search by find basket

2 files changed, 11 insertions, 1 deletions

diff --git a/ishtar_common/forms_common.py b/ishtar_common/forms_common.py
index eb4582c3e..d34c0d23f 100644
--- a/ishtar_common/forms_common.py
+++ b/ishtar_common/forms_common.py
@@ -53,7 +53,7 @@ from ishtar_common.utils import is_downloadable, clean_session_cache, \
 
 from archaeological_operations.models import Operation
 from archaeological_context_records.models import ContextRecord
-from archaeological_finds.models import Find
+from archaeological_finds.models import Find, FindBasket
 from archaeological_warehouse.models import Container
 
 
@@ -1374,6 +1374,12 @@ class DocumentSelect(HistorySelect):
             reverse_lazy('autocomplete-contextrecord'),
             associated_model=ContextRecord),
         validators=[models.valid_id(ContextRecord)])
+    find_basket = forms.IntegerField(
+        label=_("Basket - Finds"),
+        widget=widgets.JQueryAutoComplete(
+            reverse_lazy('autocomplete-findbasket'),
+            associated_model=FindBasket),
+        validators=[models.valid_id(FindBasket)], required=False)
     find = forms.IntegerField(
         label=_(u"Find"), required=False,
         widget=widgets.JQueryAutoComplete(
diff --git a/ishtar_common/models.py b/ishtar_common/models.py
index d8283e23f..0b89e6416 100644
--- a/ishtar_common/models.py
+++ b/ishtar_common/models.py
@@ -5057,6 +5057,10 @@ class Document(BaseHistorizedItem, QRCodeItem, OwnPerms, ImageModel,
             pgettext_lazy("key for text search", "context-record"),
             'context_records__cached_label__iexact'
         ),
+        'find_basket': SearchAltName(
+            pgettext_lazy("key for text search", "basket-finds"),
+            'finds__basket__label__iexact'
+        ),
         'find': SearchAltName(
             pgettext_lazy("key for text search", "find"),
             'finds__cached_label__iexact'