diff --git a/hyperglass/execution/drivers/ssh_netmiko.py b/hyperglass/execution/drivers/ssh_netmiko.py index 6ed33cb..5e055b4 100644 --- a/hyperglass/execution/drivers/ssh_netmiko.py +++ b/hyperglass/execution/drivers/ssh_netmiko.py @@ -108,7 +108,7 @@ class NetmikoConnection(SSHConnection): log.error(str(scrape_error)) raise DeviceTimeout( params.messages.connection_error, - device_name=self.device.display_name, + device_name=self.device.name, proxy=None, error=params.messages.request_timeout, ) @@ -121,14 +121,14 @@ class NetmikoConnection(SSHConnection): raise AuthError( params.messages.connection_error, - device_name=self.device.display_name, + device_name=self.device.name, proxy=None, error=params.messages.authentication_error, ) if not responses: raise ScrapeError( params.messages.connection_error, - device_name=self.device.display_name, + device_name=self.device.name, proxy=None, error=params.messages.no_response, ) diff --git a/hyperglass/execution/drivers/ssh_scrapli.py b/hyperglass/execution/drivers/ssh_scrapli.py index eed0674..7a9761b 100644 --- a/hyperglass/execution/drivers/ssh_scrapli.py +++ b/hyperglass/execution/drivers/ssh_scrapli.py @@ -126,7 +126,7 @@ class ScrapliConnection(SSHConnection): log.error(err) raise DeviceTimeout( params.messages.connection_error, - device_name=self.device.display_name, + device_name=self.device.name, proxy=None, error=params.messages.request_timeout, ) @@ -139,7 +139,7 @@ class ScrapliConnection(SSHConnection): raise AuthError( params.messages.connection_error, - device_name=self.device.display_name, + device_name=self.device.name, proxy=None, error=params.messages.authentication_error, ) @@ -147,7 +147,7 @@ class ScrapliConnection(SSHConnection): log.error(err) raise ScrapeError( params.messages.connection_error, - device_name=self.device.display_name, + device_name=self.device.name, proxy=None, error=params.messages.no_response, ) @@ -155,7 +155,7 @@ class ScrapliConnection(SSHConnection): if not responses: raise ScrapeError( params.messages.connection_error, - device_name=self.device.display_name, + device_name=self.device.name, proxy=None, error=params.messages.no_response, )