46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright (C) 2009-2018 the sqlparse authors and contributors
|
|
# <see AUTHORS file>
|
|
#
|
|
# This module is part of python-sqlparse and is released under
|
|
# the BSD License: https://opensource.org/licenses/BSD-3-Clause
|
|
|
|
"""Python 2/3 compatibility.
|
|
|
|
This module only exists to avoid a dependency on six
|
|
for very trivial stuff. We only need to take care of
|
|
string types, buffers and metaclasses.
|
|
|
|
Parts of the code is copied directly from six:
|
|
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 unicode_compatible(cls):
|
|
return cls
|
|
|
|
text_type = str
|
|
string_types = (str,)
|
|
from io import StringIO
|
|
file_types = (StringIO, TextIOBase)
|
|
|
|
|
|
elif PY2:
|
|
def unicode_compatible(cls):
|
|
cls.__unicode__ = cls.__str__
|
|
cls.__str__ = lambda x: x.__unicode__().encode('utf-8')
|
|
return cls
|
|
|
|
text_type = unicode
|
|
string_types = (str, unicode,)
|
|
from StringIO import StringIO
|
|
file_types = (file, StringIO, TextIOBase)
|