Skip to content
Snippets Groups Projects
Commit a5596463 authored by Neda Moeini's avatar Neda Moeini
Browse files

Updated delete Netbox device functionality to delete the device if it exists...

Updated delete Netbox device functionality to delete the device if it exists and don't raise AttributeError.
parent 71e75ec9
No related branches found
No related tags found
1 merge request!190Updated delete Netbox device functionality to delete the device if it exists...
Pipeline #86095 failed
...@@ -125,9 +125,8 @@ class NetboxClient: ...@@ -125,9 +125,8 @@ class NetboxClient:
def delete_interface(self, device_name: str, iface_name: str) -> None: def delete_interface(self, device_name: str, iface_name: str) -> None:
"""Delete an interface from a device by name.""" """Delete an interface from a device by name."""
interface = self.get_interface_by_name_and_device(iface_name, device_name) interface = self.get_interface_by_name_and_device(iface_name, device_name)
if interface: if interface is not None:
return interface.delete() return interface.delete()
return None
def create_device_type(self, manufacturer: str, model: str, slug: str) -> DeviceTypes: def create_device_type(self, manufacturer: str, model: str, slug: str) -> DeviceTypes:
"""Create a new device type in Netbox.""" """Create a new device type in Netbox."""
...@@ -199,8 +198,11 @@ class NetboxClient: ...@@ -199,8 +198,11 @@ class NetboxClient:
return device return device
def delete_device(self, device_name: str) -> None: def delete_device(self, device_name: str) -> None:
"""Delete device by name.""" """Delete device by name if exists."""
self.netbox.dcim.devices.get(name=device_name).delete() try:
self.netbox.dcim.devices.get(name=device_name).delete()
except AttributeError:
pass
def attach_interface_to_lag( def attach_interface_to_lag(
self, self,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment