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

Update netbox_initial_setup based on the latest changes.

parent f2faa18c
No related branches found
No related tags found
1 merge request!77Netbox integration including intial CLI for populating base data and ...
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}")
......
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