compat.py 1.06 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
# -*- 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)