Move errors from import to connect (#50034)

This commit is contained in:
Nathaniel Case 2018-12-17 12:33:44 -05:00 committed by GitHub
parent d4ee599fe9
commit bf4ad56479
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 5 deletions

View file

@ -138,10 +138,7 @@ try:
from napalm.base import ModuleImportError
HAS_NAPALM = True
except ImportError:
raise AnsibleError(
'Napalm is required to use the napalm connection type.\n'
'Please run pip install napalm'
)
HAS_NAPALM = False
display = Display()
@ -158,6 +155,11 @@ class Connection(NetworkConnectionBase):
self.napalm = None
def _connect(self):
if not HAS_NAPALM:
raise AnsibleError(
'Napalm is required to use the napalm connection type.\n'
'Please run pip install napalm'
)
super(Connection, self)._connect()
if not self.connected: