QA: duplicate find

1 files changed, 29 insertions, 0 deletions

diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index b8a23367b..ed0389cff 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -936,6 +936,35 @@ class QAFindBasketFormView(QAItemForm):
         return HttpResponseRedirect(reverse("success"))
 
 
+class QAFindDuplicateFormView(QAItemForm):
+    template_name = 'ishtar/forms/qa_find_duplicate.html'
+    model = models.Find
+    page_name = _(u"Duplicate")
+    form_class = forms.QAFindDuplicateForm
+
+    def get_quick_action(self):
+        return models.Find.QUICK_ACTIONS[1]
+
+    def get_form_kwargs(self):
+        kwargs = super(QAFindDuplicateFormView, self).get_form_kwargs()
+        kwargs['user'] = self.request.user
+        return kwargs
+
+    def form_valid(self, form):
+        form.save()
+        return HttpResponseRedirect(reverse("success"))
+
+    def get_context_data(self, **kwargs):
+        data = super(QAFindDuplicateFormView, self).get_context_data(
+            **kwargs)
+        data['action_name'] = _(u"Duplicate")
+        bf = self.items[0].get_first_base_find()
+        if bf:
+            data['context_record'] = bf.context_record
+            data['operation'] = bf.context_record.operation
+        return data
+
+
 class QAFindTreatmentFormView(QAItemForm):
     template_name = 'ishtar/forms/qa_find_treatment.html'
     model = models.Find