DGNum/netbox-agent
6
0
Fork 0
Code Issues 28 Pull requests 9 Projects Releases 14 Packages Wiki Activity Actions

handle network update

Browse source
This commit is contained in:
Solvik Blum 2019-08-05 16:51:01 +02:00
parent e80d8b941c
commit 165f911631
No known key found for this signature in database
GPG key ID: CC12B3DC262B6C47
2 changed files with 48 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

53
netbox_agent/network.py
View file

@ -1,5 +1,6 @@
import os
import re
from itertools import chain
from netaddr import IPAddress
import netifaces
@ -86,7 +87,7 @@ class Network():
type=self.get_netbox_type_for_nic(nic),
)
def update_netbox_network_cards(self):
def create_netbox_network_cards(self):
device = self.server.get_netbox_server()
for nic in self.nics:
interface = nb.dcim.interfaces.get(
@ -112,9 +113,47 @@ class Network():
interface=new_interface.id,
status=1,
)
# or we check if it needs update
else:
# FIXME: implement update
# update name or ip
# see https://github.com/Solvik/netbox_agent/issues/9
pass
def update_netbox_network_cards(self):
device = self.server.get_netbox_server()
# delete IP on netbox that are not known on this server
netbox_ips = nb.ipam.ip_addresses.filter(
device=device
)
all_local_ips = list(chain.from_iterable([x['ip'] for x in self.nics if x['ip'] != None]))
for netbox_ip in netbox_ips:
if netbox_ip.address not in all_local_ips:
netbox_ip.interface = None
netbox_ip.save()
# update each nic
for nic in self.nics:
interface = nb.dcim.interfaces.get(
mac_address=nic['mac'],
)
nic_update = False
if nic['name'] != interface.name:
nic_update = True
interface.name = nic['name']
if nic['ip']:
# sync local IPs
for ip in nic['ip']:
netbox_ip = nb.ipam.ip_addresses.get(
address=ip,
)
if not netbox_ip:
# create netbbox_ip on device
netbox_ip = nb.ipam.ip_addresses.create(
address=ip,
interface=interface.id,
status=1,
)
else:
if netbox_ip.device != device:
netbox_ip.device = device
netbox_ip.save()
if nic_update:
interface.save()

5
netbox_agent/server.py
View file

@ -152,7 +152,7 @@ class ServerBase():
if not server:
self._netbox_create_server()
self.network.update_netbox_network_cards()
self.network.create_netbox_network_cards()
def netbox_update(self):
"""
@ -192,7 +192,6 @@ class ServerBase():
if move_device_bay or device_bay.name != 'Blade {}'.format(self.get_blade_slot()):
device_bay.installed_device = None
device_bay.save()
# Find the slot and update it with our blade
device_bays = nb.dcim.device_bays.filter(
device_id=chassis.id,
@ -209,7 +208,7 @@ class ServerBase():
update = True
server.hostname = self.get_hostname()
# check network cards
#self.network.update_netbox_network_cards()
self.network.update_netbox_network_cards()
if update:
server.save()