Graph generation: refactor view - add an option in profile

1 files changed, 7 insertions, 2 deletions

diff --git a/ishtar_common/models.py b/ishtar_common/models.py
index e8b6fc518..b2f235207 100644
--- a/ishtar_common/models.py
+++ b/ishtar_common/models.py
@@ -1241,8 +1241,11 @@ class RelationItem(models.Model):
     class Meta:
         abstract = True
 
-    def generate_relation_image(self):
-        generate_relation_graph(self)
+    def generate_relation_image(self, highlight_current=True,
+                                render_above=True, render_bellow=True):
+        generate_relation_graph(self, highlight_current=highlight_current,
+                                render_above=render_above,
+                                render_bellow=render_bellow)
 
 
 class JsonDataSection(models.Model):
@@ -2633,6 +2636,8 @@ class IshtarSiteProfile(models.Model, Cached):
     )
     use_town_for_geo = models.BooleanField(
         _("Use town to locate when coordinates are missing"), default=True)
+    relation_graph = models.BooleanField(_("Generate relation graph"),
+                                         default=False)
     underwater = models.BooleanField(_("Underwater module"), default=False)
     parcel_mandatory = models.BooleanField(
         _("Parcel are mandatory for context records"), default=True)