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

Fixed get params bug in netbox client.

parent c14d32d6
Branches
Tags
1 merge request!77Netbox integration including intial CLI for populating base data and ...
Hide whitespace changes
Inline Side-by-side
gso/services/netbox_client.py
+ 3
− 2
View file @ 3e9c2792
...@@ -9,7 +9,7 @@ import pydantic ...@@ -9,7 +9,7 @@ import pydantic
import pynetbox import pynetbox
from gso.products import Router 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.device_info import TierInfo, FEASIBLE_LAG_RANGE
from gso.utils.exceptions import NotFoundError, WorkflowStateException from gso.utils.exceptions import NotFoundError, WorkflowStateException
...@@ -61,7 +61,8 @@ class Site(pydantic.BaseModel): ...@@ -61,7 +61,8 @@ class Site(pydantic.BaseModel):
class NetBoxClient: class NetBoxClient:
def __init__(self): 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): def get_all_devices(self):
return list(self.netbox.dcim.devices.all()) return list(self.netbox.dcim.devices.all())
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment