nextcloud.compat

Complexity

Total Complexity

8

Size/Duplication

Total Lines	42
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	encode_requests_password()	0	21	5
A	encode_string()	0	11	3

# -*- coding: utf-8 -*-
"""
Tools for python2/3 unicode compatibility
"""
import six


def encode_requests_password(word):
    """
    Convert the string to bytes (readable by the server)

    :param word: input string
    :returns:    bytes with appropriate encoding
    """
    if isinstance(word, bytes):
        return word

    ret = word
    if six.PY2:
        if isinstance(word, six.text_type):
            # trick to work with tricks in requests lib
            ret = word.encode('utf-8').decode('latin-1')
    else:
        try:
            ret = bytes(word, 'ascii')
        except UnicodeEncodeError:
            ret = bytes(word, 'utf-8')
    return ret


def encode_string(string):
    """Encodes a unicode instance to utf-8. If a str is passed it will
    simply be returned

    :param string: str or unicode to encode
    :returns     : encoded output as str
    """
    if six.PY2:
        if isinstance(string, six.text_type):
            return string.encode('utf-8')
    return string