Context record relations: manage full graph from operation

1 files changed, 25 insertions, 6 deletions

diff --git a/archaeological_context_records/views.py b/archaeological_context_records/views.py
index 122da8c2f..e27aacb41 100644
--- a/archaeological_context_records/views.py
+++ b/archaeological_context_records/views.py
@@ -165,25 +165,44 @@ def reset_wizards(request):
 
 
 class GenerateRelationImage(IshtarMixin, LoginRequiredMixin, RedirectView):
+    upper_model = models.Operation
     model = models.ContextRecord
 
     def get_redirect_url(self, *args, **kwargs):
+        if self.upper_item:
+            return reverse(
+                'display-item',
+                args=[self.upper_model.SLUG, self.upper_item.pk]) + "#cr"
         return reverse('display-item',
                        args=[self.model.SLUG, self.item.pk]) + "#relations"
 
     def get(self, request, *args, **kwargs):
-        try:
-            self.item = self.model.objects.get(pk=kwargs['pk'])
-        except self.model.DoesNotExist:
-            raise Http404()
-        render_above, render_bellow = True, True
+        self.upper_item = None
         render_type = kwargs.get("type", None)
+        if render_type == "full" and self.upper_model:
+            try:
+                self.upper_item = self.upper_model.objects.get(pk=kwargs['pk'])
+            except self.model.DoesNotExist:
+                raise Http404()
+            q = self.upper_item.context_record_relations_q()
+            if not q.count():
+                return super(GenerateRelationImage, self).get(request, *args,
+                                                              **kwargs)
+            self.item = q.all()[0].left_record
+        else:
+            try:
+                self.item = self.model.objects.get(pk=kwargs['pk'])
+            except self.model.DoesNotExist:
+                raise Http404()
+        render_above, render_bellow, full = True, True, False
         if render_type == "above":
             render_bellow = False
         elif render_type == "bellow":
             render_above = False
+        elif render_type == "full":
+            full = True
         self.item.generate_relation_image(
-            render_bellow=render_bellow, render_above=render_above)
+            render_bellow=render_bellow, render_above=render_above, full=full)
         return super(GenerateRelationImage, self).get(request, *args, **kwargs)