Added stream to CSV

jazzband · Aug 8, 2018 · 36d3a40 · 36d3a40
1 parent 4c300e6
commit 36d3a40
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/tablib/formats/_csv.py b/tablib/formats/_csv.py
@@ -14,8 +14,8 @@
 DEFAULT_DELIMITER = ','
 
 
-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)
@@ -27,6 +27,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()