Source code for oss2.compat

# -*- coding: utf-8 -*-

"""
Compatible Python versions
"""

import sys

is_py2 = (sys.version_info[0] == 2)
is_py3 = (sys.version_info[0] == 3)
is_py33 = (sys.version_info[0] == 3 and sys.version_info[1] == 3)


try:
    import simplejson as json
except (ImportError, SyntaxError):
    import json


if is_py2:
    from urllib import quote as urlquote, unquote as urlunquote
    from urlparse import urlparse


    def to_bytes(data):
        """Covert to UTF-8 encoding if the input is unicode; otherwise return the original data."""
        if isinstance(data, unicode):
            return data.encode('utf-8')
        else:
            return data

    def to_string(data):
        """Convert to str object"""
        return to_bytes(data)

    def to_unicode(data):
        """Convert the input to unicode if it's utf-8 bytes."""
        if isinstance(data, bytes):
            return data.decode('utf-8')
        else:
            return data

    def stringify(input):
        if isinstance(input, dict):
            return dict([(stringify(key), stringify(value)) for key,value in input.iteritems()])
        elif isinstance(input, list):
            return [stringify(element) for element in input]
        elif isinstance(input, unicode):
            return input.encode('utf-8')
        else:
            return input

    builtin_str = str
    bytes = str
    str = unicode


elif is_py3:
    from urllib.parse import quote as urlquote, unquote as urlunquote
    from urllib.parse import urlparse

[docs] def to_bytes(data): """Covert to UTF-8 encoding if the input is unicode; otherwise return the original data.""" if isinstance(data, str): return data.encode(encoding='utf-8') else: return data
[docs] def to_string(data): """Convert the input to unicode if it's UTF-8 bytes.""" if isinstance(data, bytes): return data.decode('utf-8') else: return data
[docs] def to_unicode(data): """Convert the input to unicode if it's UTF-8 bytes.""" return to_string(data)
[docs] def stringify(input): return input
builtin_str = str bytes = bytes str = str