Make UnsafeProxy work on Python 3

Instead of str and unicode we have to deal with bytes and str.
pull/12723/head
Marius Gedminas 9 years ago
parent af6a316d9b
commit e511d6d458

@ -56,20 +56,41 @@ __all__ = ['UnsafeProxy', 'AnsibleUnsafe', 'wrap_var']
class AnsibleUnsafe(object): class AnsibleUnsafe(object):
__UNSAFE__ = True __UNSAFE__ = True
class AnsibleUnsafeStr(str, AnsibleUnsafe): try:
pass unicode
except NameError:
# Python 3
class AnsibleUnsafeBytes(bytes, AnsibleUnsafe):
pass
class AnsibleUnsafeUnicode(unicode, AnsibleUnsafe): class AnsibleUnsafeStr(str, AnsibleUnsafe):
pass pass
class UnsafeProxy(object):
def __new__(cls, obj, *args, **kwargs):
if obj.__class__ == str:
return AnsibleUnsafeStr(obj)
elif obj.__class__ == bytes:
return AnsibleUnsafeBytes(obj)
else:
return obj
else:
# Python 2
class AnsibleUnsafeStr(str, AnsibleUnsafe):
pass
class AnsibleUnsafeUnicode(unicode, AnsibleUnsafe):
pass
class UnsafeProxy(object):
def __new__(cls, obj, *args, **kwargs):
if obj.__class__ == unicode:
return AnsibleUnsafeUnicode(obj)
elif obj.__class__ == str:
return AnsibleUnsafeStr(obj)
else:
return obj
class UnsafeProxy(object):
def __new__(cls, obj, *args, **kwargs):
if obj.__class__ == unicode:
return AnsibleUnsafeUnicode(obj)
elif obj.__class__ == str:
return AnsibleUnsafeStr(obj)
else:
return obj
def _wrap_dict(v): def _wrap_dict(v):
for k in v.keys(): for k in v.keys():
@ -77,12 +98,14 @@ def _wrap_dict(v):
v[k] = wrap_var(v[k]) v[k] = wrap_var(v[k])
return v return v
def _wrap_list(v): def _wrap_list(v):
for idx, item in enumerate(v): for idx, item in enumerate(v):
if item is not None: if item is not None:
v[idx] = wrap_var(item) v[idx] = wrap_var(item)
return v return v
def wrap_var(v): def wrap_var(v):
if isinstance(v, dict): if isinstance(v, dict):
v = _wrap_dict(v) v = _wrap_dict(v)

Loading…
Cancel
Save