Simple treatment form. Treatment listing. (refs #3365)

1 files changed, 13 insertions, 2 deletions

diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index 89babe9b7..f0c67c231 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -45,6 +45,9 @@ get_find = get_item(models.Find, 'get_find', 'find')
 get_find_for_ope = get_item(models.Find, 'get_find', 'find',
                             own_table_cols=models.Find.TABLE_COLS_FOR_OPE)
 
+show_treatment = show_item(models.Treatment, 'treatment')
+get_treatment = get_item(models.Treatment, 'get_treatement', 'treatment')
+
 show_findsource = show_item(models.FindSource, 'findsource')
 
 get_findsource = get_item(models.FindSource, 'get_findsource', 'findsource')
@@ -282,12 +285,20 @@ get_downstreamtreatment = get_item(
     models.FindDownstreamTreatments, 'get_downstreamtreatment',
     'downtreatment')
 
-treatment_creation_wizard = TreatmentWizard.as_view([
+treatment_wizard_steps = [
     ('basetreatment-treatment_creation', BaseTreatmentForm),
     ('selecfind-treatment_creation', UpstreamFindFormSelection),
     ('resultfind-treatment_creation', ResultFindForm),
     ('resultfinds-treatment_creation', ResultFindFormSet),
-    ('final-treatment_creation', FinalForm)],
+    ('final-treatment_creation', FinalForm)]
+
+treatment_search_wizard = SearchWizard.as_view([
+    ('general-treatment_search', TreatmentFormSelection)],
+    label=_(u"Treatment search"),
+    url_name='treatment_search',)
+
+treatment_creation_wizard = TreatmentWizard.as_view(
+    treatment_wizard_steps,
     condition_dict={
     'selecfind-treatment_creation':
         check_not_exist('basetreatment-treatment_creation',