223 lines
8.2 KiB
Python
223 lines
8.2 KiB
Python
#!/usr/bin/python
|
|
|
|
# Copyright: (c) 2018, Terry Jones <terry.jones@example.org>
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
DOCUMENTATION = r'''
|
|
---
|
|
module: CreoleSet
|
|
|
|
short_description: This module aims to set variables on EOLE modules
|
|
|
|
version_added: "1.0.0"
|
|
|
|
description: This module aims to set variables on EOLE modules orderly and atomicly.
|
|
|
|
options:
|
|
variables:
|
|
description: Name of the variable to change.
|
|
required: true
|
|
type: list
|
|
state:
|
|
description: state of variables
|
|
required: false
|
|
type: str
|
|
author:
|
|
- Cadoles
|
|
'''
|
|
|
|
EXAMPLES = r'''
|
|
variables:
|
|
- name: activer_ajout_hosts
|
|
value: 'oui'
|
|
- name: adresse_ip_hosts
|
|
value:
|
|
- 192.168.1.1
|
|
- name: nom_long_hosts
|
|
value:
|
|
- "seth.domain.org cloud.domain.org"
|
|
state: exact
|
|
'''
|
|
|
|
RETURN = r'''
|
|
# These are examples of possible return values, and in general should use other names for return values.
|
|
diff:
|
|
description: variable modifications status.
|
|
type: list
|
|
returned: always
|
|
sample: ['activer_ajout_hosts: non => oui']
|
|
'''
|
|
|
|
import sys
|
|
from ansible.module_utils.basic import AnsibleModule
|
|
from creole.loader import creole_loader, config_save_values
|
|
|
|
def process_value(variable, value):
|
|
if variable.impl_is_multi():
|
|
if not isinstance(value, list):
|
|
return [value]
|
|
else:
|
|
if isinstance(value, list):
|
|
raise Exception('Variable {} is not multi'.format(variable))
|
|
return value
|
|
|
|
def yml_params_to_unicode(param):
|
|
def convert_param(param):
|
|
if isinstance(param, str):
|
|
return param.decode('utf-8')
|
|
if isinstance(param, list):
|
|
return [convert_param(p) for p in param]
|
|
if isinstance(param, dict):
|
|
return {convert_param(key): convert_param(value) for key,value in param.items()}
|
|
return param
|
|
return convert_param(param)
|
|
|
|
|
|
|
|
def run_module():
|
|
# define available arguments/parameters a user can pass to the module
|
|
module_args = dict(
|
|
variables=dict(type='list', required=True),
|
|
state=dict(type='str', required=False, default='exact'),
|
|
)
|
|
|
|
# seed the result dict in the object
|
|
# we primarily care about changed and state
|
|
# changed is if this module effectively modified the target
|
|
# state will include any data that you want your module to pass back
|
|
# for consumption, for example, in a subsequent task
|
|
result = dict(
|
|
changed=False,
|
|
diff=[],
|
|
debug=[],
|
|
)
|
|
|
|
# the AnsibleModule object will be our abstraction working with Ansible
|
|
# this includes instantiation, a couple of common attr would be the
|
|
# args/params passed to the execution, as well as if the module
|
|
# supports check mode
|
|
module = AnsibleModule(
|
|
argument_spec=module_args,
|
|
supports_check_mode=True
|
|
)
|
|
|
|
# if the user is working with this module in only check mode we do not
|
|
# want to make any changes to the environment, just return the current
|
|
# state with no modifications
|
|
c = creole_loader(rw=True, reload_config=True)
|
|
d = c.cfgimpl_get_description()
|
|
diff = []
|
|
c.cfgimpl_get_settings().remove('disabled')
|
|
if sys.version_info < (3,):
|
|
module.params['variables'] = yml_params_to_unicode(module.params['variables'])
|
|
|
|
variables = {}
|
|
for variable in module.params['variables']:
|
|
creole_variable = c.find_first(byname=variable['name'])
|
|
variables[creole_variable] = process_value(creole_variable, variable['value'])
|
|
|
|
slaves = [v for v in variables if v.impl_is_master_slaves('slave')]
|
|
|
|
masters = {v: [] for v in variables if v.impl_is_master_slaves('master')}
|
|
|
|
for slave in slaves:
|
|
master = slave.impl_get_master_slaves().getmaster(slave)
|
|
masters[master].append(slave)
|
|
|
|
ordered_variables = [(v, []) for v in variables if not v.impl_is_master_slaves()]
|
|
|
|
ordered_variables.extend(masters.items())
|
|
|
|
for variable, sub_variables in ordered_variables:
|
|
new_value = variables[variable]
|
|
var_path = d.impl_get_path_by_opt(variable)
|
|
old_value = c.getattr(var_path)
|
|
new_value_set = {var_path: {'value': new_value, 'sub': [(d.impl_get_path_by_opt(sub), variables[sub]) for sub in sub_variables]}}
|
|
|
|
if variable.impl_is_multi():
|
|
method = module.params['state']
|
|
if method == 'exact':
|
|
if old_value != new_value:
|
|
diff.append('{}: {} => {}'.format(var_path, old_value, new_value))
|
|
if sub_variables:
|
|
sub_old_values = []
|
|
for sub_variable in sub_variables:
|
|
sub_value = variables[sub_variable]
|
|
sub_var_path = d.impl_get_path_by_opt(sub_variable)
|
|
sub_old_value = c.getattr(sub_var_path)
|
|
sub_old_values.append(sub_old_value)
|
|
if sub_old_value != sub_value:
|
|
diff.append('{}: {} => {}'.format(sub_var_path, sub_old_value, sub_value))
|
|
homeconfig, name = c.cfgimpl_get_home_by_path(var_path)
|
|
homeconfig.__delattr__(name)
|
|
c.cfgimpl_get_settings().remove('validator')
|
|
c.setattr(var_path, new_value)
|
|
for sub_variable, sub_old_value in zip(sub_variables, sub_old_values):
|
|
sub_value = variables[sub_variable]
|
|
sub_var_path = d.impl_get_path_by_opt(sub_variable)
|
|
if sub_old_value != sub_value:
|
|
for i in range(len(new_value)):
|
|
c.getattr(sub_var_path)[i] = sub_value[i]
|
|
c.cfgimpl_get_settings().append('validator')
|
|
else:
|
|
c.setattr(var_path, new_value)
|
|
|
|
elif method == 'present':
|
|
# is master value present in old values?
|
|
indexes = {}
|
|
for nv in new_value:
|
|
indexes[nv] = []
|
|
for index, ov in enumerate(old_value):
|
|
if nv == ov:
|
|
indexes[nv].append(index)
|
|
|
|
for nv in indexes:
|
|
nv_index = new_value_set[var_path]['value'].index(nv)
|
|
new_value_subset = {so: sv[nv_index] for so,sv in new_value_set[var_path]['sub']}
|
|
is_present = False
|
|
for index in indexes[nv]:
|
|
old_value_set = {d.impl_get_path_by_opt(sv): c.getattr(d.impl_get_path_by_opt(sv))[index] for sv in sub_variables}
|
|
if old_value_set == new_value_subset:
|
|
is_present = True
|
|
break
|
|
|
|
if not is_present:
|
|
old_value.append(nv)
|
|
diff.append('{}: {} => {}'.format(var_path, old_value[:-1], old_value))
|
|
for sub_variable, sub_value in new_value_set[var_path]['sub']:
|
|
c.getattr(sub_variable)[-1] = sub_value[nv_index]
|
|
diff.append('{}: {} => {}'.format(sub_variable, c.getattr(sub_variable)[:-1], c.getattr(sub_variable)))
|
|
|
|
else:
|
|
if old_value != new_value:
|
|
diff.append('{}: {} => {}'.format(var_path, old_value, new_value))
|
|
c.setattr(var_path, new_value)
|
|
for sub_variable in sub_variables:
|
|
sub_value = variables[sub_variable]
|
|
sub_var_path = d.impl_get_path_by_opt(sub_variable)
|
|
sub_old_value = c.getattr(sub_var_path)
|
|
if sub_old_value != sub_value:
|
|
diff.append('{}: {} => {}'.format(sub_var_path, sub_old_value, sub_value))
|
|
c.setattr(sub_var_path, sub_value)
|
|
|
|
|
|
result['diff'] = diff
|
|
c.cfgimpl_get_settings().append('disabled')
|
|
c.make_dict()
|
|
|
|
if not module.check_mode and diff:
|
|
config_save_values(c, 'creole')
|
|
result['changed'] = True
|
|
|
|
module.exit_json(**result)
|
|
|
|
|
|
def main():
|
|
run_module()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|