summaryrefslogtreecommitdiff
path: root/sqlparse/compat.py
blob: 84d0c965b47d5a04cc702f421da047ea199ecc8f (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 of
string types, buffers and metaclasses.

Parts of the code is copied directly from six:
https://bitbucket.org/gutworth/six
"""

import sys

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

if PY3:
    def u(s, encoding=None):
        return str(s)


    range = range
    text_type = str
    string_types = (str,)
    from io import StringIO


elif PY2:
    def u(s, encoding=None):
        encoding = encoding or 'unicode-escape'
        try:
            return unicode(s)
        except UnicodeDecodeError:
            return unicode(s, encoding)


    range = xrange
    text_type = unicode
    string_types = (basestring,)
    from StringIO import StringIO