Sheet actions: add duplicate for site, operation, document and context record

1 files changed, 23 insertions, 0 deletions

diff --git a/archaeological_context_records/views.py b/archaeological_context_records/views.py
index 74def46a4..215dd2b4a 100644
--- a/archaeological_context_records/views.py
+++ b/archaeological_context_records/views.py
@@ -233,3 +233,26 @@ class QAContextRecordLockView(QABaseLockView):
     model = models.ContextRecord
     base_url = "contextrecord-qa-lock"
 
+
+class QAContextRecordDuplicateFormView(QAItemForm):
+    template_name = 'ishtar/forms/qa_contextrecord_duplicate.html'
+    model = models.ContextRecord
+    page_name = _("Duplicate")
+    form_class = forms.QAContextRecordDuplicateForm
+    base_url = "contextrecord-qa-duplicate"
+
+    def get_form_kwargs(self):
+        kwargs = super(QAContextRecordDuplicateFormView, 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(QAContextRecordDuplicateFormView, self).get_context_data(
+            **kwargs)
+        data['action_name'] = _(u"Duplicate")
+        data["operation"] = self.items[0].operation
+        return data