posting picture back to python actually works like a charm with matpl…

…otlib redraw
benmaier · Mar 27, 2019 · 90d42dc · 90d42dc
1 parent 4a83807
commit 90d42dc
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/netwulf/interactive.py b/netwulf/interactive.py
@@ -275,6 +275,14 @@ def visualize(network,
         print('changing directory back to', cwd)
 
     os.chdir(cwd)
+
+    # see whether or not the whole thing was started from a jupyter notebook and if yes,
+    # actually re-draw the figure and display it
+    env = os.environ
+    program = os.path.basename(env['_'])
+    if 'jupyter' in program:        
+        fig, ax = wulf.draw_netwulf(posted_network_properties)
+
 
     return posted_network_properties, posted_config