diff --git a/nb-dt-import.py b/nb-dt-import.py index 844a2c7..49c3997 100755 --- a/nb-dt-import.py +++ b/nb-dt-import.py @@ -10,8 +10,9 @@ import os import settings REPO_URL = settings.REPO_URL + parser = argparse.ArgumentParser(description='Import Netbox Device Types') -parser.add_argument('--vendors', nargs='+', +parser.add_argument('--vendors', nargs='+', default=settings.VENDORS, help="List of vendors to import eg. apc cisco") parser.add_argument('--url', '--git', default=REPO_URL, help="Git URL with valid Device Type YAML files") diff --git a/settings.py b/settings.py index 5785129..40e42bd 100644 --- a/settings.py +++ b/settings.py @@ -6,6 +6,12 @@ REPO_URL = str(os.getenv("REPO_URL")) NETBOX_URL = str(os.getenv("NETBOX_URL")) NETBOX_TOKEN = str(os.getenv("NETBOX_TOKEN")) +# optionnally load vendors through a space separated list as env var +try: + VENDORS = os.getenv("VENDORS").split(" ") +except AttributeError: + VENDORS = None + MANDATORY_ENV_VARS = ["REPO_URL", "NETBOX_URL", "NETBOX_TOKEN"] for var in MANDATORY_ENV_VARS: