io.BytesIO is also available in Python 2.7

tablib/compat.py

@@ -14,7 +14,6 @@ is_py3 = (sys.version_info[0] > 2)
 
 
 if is_py3:
-    from io import BytesIO
     from io import StringIO
     from statistics import median
     from itertools import zip_longest as izip_longest
@@ -25,7 +24,6 @@ if is_py3:
     xrange = range
 
 else:
-    from cStringIO import StringIO as BytesIO
     from StringIO import StringIO
     from tablib.packages.statistics import median
     from itertools import izip_longest

tablib/formats/_df.py

@@ -1,14 +1,8 @@
 """ Tablib - DataFrame Support.
 """
 
-
 import sys
-
+from io import BytesIO
-
-if sys.version_info[0] > 2:
-    from io import BytesIO
-else:
-    from cStringIO import StringIO as BytesIO
 
 try:
     from pandas import DataFrame

tablib/formats/_html.py

@@ -3,14 +3,10 @@
 """ Tablib - HTML export support.
 """
 
-import sys
-
-if sys.version_info[0] > 2:
-    from io import BytesIO as StringIO
-else:
-    from cStringIO import StringIO
-
 import codecs
+import sys
+from io import BytesIO
+
 from MarkupPy import markup
 import tablib
 from tablib.compat import unicode
@@ -24,7 +20,7 @@ extensions = ('html', )
 def export_set(dataset):
     """HTML representation of a Dataset."""
 
-    stream = StringIO()
+    stream = BytesIO()
 
     page = markup.page()
     page.table.open()
@@ -55,7 +51,7 @@ def export_set(dataset):
 def export_book(databook):
     """HTML representation of a Databook."""
 
-    stream = StringIO()
+    stream = BytesIO()
 
     # Allow unicode characters in output
     wrapper = codecs.getwriter("utf8")(stream)

tablib/formats/_ods.py

@@ -3,8 +3,9 @@
 """ Tablib - ODF Support.
 """
 
+from io import BytesIO
 from odf import opendocument, style, table, text
-from tablib.compat import BytesIO, unicode
+from tablib.compat import unicode
 
 title = 'ods'
 extensions = ('ods',)

tablib/formats/_xls.py

@@ -4,8 +4,9 @@
 """
 
 import sys
+from io import BytesIO
 
-from tablib.compat import BytesIO, xrange
+from tablib.compat import xrange
 import tablib
 import xlrd
 import xlwt

tablib/formats/_xlsx.py

@@ -4,12 +4,7 @@
 """
 
 import sys
-
+from io import BytesIO
-
-if sys.version_info[0] > 2:
-    from io import BytesIO
-else:
-    from cStringIO import StringIO as BytesIO
 
 import openpyxl
 import tablib