mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-05-08 14:22:46 +00:00
Make UnsafeProxy work on Python 3
Instead of str and unicode we have to deal with bytes and str.
This commit is contained in:
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user