diff --git a/gso/services/netbox_client.py b/gso/services/netbox_client.py index f22bd4319ee5c60040263eaee02b6a3d866bf956..a66aae8692b91249a1c460b5aa5d120c74df6237 100644 --- a/gso/services/netbox_client.py +++ b/gso/services/netbox_client.py @@ -3,7 +3,6 @@ from uuid import UUID import pydantic import pynetbox -from pynetbox import RequestError from pynetbox.models.dcim import Devices, DeviceTypes, Interfaces from gso.products import Router @@ -51,14 +50,8 @@ class NetBoxClient: def __init__(self) -> None: self.netbox_params = load_oss_params().NETBOX - self.netbox = self._connect() - - def _connect(self): - try: - return pynetbox.api(self.netbox_params.api, self.netbox_params.token) - except RequestError as e: - raise Exception('NetBox API Error', e) - + self.netbox = pynetbox.api(self.netbox_params.api, self.netbox_params.token) + def get_all_devices(self) -> list[Devices]: return list(self.netbox.dcim.devices.all())