diff --git a/.env.example b/.env.example index 83cd7be..70e1d3e 100644 --- a/.env.example +++ b/.env.example @@ -1,2 +1,3 @@ NETBOX_URL= NETBOX_TOKEN= +REPO_URL=https://github.com/netbox-community/devicetype-library.git diff --git a/nb-dt-import.py b/nb-dt-import.py index f0cde9b..08f3427 100755 --- a/nb-dt-import.py +++ b/nb-dt-import.py @@ -1,6 +1,7 @@ #!/usr/bin/env python3 from git import Repo, exc, RemoteProgress from collections import Counter +from datetime import datetime import yaml import pynetbox import glob @@ -8,7 +9,7 @@ import argparse import os import settings -REPO_URL = 'https://github.com/netbox-community/devicetype-library.git' +REPO_URL = settings.REPO_URL parser = argparse.ArgumentParser(description='Import Netbox Device Types') parser.add_argument('--vendor', nargs='+', help="List of vendors to import eg. apc cisco") @@ -20,6 +21,7 @@ cwd = os.getcwd() counter = Counter(added=0, updated=0, manufacturer=0) nbUrl = settings.NETBOX_URL nbToken = settings.NETBOX_TOKEN +startTime = datetime.now() def update_package(path: str): @@ -358,6 +360,8 @@ else: createDeviceTypes(deviceTypes, nb) print('---') +print('Script took {} to run'.format(datetime.now() - startTime)) print('{} devices created'.format(counter['added'])) print('{} interfaces/ports updated'.format(counter['updated'])) print('{} manufacturers created'.format(counter['manufacturer'])) + diff --git a/settings.py b/settings.py index 7012e62..5785129 100644 --- a/settings.py +++ b/settings.py @@ -2,10 +2,11 @@ import os from dotenv import load_dotenv load_dotenv() +REPO_URL = str(os.getenv("REPO_URL")) NETBOX_URL = str(os.getenv("NETBOX_URL")) NETBOX_TOKEN = str(os.getenv("NETBOX_TOKEN")) -MANDATORY_ENV_VARS = ["NETBOX_URL", "NETBOX_TOKEN"] +MANDATORY_ENV_VARS = ["REPO_URL", "NETBOX_URL", "NETBOX_TOKEN"] for var in MANDATORY_ENV_VARS: if var not in os.environ: