Manage 1<->n treatments wizard

1 files changed, 14 insertions, 2 deletions

diff --git a/archaeological_finds/forms_treatments.py b/archaeological_finds/forms_treatments.py
index 99a685825..be53dd418 100644
--- a/archaeological_finds/forms_treatments.py
+++ b/archaeological_finds/forms_treatments.py
@@ -80,7 +80,7 @@ class TreatmentFormSelection(forms.Form):
 
 
 class BaseTreatmentForm(CustomForm, ManageOldType):
-    form_label = _(u"Base treatment")
+    form_label = _(u"Treatment")
     form_admin_name = _(u"Treatment - 020 - General")
     form_slug = "treatment-020-general"
     base_models = ['treatment_type']
@@ -231,7 +231,6 @@ class BaseTreatmentForm(CustomForm, ManageOldType):
 
 
 class N1TreatmentForm(BaseTreatmentForm):
-    form_label = _(u"Base treatment")
     form_admin_name = _(u"Treatment n-1 - 020 - General")
     form_slug = "treatmentn1-020-general"
 
@@ -244,6 +243,19 @@ class N1TreatmentForm(BaseTreatmentForm):
     ]
 
 
+class OneNTreatmentForm(BaseTreatmentForm):
+    form_admin_name = _(u"Treatment 1-n - 020 - General")
+    form_slug = "treatment1n-020-general"
+
+    TYPES = [
+        FieldType('treatment_state', models.TreatmentState),
+        FieldType(
+            'treatment_type', models.TreatmentType, is_multiple=True,
+            extra_args={'dct': {'upstream_is_many': False,
+                                'downstream_is_many': True}})
+    ]
+
+
 class TreatmentModifyForm(BaseTreatmentForm):
     index = forms.IntegerField(_(u"Index"))
     id = forms.IntegerField(' ', widget=forms.HiddenInput, required=False)