summaryrefslogtreecommitdiff
path: root/archaeological_context_records/views.py
diff options
context:
space:
mode:
Diffstat (limited to 'archaeological_context_records/views.py')
-rw-r--r--archaeological_context_records/views.py25
1 files changed, 23 insertions, 2 deletions
diff --git a/archaeological_context_records/views.py b/archaeological_context_records/views.py
index 771a2ed71..6a01a48f0 100644
--- a/archaeological_context_records/views.py
+++ b/archaeological_context_records/views.py
@@ -26,11 +26,15 @@ from django.shortcuts import redirect
from django.utils.translation import ugettext_lazy as _
from django.views.generic import RedirectView
+from ishtar_common.models import QuickAction
+from archaeological_operations.models import Operation
from archaeological_context_records import models
+
from archaeological_operations.views import site_extra_context
from archaeological_context_records import forms
from ishtar_common.utils import put_session_message
-from ishtar_common.views import IshtarMixin, LoginRequiredMixin
+
+from ishtar_common.views import IshtarMixin, LoginRequiredMixin, QAItemForm
from ishtar_common.views_item import display_item, get_item, show_item, \
revert_item
from archaeological_context_records import wizards
@@ -171,4 +175,21 @@ class GenerateRelationImage(IshtarMixin, LoginRequiredMixin, RedirectView):
except models.ContextRecord.DoesNotExist:
raise Http404()
self.context_record.generate_relation_image()
- return super(GenerateRelationImage, self).get(request, *args, **kwargs) \ No newline at end of file
+ return super(GenerateRelationImage, self).get(request, *args, **kwargs)
+
+
+class QAOperationContextRecordView(QAItemForm):
+ template_name = 'ishtar/forms/qa_operation_contextrecord.html'
+ model = Operation
+ form_class = forms.QAOperationCR
+ page_name = _(u"Context record")
+
+ def get_quick_action(self):
+ return QuickAction(
+ url="operation-qa-contextrecord", icon_class="fa fa-plus",
+ text=_(u"Add context record"), target="one",
+ rights=['add_contextrecord', 'add_own_contextrecord'])
+
+ def form_valid(self, form):
+ form.save(self.items)
+ return HttpResponseRedirect(reverse("success"))