mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Fixes #1696: Fix for NAPALM v2.0+

Browse Source
This commit is contained in:
Jeremy Stretch
2017-11-08 09:51:37 -05:00
parent 00986fd7bf
commit 9d50b78b69
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
netbox/dcim/api/views.py
View File

@ -258,12 +258,19 @@ class DeviceViewSet(WritableSerializerMixin, CustomFieldModelViewSet):
device.platform device.platform
)) ))
# Check that NAPALM is installed and verify the configured driver # Check that NAPALM is installed
try: try:
import napalm import napalm
from napalm_base.exceptions import ConnectAuthError, ModuleImportError
except ImportError: except ImportError:
raise ServiceUnavailable("NAPALM is not installed. Please see the documentation for instructions.") raise ServiceUnavailable("NAPALM is not installed. Please see the documentation for instructions.")
# TODO: Remove support for NAPALM < 2.0
try:
from napalm.base.exceptions import ConnectAuthError, ModuleImportError
except ImportError:
from napalm_base.exceptions import ConnectAuthError, ModuleImportError
# Validate the configured driver
try: try:
driver = napalm.get_network_driver(device.platform.napalm_driver) driver = napalm.get_network_driver(device.platform.napalm_driver)
except ModuleImportError: except ModuleImportError: