From c14d32d69a12da8197c8284023b805baeb30faed Mon Sep 17 00:00:00 2001
From: Neda Moeini <neda.moeini@GA0479-NMOEINI.local>
Date: Fri, 22 Sep 2023 14:28:09 +0200
Subject: [PATCH] Update netbox_initial_setup based on the latest changes.
---
gso/cli/netbox.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/gso/cli/netbox.py b/gso/cli/netbox.py
index 00f8f399..d3016443 100644
--- a/gso/cli/netbox.py
+++ b/gso/cli/netbox.py
@@ -1,7 +1,7 @@
import typer
from pynetbox import RequestError
-from gso.services.netbox_client import NetBoxClient, create_device_role, create_device_site
+from gso.services.netbox_client import NetBoxClient
app: typer.Typer = typer.Typer()
@@ -18,21 +18,21 @@ def netbox_initial_setup() -> None:
typer.echo("Connecting to NetBox ...")
try:
- nbclient = NetBoxClient().connect()
+ nbclient = NetBoxClient()
except RequestError as e:
typer.echo(f"Error connecting to NetBox: {e}")
return
typer.echo("Creating GEANT site ...")
try:
- create_device_site(nbclient, "GEANT", "geant")
+ nbclient.create_device_site("GEANT", "geant")
typer.echo("Site created successfully.")
except RequestError as e:
typer.echo(f"Error creating site: {e}")
typer.echo("Creating Router device role ...")
try:
- create_device_role(nbclient, "router", "router")
+ nbclient.create_device_role("router", "router")
typer.echo("Device role created successfully.")
except RequestError as e:
typer.echo(f"Error creating device role: {e}")
--
GitLab