Move file_types list to compat module.

2 files changed, 6 insertions, 9 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
diff --git a/sqlparse/lexer.py b/sqlparse/lexer.py
index 06318c6..e7996b2 100644
--- a/sqlparse/lexer.py
+++ b/sqlparse/lexer.py
@@ -12,20 +12,12 @@
 # It's separated from the rest of pygments to increase performance
 # and to allow some customizations.
 
-from io import TextIOBase
-
 from sqlparse import tokens
 from sqlparse.keywords import SQL_REGEX
-from sqlparse.compat import StringIO, string_types, u
+from sqlparse.compat import file_types, string_types, u
 from sqlparse.utils import consume
 
 
-try:
-    file_types = (file, StringIO, TextIOBase)
-except NameError:  # Python 3
-    file_types = (StringIO, TextIOBase)
-
-
 class Lexer(object):
     """Lexer
     Empty class. Leaving for backwards-compatibility