diff options
Diffstat (limited to 'sqlparse/compat.py')
| -rw-r--r-- | sqlparse/compat.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/sqlparse/compat.py b/sqlparse/compat.py index a41b18b..d6a9144 100644 --- a/sqlparse/compat.py +++ b/sqlparse/compat.py @@ -16,10 +16,12 @@ https://bitbucket.org/gutworth/six """ import sys +from io import TextIOBase PY2 = sys.version_info[0] == 2 PY3 = sys.version_info[0] == 3 + if PY3: def u(s, encoding=None): return str(s) @@ -32,6 +34,7 @@ if PY3: text_type = str string_types = (str,) from io import StringIO + file_types = (StringIO, TextIOBase) elif PY2: @@ -52,3 +55,5 @@ elif PY2: text_type = unicode string_types = (str, unicode,) from StringIO import StringIO + file_types = (file, StringIO, TextIOBase) + from StringIO import StringIO |