QA packaging

1 files changed, 21 insertions, 2 deletions

diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index fd041b511..30a382e2c 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -601,7 +601,7 @@ class QAFindForm(QAItemEditForm):
     form_class = QAFindFormMulti
 
 
-class QAFindBasketForm(QAItemForm):
+class QAFindBasketFormView(QAItemForm):
     template_name = 'ishtar/forms/qa_find_basket.html'
     model = models.Find
     form_class = QAFindBasketForm
@@ -612,10 +612,29 @@ class QAFindBasketForm(QAItemForm):
         return models.Find.QUICK_ACTIONS[1]
 
     def get_form_kwargs(self):
-        kwargs = super(QAFindBasketForm, self).get_form_kwargs()
+        kwargs = super(QAFindBasketFormView, self).get_form_kwargs()
         kwargs['user'] = self.request.user
         return kwargs
 
     def form_valid(self, form):
         form.save(self.items)
         return HttpResponseRedirect(reverse("success"))
+
+
+class QAFindTreatmentFormView(QAItemForm):
+    template_name = 'ishtar/forms/qa_find_treatment.html'
+    model = models.Find
+    form_class = QAFindTreatmentForm
+    page_name = _(u"Packaging")
+
+    def get_quick_action(self):
+        return models.Find.QUICK_ACTIONS[2]
+
+    def get_form_kwargs(self):
+        kwargs = super(QAFindTreatmentFormView, self).get_form_kwargs()
+        kwargs['user'] = self.request.user
+        return kwargs
+
+    def form_valid(self, form):
+        form.save(self.items, self.request.user)
+        return HttpResponseRedirect(reverse("success"))