From 182f4b4f233986bda0fc277ba7bbc9c71d8de692 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Mon, 8 Jul 2019 12:11:24 +0200 Subject: Graph generation: refactor view - add an option in profile --- archaeological_context_records/views.py | 21 ++++++++++++++------- 1 file changed, 14 insertions(+), 7 deletions(-) (limited to 'archaeological_context_records/views.py') diff --git a/archaeological_context_records/views.py b/archaeological_context_records/views.py index 73c77fa48..fbe382623 100644 --- a/archaeological_context_records/views.py +++ b/archaeological_context_records/views.py @@ -164,18 +164,25 @@ def reset_wizards(request): class GenerateRelationImage(IshtarMixin, LoginRequiredMixin, RedirectView): + model = models.ContextRecord + def get_redirect_url(self, *args, **kwargs): - return reverse('display-' + models.ContextRecord.SLUG, - args=[self.context_record.pk]) + return reverse('display-item', + args=[self.model.SLUG, self.item.pk]) def get(self, request, *args, **kwargs): try: - self.context_record = models.ContextRecord.objects.get( - pk=kwargs['pk'] - ) - except models.ContextRecord.DoesNotExist: + self.item = self.model.objects.get(pk=kwargs['pk']) + except self.model.DoesNotExist: raise Http404() - self.context_record.generate_relation_image() + render_above, render_bellow = True, True + render_type = kwargs.get("type", None) + if render_type == "above": + render_bellow = False + elif render_type == "bellow": + render_above = False + self.item.generate_relation_image( + render_bellow=render_bellow, render_above=render_above) return super(GenerateRelationImage, self).get(request, *args, **kwargs) -- cgit v1.2.3