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
No related branches found
No related tags found
1 merge request!77Netbox integration including intial CLI for populating base data and ...
......@@ -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())
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment