diff --git a/tablib/formats/_csv.py b/tablib/formats/_csv.py index 8174501..46a26d5 100644 --- a/tablib/formats/_csv.py +++ b/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): diff --git a/tablib/formats/_html.py b/tablib/formats/_html.py index bfa3b5c..64be2af 100644 --- a/tablib/formats/_html.py +++ b/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 diff --git a/tablib/formats/_tsv.py b/tablib/formats/_tsv.py index 708c4b9..ab10d32 100644 --- a/tablib/formats/_tsv.py +++ b/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):