Merge pull request #17254 from privateip/netcli

adds new property command_string to Command object
This commit is contained in:
Peter Sprygada 2016-08-25 22:51:02 -04:00 committed by GitHub
commit fb9b98a543

View file

@ -102,6 +102,7 @@ class Command(object):
self.command = command
self.output = output
self.command_string = command
self.prompt = prompt
self.response = response
@ -110,7 +111,7 @@ class Command(object):
self.delay = delay
def __str__(self):
return self.command
return self.command_string
class CommandRunner(object):
@ -145,7 +146,7 @@ class CommandRunner(object):
return cmdobj.response
except KeyError:
for cmd in self.commands:
if str(cmd) == command and cmd.output == output:
if cmd.command == command and cmd.output == output:
self._cache[(command, output)] = cmd
return cmd.response
raise ValueError("command '%s' not found" % command)