Add documents to containers

1 files changed, 8 insertions, 0 deletions

diff --git a/ishtar_common/forms_common.py b/ishtar_common/forms_common.py
index 16c6d269e..fadda7593 100644
--- a/ishtar_common/forms_common.py
+++ b/ishtar_common/forms_common.py
@@ -51,6 +51,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_warehouse.models import Container
 
 
 def get_town_field(label=_(u"Town"), required=True):
@@ -1270,6 +1271,12 @@ class DocumentSelect(TableSelect):
         validators=[models.valid_id(Find)])
     find__denomination = forms.CharField(label=_(u"Find - denomination"),
                                          required=False)
+    container = forms.IntegerField(
+        label=_(u"Container"), required=False,
+        widget=widgets.JQueryAutoComplete(
+            reverse_lazy('autocomplete-container'),
+            associated_model=Container),
+        validators=[models.valid_id(Container)])
 
     TYPES = [
         FieldType('source_type', models.SourceType),
@@ -1278,6 +1285,7 @@ class DocumentSelect(TableSelect):
     PROFILE_FILTER = {
         'context_record': ['context_record'],
         'find': ['find'],
+        'warehouse': ['container']
     }