2.x bytesio fix

tablib/compat.py

@@ -28,6 +28,7 @@ if is_py3:
     from tablib.packages.odf3 import opendocument, style, text, table
 
     import csv
+    from io import StringIO
     # py3 mappings
 
     unicode = str
@@ -36,6 +37,7 @@ if is_py3:
 
 else:
     from cStringIO import StringIO as BytesIO
+    from cStringIO import StringIO
     import tablib.packages.xlwt as xlwt
     from tablib.packages import markup
     from itertools import ifilter

tablib/formats/_csv.py

@@ -7,7 +7,7 @@
 import os
 
 import tablib
-from tablib.compat import is_py3, BytesIO, csv
+from tablib.compat import is_py3, csv, StringIO
 
 
 title = 'csv'
@@ -20,7 +20,7 @@ DEFAULT_ENCODING = 'utf-8'
 
 def export_set(dataset):
     """Returns CSV representation of Dataset."""
-    stream = BytesIO()
+    stream = StringIO()
 
     if is_py3:
         _csv = csv.writer(stream)

tablib/formats/_tsv.py

@@ -6,7 +6,7 @@
 import os
 
 import tablib
-from tablib.compat import is_py3, csv, BytesIO
+from tablib.compat import is_py3, csv, StringIO
 
 
 
@@ -17,7 +17,7 @@ DEFAULT_ENCODING = 'utf-8'
 
 def export_set(dataset):
     """Returns a TSV representation of Dataset."""
-    stream = BytesIO()
+    stream = StringIO()
 
     if is_py3:
         _tsv = csv.writer(stream, delimiter="\t")