Merge pull request #16572 from privateip/netcmd

adds new method to return specific response from command to netcmd
This commit is contained in:
Peter Sprygada 2016-07-04 22:34:50 -04:00 committed by GitHub
commit d9546d6367

View file

@ -163,9 +163,23 @@ class CommandRunner(object):
self.retries = 10 self.retries = 10
self.interval = 1 self.interval = 1
self._cache = dict()
def add_command(self, command, output=None): def add_command(self, command, output=None):
self.module.cli.add_commands(command, output=output) self.module.cli.add_commands(command, output=output)
def get_command(self, command):
try:
cmdobj = self._cache[command]
return cmdobj.response
except KeyError:
for cmd in self.module.cli.commands:
if str(cmd) == command:
self._cache[command] = cmd
return cmd.response
raise ValueError("command '%s' not found" % command)
def add_conditional(self, condition): def add_conditional(self, condition):
self.conditionals.add(Conditional(condition)) self.conditionals.add(Conditional(condition))