diff options
Diffstat (limited to 'archaeological_context_records/views.py')
| -rw-r--r-- | archaeological_context_records/views.py | 25 | 
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")) | 
