From 527e4196b261bc013d4cab08150853169136c0cf Mon Sep 17 00:00:00 2001 From: Peter Sprygada Date: Mon, 4 Jul 2016 22:12:43 -0400 Subject: [PATCH] adds new method to return specific response from command to netcmd This adds a new method that will return the output from a specified command that has already been excuted by the CommandRunner. The new method, get_command takes a single argument which is the full name of the command to retrieve. --- lib/ansible/module_utils/netcmd.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/lib/ansible/module_utils/netcmd.py b/lib/ansible/module_utils/netcmd.py index c5eb197d13..651c437dd0 100644 --- a/lib/ansible/module_utils/netcmd.py +++ b/lib/ansible/module_utils/netcmd.py @@ -163,9 +163,23 @@ class CommandRunner(object): self.retries = 10 self.interval = 1 + self._cache = dict() + def add_command(self, command, output=None): 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): self.conditionals.add(Conditional(condition))