77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright 2020 Red Hat
|
|
# GNU General Public License v3.0+
|
|
# (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
|
|
|
|
"""
|
|
flatten a complex object to dot bracket notation
|
|
"""
|
|
from __future__ import absolute_import, division, print_function
|
|
|
|
__metaclass__ = type
|
|
|
|
import re
|
|
from ansible.module_utils.common._collections_compat import (
|
|
Mapping,
|
|
MutableMapping,
|
|
)
|
|
|
|
# Note, this file can only be used on the control node
|
|
# where ansible is installed
|
|
# limit imports to filter and lookup plugins
|
|
try:
|
|
from ansible.errors import AnsibleError
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
def get_path(var, path, environment, wantlist=False):
|
|
"""Get the value of a path within an object
|
|
|
|
:param var: The var from which the value is retrieved
|
|
:type var: should be dict or list, but jinja can sort that out
|
|
:param path: The path to get
|
|
:type path: should be a string but jinja can sort that out
|
|
:param environment: The jinja Environment
|
|
:type environment: Environment
|
|
:return: The result of the jinja evaluation
|
|
:rtype: any
|
|
"""
|
|
string_to_variable = "{{ %s }}" % path
|
|
result = environment.from_string(string_to_variable).render(**var)
|
|
if wantlist:
|
|
return list(result)
|
|
return result
|
|
|
|
|
|
def to_paths(var, prepend=False, wantlist=False):
|
|
if prepend:
|
|
if not isinstance(prepend, str):
|
|
raise AnsibleError("The value of 'prepend' must be a string.")
|
|
var = {prepend: var}
|
|
|
|
out = {}
|
|
|
|
def flatten(data, name=""):
|
|
if isinstance(data, (dict, Mapping, MutableMapping)):
|
|
for key, val in data.items():
|
|
if name:
|
|
if re.match("^[a-zA-Z_][a-zA-Z0-9_]*$", key):
|
|
nname = name + ".{key}".format(key=key)
|
|
else:
|
|
nname = name + "['{key}']".format(key=key)
|
|
else:
|
|
nname = key
|
|
flatten(val, nname)
|
|
elif isinstance(data, list):
|
|
for idx, val in enumerate(data):
|
|
flatten(val, "{name}[{idx}]".format(name=name, idx=idx))
|
|
else:
|
|
out[name] = data
|
|
|
|
flatten(var)
|
|
if wantlist:
|
|
return [out]
|
|
return out
|