IPython support.

leather/chart.py

@@ -12,7 +12,7 @@
 from leather.shapes import Bars, Columns, Dots, Lines
 import leather.svg as svg
 from leather import theme
-from leather.utils import X, Y, DIMENSIONS, Box
+from leather.utils import X, Y, DIMENSIONS, Box, IPythonSVG
 
 
 class Chart(object):
@@ -243,8 +243,9 @@ def to_svg(self, path=None, width=None, height=None):
         as though they were pixels.
 
         :param path:
-            Filepath or file-like object to write to. If not specified then
-            the SVG will be returned as a string.
+            Filepath or file-like object to write to. If omitted then the SVG
+            will be returned as a string. If running within IPython, then this
+            will return a SVG object to be displayed.
         :param width:
             The output width, in SVG user units.
         :param height:
@@ -285,4 +286,4 @@ def to_svg(self, path=None, width=None, height=None):
                 if close and f is not None:
                     f.close()
         else:
-            return svg_text
+            return IPythonSVG(svg_text)

leather/grid.py

@@ -8,6 +8,7 @@
 
 import leather.svg as svg
 from leather import theme
+from leather.utils import IPythonSVG
 
 
 class Grid(object):
@@ -83,4 +84,4 @@ def to_svg(self, path=None, width=None, height=None):
                 if close and f is not None:
                     f.close()
         else:
-            return svg_text
+            return IPythonSVG(svg_text)

leather/utils.py

@@ -2,6 +2,13 @@
 
 from collections import namedtuple
 
+try:
+    __IPYTHON__
+    from IPython.display import SVG as IPythonSVG
+except (NameError, ImportError):
+    IPythonSVG = lambda x: x
+
+
 #: X data dimension index
 X = 0
 

tests/utils.py

@@ -6,6 +6,7 @@
     import unittest
 
 from lxml import etree
+import six
 
 
 class XMLTest(unittest.TestCase):
@@ -17,6 +18,7 @@ def render_chart(self, chart):
         Verify the column names in the given table match what is expected.
         """
         text = chart.to_svg()
+        print(dir(text))
         text = text.replace(' xmlns="http://www.w3.org/2000/svg"', '')
 
         return etree.fromstring(text)