Same codebase for 2.x and 3.x!

tablib/formats/_csv.py

@@ -3,7 +3,13 @@
 """ Tablib - CSV Support.
 """
 
-import io
+import sys
+if sys.version_info.major > 2:
+    from io import StringIO
+else:
+    from cStringIO import StringIO
+
+
 import csv
 import os
 
@@ -17,7 +23,7 @@ extentions = ('csv',)
 
 def export_set(dataset):
     """Returns CSV representation of Dataset."""
-    stream = io.StringIO()
+    stream = StringIO()
     _csv = csv.writer(stream)
 
     for row in dataset._package(dicts=False):

tablib/formats/_html.py

@@ -4,12 +4,13 @@
 """
 
 import sys
-from io import StringIO
 
 
 if sys.version_info.major > 2:
+    from io import StringIO
     from tablib.packages import markup3 as markup
 else:
+    from cStringIO import StringIO
     from tablib.packages import markup
 
 import tablib

tablib/formats/_tsv.py

@@ -3,7 +3,12 @@
 """ Tablib - TSV (Tab Separated Values) Support.
 """
 
-import io
+import sys
+if sys.version_info.major > 2:
+    from io import StringIO
+else:
+    from cStringIO import StringIO
+    
 import csv
 import os
 
@@ -17,7 +22,7 @@ extentions = ('tsv',)
 
 def export_set(dataset):
     """Returns a TSV representation of Dataset."""
-    stream = io.StringIO()
+    stream = StringIO()
     _tsv = csv.writer(stream, delimiter='\t')
 
     for row in dataset._package(dicts=False):