draft and temporary testing for #5231

1 files changed, 17 insertions, 0 deletions

diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index d8fd48353..4d8430359 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -654,6 +654,17 @@ treatment_creation_n1_wizard = wizards.TreatmentN1Wizard.as_view(
     url_name="treatment_creation_n1",
 )
 
+
+def treatmentN1_create(request, pk):
+    if not wizard_is_available(treatment_creation_n1_wizard, models.Treatment):
+        return HttpResponseRedirect("/")
+    wizards.TreatmentWizard.session_set_value(
+        request, "selecfind-treatment_creation_n1", "pk", pk, reset=True
+    )
+    return redirect(
+        "treatment_creation_n1", kwargs={"step": "file-treatment_creation_n1"}
+    )
+
 treatment_1n_wizard_steps = [
     ("selecfind-treatment_creation_1n", forms.SingleUpstreamFindFormSelection),
     ("file-treatment_creation_1n", forms.TreatmentFormFileChoice),
@@ -1307,6 +1318,12 @@ class QAFindLockView(QABaseLockView):
     base_url = "find-qa-lock"
 
 
+class QAFindTreatmentN1View(QAItemForm):
+    model = models.Treatment
+    form_class = forms.QAFindTreatmentN1Form
+    base_url = "find-qa-treatmentN1"
+
+
 def get_geo_items(request, get_polygons, current_right=None):
     operation_pk = request.GET.get("operation_pk")
     context_record_pk = request.GET.get("context_record_pk")