Display generation button for relation image on the sheet

1 files changed, 21 insertions, 2 deletions

diff --git a/archaeological_context_records/views.py b/archaeological_context_records/views.py
index 53ef90a40..9c2f14406 100644
--- a/archaeological_context_records/views.py
+++ b/archaeological_context_records/views.py
@@ -21,12 +21,14 @@ import json
 
 from django.core.urlresolvers import reverse
 from django.db.models import Q
-from django.http import HttpResponse, HttpResponseRedirect
+from django.http import HttpResponse, HttpResponseRedirect, Http404
 from django.shortcuts import redirect
+from django.views.generic import RedirectView
 from django.utils.translation import ugettext_lazy as _
 
 from ishtar_common.forms_common import AuthorFormset, SourceForm
-from ishtar_common.views import get_item, show_item, revert_item
+from ishtar_common.views import get_item, show_item, revert_item, \
+    IshtarMixin, LoginRequiredMixin, display_item
 from archaeological_operations.views import site_extra_context
 from ishtar_common.wizards import SearchWizard
 
@@ -41,6 +43,7 @@ show_contextrecord = show_item(
     extra_dct=site_extra_context
 )
 revert_contextrecord = revert_item(models.ContextRecord)
+display_contextrecord = display_item(models.ContextRecord)
 
 
 def autocomplete_contextrecord(request):
@@ -194,3 +197,19 @@ def reset_wizards(request):
             (RecordSourceWizard, 'record_source_modification'),
             (RecordSourceDeletionWizard, 'record_source_deletion')):
         wizard_class.session_reset(request, url_name)
+
+
+class GenerateRelationImage(IshtarMixin, LoginRequiredMixin, RedirectView):
+    def get_redirect_url(self, *args, **kwargs):
+        return reverse('display-' + models.ContextRecord.SLUG,
+                       args=[self.context_record.pk])
+
+    def get(self, request, *args, **kwargs):
+        try:
+            self.context_record = models.ContextRecord.objects.get(
+                pk=kwargs['pk']
+            )
+        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