Merge pull request #3010 from grybak-arista/exit-filter

Filter exit commands when no other commands are included
This commit is contained in:
James Cammarata 2016-03-17 13:11:30 -04:00
commit d7422c02ad

View file

@ -117,6 +117,8 @@ responses:
sample: ['...', '...'] sample: ['...', '...']
""" """
import re
def compare(this, other): def compare(this, other):
parents = [item.text for item in this.parents] parents = [item.text for item in this.parents]
for entry in other: for entry in other:
@ -199,6 +201,30 @@ def main():
commands = flatten(commands, list()) commands = flatten(commands, list())
# Filter out configuration mode commands followed immediately by an
# exit command indented by one level only, e.g.
# - route-map map01 permit 10
# - exit
#
# Build a temporary list as we filter, then copy the temp list
# back onto the commands list.
temp = []
ind_prev = 999
count = 0
for c in commands:
ind_this = c.count(' ')
if re.search(r"^\s*exit$", c) and ind_this == ind_prev + 1:
temp.pop()
count -= 1
if count != 0:
ind_prev = temp[-1].count(' ')
continue
temp.append(c)
ind_prev = ind_this
count += 1
commands = temp
if commands: if commands:
if not module.check_mode: if not module.check_mode:
commands = [str(c).strip() for c in commands] commands = [str(c).strip() for c in commands]
@ -219,4 +245,3 @@ from ansible.module_utils.netcfg import *
from ansible.module_utils.eos import * from ansible.module_utils.eos import *
if __name__ == '__main__': if __name__ == '__main__':
main() main()