Manage search on json fields - HistorySelect for all concerned forms

1 files changed, 9 insertions, 3 deletions

diff --git a/archaeological_finds/forms_treatments.py b/archaeological_finds/forms_treatments.py
index 5471cea23..203eb0209 100644
--- a/archaeological_finds/forms_treatments.py
+++ b/archaeological_finds/forms_treatments.py
@@ -33,7 +33,7 @@ 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
+    ManageOldType, CustomForm, FieldType, IshtarForm, HistorySelect
 from ishtar_common.models import Person, valid_id, Organization, \
     get_current_profile
 
@@ -42,8 +42,10 @@ logger = logging.getLogger(__name__)
 # Treatment
 
 
-class TreatmentSelect(TableSelect):
+class TreatmentSelect(HistorySelect):
     _model = models.Treatment
+    form_admin_name = _(u"Treatment - 001 - Search")
+    form_slug = "treatment-001-search"
 
     search_vector = forms.CharField(
         label=_(u"Full text search"), widget=widgets.SearchWidget(
@@ -614,7 +616,11 @@ class AdministrativeActTreatmentModifForm(
 # treatment requests
 
 
-class TreatmentFileSelect(TableSelect):
+class TreatmentFileSelect(HistorySelect):
+    _model = models.TreatmentFile
+    form_admin_name = _(u"Treatment file - 001 - Search")
+    form_slug = "treatmentfile-001-search"
+
     search_vector = forms.CharField(
         label=_(u"Full text search"), widget=widgets.SearchWidget(
             'archaeological-finds', 'treatmentfile'