From a2b72339c7bf6d15a2e6bd7a12098f7bcc9ac891 Mon Sep 17 00:00:00 2001 From: Cyril Levis Date: Sat, 6 Jun 2020 18:49:28 +0200 Subject: [PATCH] Some fixes that prevent crash on register and update --- netbox_agent/network.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/netbox_agent/network.py b/netbox_agent/network.py index c635236..8659ee2 100644 --- a/netbox_agent/network.py +++ b/netbox_agent/network.py @@ -438,8 +438,8 @@ class ServerNetwork(Network): self.server = server self.device = self.server.get_netbox_server() self.nb_net = nb.dcim - self.custom_arg = {'device': self.device.id} - self.custom_arg_id = {'device_id': self.device.id} + self.custom_arg = {'device': getattr(self.device, "id", None)} + self.custom_arg_id = {'device_id': getattr(self.device, "id", None)} def get_network_type(self): return 'server' @@ -558,8 +558,8 @@ class VirtualNetwork(Network): self.server = server self.device = self.server.get_netbox_vm() self.nb_net = nb.virtualization - self.custom_arg = {'virtual_machine': self.device.id} - self.custom_arg_id = {'virtual_machine_id': self.device.id} + self.custom_arg = {'virtual_machine': getattr(self.device, "id", None)} + self.custom_arg_id = {'virtual_machine_id': getattr(self.device, "id", None)} dcim_c = nb.virtualization.interfaces.choices() for _choice_type in dcim_c: -- 2.48.1