Fixed the import StringIO issue.

demidovakatya · Apr 22, 2014 · c383374 · c383374
1 parent a102844
commit c383374
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/sqlparse/lexer.py b/sqlparse/lexer.py
@@ -16,7 +16,15 @@
 
 from sqlparse import tokens
 from sqlparse.keywords import KEYWORDS, KEYWORDS_COMMON
-from io import StringIO
+
+try:
+    # Python 3
+    from io import BytesIO as StringIO
+except (ImportError):
+    # Python 2
+    from StringIO import StringIO
+
+
 import collections