diff options
Diffstat (limited to 'archaeological_finds/views.py')
| -rw-r--r-- | archaeological_finds/views.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py index 783b336f2..a449a2115 100644 --- a/archaeological_finds/views.py +++ b/archaeological_finds/views.py @@ -329,8 +329,13 @@ class DeleteFindBasketView(IshtarMixin, LoginRequiredMixin, FormView): form.save() return HttpResponseRedirect(self.get_success_url()) -get_treatment = get_item( - models.FindTreatments, 'get_treatment', 'treatment', +get_upstreamtreatment = get_item( + models.FindUpstreamTreatments, 'get_upstreamtreatment', 'uptreatment', + extra_request_keys={'find_id': 'find_id'}) + +get_downstreamtreatment = get_item( + models.FindDownstreamTreatments, 'get_downstreamtreatment', + 'downtreatment', extra_request_keys={'find_id': 'find_id'}) treatment_creation_wizard = TreatmentWizard.as_view([ |