added a png repr that uses Dot.write_png

log0 · Jun 15, 2016 · 3b54ff0 · 3b54ff0
1 parent dda3d15
commit 3b54ff0
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pydot/pydot.py b/pydot/pydot.py
@@ -27,6 +27,7 @@
 import subprocess
 import tempfile
 import copy
+
 try:
     import dot_parser
 except Exception, e:
@@ -1912,6 +1913,15 @@ def write(self, path, prog=None, format='raw'):
         return True
 
 
+    def _repr_png_(self):
+        """IPython display hook support.
+        """
+        png_filename = tempfile.mktemp('.png')
+        self.write_png(png_filename)
+        with open(png_filename, 'rb') as f:
+            return f.read()
+
+
 
     def create(self, prog=None, format='ps'):
         """Creates and returns a Postscript representation of the graph.