mirror of
https://github.com/thegeeklab/ansible-later.git
synced 2024-11-14 01:00:39 +00:00
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
# Copyright: (c) 2018, Ansible Project
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
|
|
# Make coding more python3-ish
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
import json
|
|
|
|
from collections import Mapping
|
|
from datetime import date, datetime
|
|
|
|
from ansible.module_utils._text import to_text
|
|
from ansible.parsing.yaml.objects import AnsibleVaultEncryptedUnicode
|
|
from ansible.utils.unsafe_proxy import AnsibleUnsafe, wrap_var
|
|
from ansible.parsing.vault import VaultLib
|
|
|
|
|
|
class AnsibleJSONDecoder(json.JSONDecoder):
|
|
|
|
_vaults = {}
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
kwargs['object_hook'] = self.object_hook
|
|
super(AnsibleJSONDecoder, self).__init__(*args, **kwargs)
|
|
|
|
@classmethod
|
|
def set_secrets(cls, secrets):
|
|
cls._vaults['default'] = VaultLib(secrets=secrets)
|
|
|
|
def object_hook(self, pairs):
|
|
for key in pairs:
|
|
value = pairs[key]
|
|
|
|
if key == '__ansible_vault':
|
|
value = AnsibleVaultEncryptedUnicode(value)
|
|
if self._vaults:
|
|
value.vault = self._vaults['default']
|
|
return value
|
|
elif key == '__ansible_unsafe':
|
|
return wrap_var(value.get('__ansible_unsafe'))
|
|
|
|
return pairs
|
|
|
|
|
|
# TODO: find way to integrate with the encoding modules do in module_utils
|
|
class AnsibleJSONEncoder(json.JSONEncoder):
|
|
'''
|
|
Simple encoder class to deal with JSON encoding of Ansible internal types
|
|
'''
|
|
def default(self, o):
|
|
if isinstance(o, AnsibleVaultEncryptedUnicode):
|
|
# vault object
|
|
value = {'__ansible_vault': to_text(o._ciphertext, errors='surrogate_or_strict', nonstring='strict')}
|
|
elif isinstance(o, AnsibleUnsafe):
|
|
# unsafe object
|
|
value = {'__ansible_unsafe': to_text(o, errors='surrogate_or_strict', nonstring='strict')}
|
|
elif isinstance(o, Mapping):
|
|
# hostvars and other objects
|
|
value = dict(o)
|
|
elif isinstance(o, (date, datetime)):
|
|
# date object
|
|
value = o.isoformat()
|
|
else:
|
|
# use default encoder
|
|
value = super(AnsibleJSONEncoder, self).default(o)
|
|
return value
|