diff --git a/gso/services/netbox_client.py b/gso/services/netbox_client.py index 84d9c7e5b8bdd3c7abb6bc992d6e4841c25b3918..67cfe77e767346d1dae215fc28d589960f7578db 100644 --- a/gso/services/netbox_client.py +++ b/gso/services/netbox_client.py @@ -9,7 +9,7 @@ import pydantic import pynetbox from gso.products import Router -from gso.settings import NetBoxParams +from gso.settings import NetBoxParams, load_oss_params from gso.utils.device_info import TierInfo, FEASIBLE_LAG_RANGE from gso.utils.exceptions import NotFoundError, WorkflowStateException @@ -61,7 +61,8 @@ class Site(pydantic.BaseModel): class NetBoxClient: def __init__(self): - self.netbox = pynetbox.api(NetBoxParams.api, NetBoxParams.token) + netbox_params = load_oss_params().NETBOX + self.netbox = pynetbox.api(netbox_params.api, netbox_params.token) def get_all_devices(self): return list(self.netbox.dcim.devices.all())