Merge pull request #46 from AdriaanRol/feat/plotless_option

plotless option
AdriaanRol · Jun 19, 2017 · 64b21aa · 64b21aa
2 parents 7658917 + 03b0f16
commit 64b21aa
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/autodepgraph/graph.py b/autodepgraph/graph.py
@@ -12,6 +12,12 @@
 except ImportError:
     plot_mode = 'mpl'
 
+try:
+    import pygraphviz
+except ImportError:
+    logging.warning('pygraphviz is not installed, plotting will be disabled')
+    plot_mode = 'none'
+
 
 class Graph(Instrument):
     """
@@ -111,6 +117,8 @@ def clear_node_state(self):
     def update_monitor(self):
         if self.plot_mode == 'mpl':
             self.update_monitor_mpl()
+        elif self.plot_mode == 'none':
+            return
         else:
             self.update_monitor_pg()