iot support Python 3

[rainbowstream.git] / rainbowstream / py3patch.py

diff --git a/rainbowstream/py3patch.py b/rainbowstream/py3patch.py
index e1eec6e2ef70fc2827a622af7afc0689f75caa64..543b9a1462a27ef2fb5cb7100058159654b32ea4 100644 (file)
--- a/rainbowstream/py3patch.py
+++ b/rainbowstream/py3patch.py
@@ -5,9 +5,9 @@ import sys
 
 # StringIO module
 try:
-    from StringIO import StringIO
+    from StringIO import StringIO, BytesIO
 except:
-    from io import StringIO
+    from io import StringIO, BytesIO
 
 # raw_input and map functiion behaviour
 if sys.version[0]=="3":