handle network update

netbox_agent/network.py

@@ -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
+
+    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:
-                # FIXME: implement update
+                        if netbox_ip.device != device:
-                # update name or ip
+                            netbox_ip.device = device
-                # see https://github.com/Solvik/netbox_agent/issues/9
+                            netbox_ip.save()
-                pass
+            if nic_update:
+                interface.save()

netbox_agent/server.py

@@ -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()