NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-24 17:38:37 -06:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Merge pull request #4844 from jvanderaa/napalm_lookup_hostname

Browse Source 
Adds name lookup to NAPALM if no primary IP address exists for device
This commit is contained in:
Jeremy Stretch 2020-07-13 13:32:32 -04:00 committed by GitHub
commit df20abf283
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,3 +1,4 @@
import socket
from collections import OrderedDict from collections import OrderedDict
from django.conf import settings from django.conf import settings
@ -371,8 +372,6 @@ class DeviceViewSet(CustomFieldModelViewSet):
Execute a NAPALM method on a Device Execute a NAPALM method on a Device
""" """
device = get_object_or_404(Device, pk=pk) device = get_object_or_404(Device, pk=pk)
if not device.primary_ip:
raise ServiceUnavailable("This device does not have a primary IP address configured.")
if device.platform is None: if device.platform is None:
raise ServiceUnavailable("No platform is configured for this device.") raise ServiceUnavailable("No platform is configured for this device.")
if not device.platform.napalm_driver: if not device.platform.napalm_driver:
@ -380,6 +379,22 @@ class DeviceViewSet(CustomFieldModelViewSet):
device.platform device.platform
)) ))
# Check for primary IP address from NetBox object
if device.primary_ip:
host = str(device.primary_ip.address.ip)
else:
# Raise exception for no IP address and no Name if device.name does not exist
if not device.name:
raise ServiceUnavailable(
"This device does not have a primary IP address or device name to lookup configured.")
try:
# Attempt to complete a DNS name resolution if no primary_ip is set
host = socket.gethostbyname(device.name)
except socket.gaierror:
# Name lookup failure
raise ServiceUnavailable(
f"Name lookup failure, unable to resolve IP address for {device.name}. Please set Primary IP or setup name resolution.")
# Check that NAPALM is installed # Check that NAPALM is installed
try: try:
import napalm import napalm
@ -399,10 +414,8 @@ class DeviceViewSet(CustomFieldModelViewSet):
if not request.user.has_perm('dcim.napalm_read'): if not request.user.has_perm('dcim.napalm_read'):
return HttpResponseForbidden() return HttpResponseForbidden()
# Connect to the device
napalm_methods = request.GET.getlist('method') napalm_methods = request.GET.getlist('method')
response = OrderedDict([(m, None) for m in napalm_methods]) response = OrderedDict([(m, None) for m in napalm_methods])
ip_address = str(device.primary_ip.address.ip)
username = settings.NAPALM_USERNAME username = settings.NAPALM_USERNAME
password = settings.NAPALM_PASSWORD password = settings.NAPALM_PASSWORD
optional_args = settings.NAPALM_ARGS.copy() optional_args = settings.NAPALM_ARGS.copy()
@ -422,8 +435,9 @@ class DeviceViewSet(CustomFieldModelViewSet):
elif key: elif key:
optional_args[key.lower()] = request.headers[header] optional_args[key.lower()] = request.headers[header]
# Connect to the device
d = driver( d = driver(
hostname=ip_address, hostname=host,
username=username, username=username,
password=password, password=password,
timeout=settings.NAPALM_TIMEOUT, timeout=settings.NAPALM_TIMEOUT,
@ -432,7 +446,7 @@ class DeviceViewSet(CustomFieldModelViewSet):
try: try:
d.open() d.open()
except Exception as e: except Exception as e:
raise ServiceUnavailable("Error connecting to the device at {}: {}".format(ip_address, e)) raise ServiceUnavailable("Error connecting to the device at {}: {}".format(host, e))
# Validate and execute each specified NAPALM method # Validate and execute each specified NAPALM method
for method in napalm_methods: for method in napalm_methods: