ansible/community.general
1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2024-09-14 20:13:21 +02:00
Code Releases Activity

Merge pull request #16588 from privateip/vyos

Browse source 
bug fixes in vyos shared module
This commit is contained in:
Peter Sprygada 2016-07-05 21:15:57 -04:00 committed by GitHub
commit c9dccd3566
1 changed files with 18 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
lib/ansible/module_utils/vyos.py
View file

@ -23,6 +23,7 @@ import re
from ansible.module_utils.network import NetworkError, get_module, get_exception from ansible.module_utils.network import NetworkError, get_module, get_exception
from ansible.module_utils.network import register_transport, to_list from ansible.module_utils.network import register_transport, to_list
from ansible.module_utils.network import Command, NetCli from ansible.module_utils.network import Command, NetCli
from ansible.module_utils.netcfg import NetworkConfig
from ansible.module_utils.shell import Shell, ShellError, HAS_PARAMIKO from ansible.module_utils.shell import Shell, ShellError, HAS_PARAMIKO
DEFAULT_COMMENT = 'configured by vyos_config' DEFAULT_COMMENT = 'configured by vyos_config'
@ -36,17 +37,18 @@ def argument_spec():
vyos_argument_spec = argument_spec() vyos_argument_spec = argument_spec()
def get_config(module): def get_config(module):
if module.params['config']: contents = module.params['config']
return module.params['config'] if not contents:
config = module.config.get_config() contents = str(module.config.get_config()).split('\n')
module.params['config'] = config module.params['config'] = contents
return config contents = '\n'.join(contents)
return NetworkConfig(contents=contents, device_os='junos')
def diff_config(candidate, config): def diff_config(candidate, config):
updates = set() updates = set()
config = [str(c).replace("'", '') for c in config] config = [str(c).replace("'", '') for c in str(config).split('\n')]
for line in candidate: for line in str(candidate).split('\n'):
item = str(line).replace("'", '') item = str(line).replace("'", '')
if not item.startswith('set') and not item.startswith('delete'): if not item.startswith('set') and not item.startswith('delete'):
@ -66,8 +68,9 @@ def diff_config(candidate, config):
return list(updates) return list(updates)
def load_config(module, candidate): def load_candidate(module, candidate):
config = get_config(module).split('\n') config = get_config(module)
updates = diff_config(candidate, config) updates = diff_config(candidate, config)
comment = module.params['comment'] comment = module.params['comment']
@ -81,7 +84,6 @@ def load_config(module, candidate):
result['diff'] = dict(prepared=diff) result['diff'] = dict(prepared=diff)
result['changed'] = True result['changed'] = True
result['updates'] = updates
if not module.check_mode: if not module.check_mode:
module.config.commit_config(comment=comment) module.config.commit_config(comment=comment)
@ -93,8 +95,14 @@ def load_config(module, candidate):
# exit from config mode # exit from config mode
module.cli('exit') module.cli('exit')
result['updates'] = updates
return result return result
def load_config(module, commands):
contents = '\n'.join(commands)
candidate = NetworkConfig(contents=contents, device_os='junos')
return load_candidate(module, candidate)
class Cli(NetCli): class Cli(NetCli):