diff options
Diffstat (limited to 'tablib')
| -rw-r--r-- | tablib/formats/_csv.py | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/tablib/formats/_csv.py b/tablib/formats/_csv.py index 8b536a7..9e8515a 100644 --- a/tablib/formats/_csv.py +++ b/tablib/formats/_csv.py @@ -13,8 +13,8 @@ extensions = ('csv',) DEFAULT_DELIMITER = unicode(',') -def export_set(dataset, **kwargs): - """Returns CSV representation of Dataset.""" +def export_stream_set(dataset, **kwargs): + """Returns CSV representation of Dataset as file-like.""" stream = StringIO() kwargs.setdefault('delimiter', DEFAULT_DELIMITER) @@ -24,6 +24,13 @@ def export_set(dataset, **kwargs): for row in dataset._package(dicts=False): _csv.writerow(row) + stream.seek(0) + return stream + + +def export_set(dataset, **kwargs): + """Returns CSV representation of Dataset.""" + stream = export_stream_set(dataset, **kwargs) return stream.getvalue() |