From f55f56ae1d5d5fb22dcf0b20fc2bbff62c86a76c Mon Sep 17 00:00:00 2001
From: Anthony Monthe <anthony.monthe@gmail.com>
Date: Wed, 8 Aug 2018 15:05:36 +0100
Subject: [PATCH] Added stream to CSV

---
 tablib/formats/_csv.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/tablib/formats/_csv.py b/tablib/formats/_csv.py
index 8b536a7b..9e8515a4 100644
--- a/tablib/formats/_csv.py
+++ b/tablib/formats/_csv.py
@@ -13,8 +13,8 @@
 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()