summaryrefslogtreecommitdiff
path: root/sqlparse/compat.py
blob: 1849c1370ff1a80710194b0898494e2e787b3610 (plain)
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
"""Python 2/3 compatibility.

This module only exists to avoid a dependency on six
for very trivial stuff. We only need to take care regarding
string types and buffers.
"""

import sys

PY2 = sys.version_info[0] == 2
PY3 = sys.version_info[0] == 3

if PY3:
    text_type = str
    string_types = (str,)
    from io import StringIO

    def u(s):
        return s

elif PY2:
    text_type = unicode
    string_types = (basestring,)
    from StringIO import StringIO  # flake8: noqa

    def u(s):
        return unicode(s, 'unicode_escape')


# Directly copied from six:
def with_metaclass(meta, *bases):
    """Create a base class with a metaclass."""
    # This requires a bit of explanation: the basic idea is to make a dummy
    # metaclass for one level of class instantiation that replaces itself with
    # the actual metaclass.
    class metaclass(meta):
        def __new__(cls, name, this_bases, d):
            return meta(name, bases, d)
    return type.__new__(metaclass, 'temporary_class', (), {})