fix pim rp_address issues (#35248)

This commit is contained in:
saichint 2018-01-23 22:18:49 -08:00 committed by Trishna Guha
commit 0b30c42902
2 changed files with 200 additions and 66 deletions

View file

@ -94,7 +94,7 @@ from ansible.module_utils.basic import AnsibleModule
from ansible.module_utils.network.common.config import CustomNetworkConfig
def get_existing(module, args):
def get_existing(module, args, gl):
existing = {}
config = str(get_config(module))
address = module.params['rp_address']
@ -105,6 +105,11 @@ def get_existing(module, args):
values = line.split()
if values[0] != address:
continue
if gl and 'group-list' not in line:
continue
elif not gl and 'group-list' in line:
if '224.0.0.0/4' not in line: # ignore default group-list
continue
existing['bidir'] = existing.get('bidir') or 'bidir' in line
if len(values) > 2:
@ -114,7 +119,8 @@ def get_existing(module, args):
elif values[1] == 'prefix-list':
existing['prefix_list'] = value
elif values[1] == 'group-list':
existing['group_list'] = value
if value != '224.0.0.0/4': # ignore default group-list
existing['group_list'] = value
return existing
@ -122,6 +128,14 @@ def get_existing(module, args):
def state_present(module, existing, proposed, candidate):
address = module.params['rp_address']
command = 'ip pim rp-address {0}'.format(address)
if module.params['group_list'] and not proposed.get('group_list'):
command += ' group-list ' + module.params['group_list']
if module.params['prefix_list']:
if not proposed.get('prefix_list'):
command += ' prefix-list ' + module.params['prefix_list']
if module.params['route_map']:
if not proposed.get('route_map'):
command += ' route-map ' + module.params['route_map']
commands = build_command(proposed, command)
if commands:
candidate.add(commands, parents=[])
@ -140,13 +154,31 @@ def build_command(param_dict, command):
def state_absent(module, existing, candidate):
address = module.params['rp_address']
commands = []
command = 'no ip pim rp-address {0}'.format(address)
if existing.get('group_list'):
if module.params['group_list'] == existing.get('group_list'):
commands = build_command(existing, command)
else:
elif not module.params['group_list']:
commands = [command]
candidate.add(commands, parents=[])
if commands:
candidate.add(commands, parents=[])
def get_proposed(pargs, existing):
proposed = {}
for key, value in pargs.items():
if key != 'rp_address':
if str(value).lower() == 'true':
value = True
elif str(value).lower() == 'false':
value = False
if existing.get(key) != value:
proposed[key] = value
return proposed
def main():
@ -180,20 +212,16 @@ def main():
'bidir'
]
existing = get_existing(module, args)
proposed_args = dict((k, v) for k, v in module.params.items()
if v is not None and k in args)
proposed = {}
for key, value in proposed_args.items():
if key != 'rp_address':
if str(value).lower() == 'true':
value = True
elif str(value).lower() == 'false':
value = False
if module.params['group_list']:
existing = get_existing(module, args, True)
proposed = get_proposed(proposed_args, existing)
if existing.get(key) != value:
proposed[key] = value
else:
existing = get_existing(module, args, False)
proposed = get_proposed(proposed_args, existing)
candidate = CustomNetworkConfig(indent=3)
if state == 'present' and (proposed or not existing):
@ -205,7 +233,19 @@ def main():
candidate = candidate.items_text()
result['commands'] = candidate
result['changed'] = True
load_config(module, candidate)
msgs = load_config(module, candidate, True)
if msgs:
for item in msgs:
if item:
if isinstance(item, dict):
err_str = item['clierror']
else:
err_str = item
if 'No policy was configured' in err_str:
if state == 'absent':
addr = module.params['rp_address']
new_cmd = 'no ip pim rp-address {0}'.format(addr)
load_config(module, new_cmd)
module.exit_json(**result)