Use BytesIO instead of StringIO for python2/3 compatibility

apache · Feb 4, 2017 · 461e41c · 461e41c
1 parent 7164061
commit 461e41c
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/superset/results_backends.py b/superset/results_backends.py
@@ -12,13 +12,9 @@
 except ImportError:
     import pickle
 
+import io
 import logging
 
-try:
-    import StringIO
-except ImportError:
-    import io as StringIO
-
 import boto3
 from werkzeug.contrib.cache import BaseCache
 
@@ -56,7 +52,7 @@ def get(self, key):
         if not self._key_exists(key):
             return None
         else:
-            value_file = StringIO.StringIO()
+            value_file = io.BytesIO()
 
             try:
                 self.s3_client.download_fileobj(
@@ -117,7 +113,7 @@ def set(self, key, value, timeout=None):
                   ``pickle.PickleError``.
         :rtype: boolean
         """
-        value_file = StringIO.StringIO()
+        value_file = io.BytesIO()
         pickle.dump(value, value_file)
 
         try: