diff options
Diffstat (limited to 'src/lib/Bcfg2/Compat.py')
-rw-r--r-- | src/lib/Bcfg2/Compat.py | 25 |
1 files changed, 7 insertions, 18 deletions
diff --git a/src/lib/Bcfg2/Compat.py b/src/lib/Bcfg2/Compat.py index 9a8d76c47..2e13301f8 100644 --- a/src/lib/Bcfg2/Compat.py +++ b/src/lib/Bcfg2/Compat.py @@ -88,6 +88,13 @@ def u_str(string, encoding=None): else: return unicode(string) + +def ensure_binary(string, encoding='utf-8'): + if type(string) == unicode: + return string.encode(encoding) + return string + + try: from functools import wraps except ImportError: @@ -95,24 +102,6 @@ except ImportError: """ implementation of functools.wraps() for python 2.4 """ return lambda f: f - -# base64 compat -if sys.hexversion >= 0x03000000: - from base64 import b64encode as _b64encode, b64decode as _b64decode - - @wraps(_b64encode) - def b64encode(val, **kwargs): # pylint: disable=C0111 - try: - return _b64encode(val, **kwargs) - except TypeError: - return _b64encode(val.encode('UTF-8'), **kwargs).decode('UTF-8') - - @wraps(_b64decode) - def b64decode(val, **kwargs): # pylint: disable=C0111 - return _b64decode(val.encode('UTF-8'), **kwargs).decode('UTF-8') -else: - from base64 import b64encode, b64decode - try: input = raw_input except NameError: |