Searc criteria: add has image/file/url criteria for all document items

1 files changed, 4 insertions, 5 deletions

diff --git a/archaeological_finds/forms_treatments.py b/archaeological_finds/forms_treatments.py
index f03dec18a..b80b6c0fe 100644
--- a/archaeological_finds/forms_treatments.py
+++ b/archaeological_finds/forms_treatments.py
@@ -33,8 +33,8 @@ from archaeological_warehouse.models import Warehouse, Container
 from bootstrap_datepicker.widgets import DatePicker
 from ishtar_common import widgets
 from ishtar_common.forms import reverse_lazy, TableSelect, FinalForm, \
-    ManageOldType, CustomForm, FieldType, IshtarForm, HistorySelect, \
-    MultiSearchForm
+    ManageOldType, CustomForm, FieldType, IshtarForm, \
+    DocumentItemSelect, MultiSearchForm
 from ishtar_common.models import Person, valid_id, valid_ids, Organization, \
     get_current_profile
 
@@ -43,7 +43,7 @@ logger = logging.getLogger(__name__)
 # Treatment
 
 
-class TreatmentSelect(HistorySelect):
+class TreatmentSelect(DocumentItemSelect):
     _model = models.Treatment
     form_admin_name = _(u"Treatment - 001 - Search")
     form_slug = "treatment-001-search"
@@ -63,7 +63,6 @@ class TreatmentSelect(HistorySelect):
             associated_model=Person),
         label=_(u"Scientific monitoring manager"))
     treatment_types = forms.ChoiceField(label=_(u"Treatment type"), choices=[])
-    documents__image__isnull = forms.NullBooleanField(label=_(u"Has an image?"))
 
     def __init__(self, *args, **kwargs):
         super(TreatmentSelect, self).__init__(*args, **kwargs)
@@ -662,7 +661,7 @@ class AdministrativeActTreatmentModifForm(
 # treatment requests
 
 
-class TreatmentFileSelect(HistorySelect):
+class TreatmentFileSelect(DocumentItemSelect):
     _model = models.TreatmentFile
     form_admin_name = _(u"Treatment file - 001 - Search")
     form_slug = "treatmentfile-001-search"