David Mc Ken bee2eb7776 Add branch support
Set default branch name of 'master' in settings.py and then checkout the branch as appropriate.
2021-10-31 17:54:46 -04:00

19 lines
604 B
Python

import os
from dotenv import load_dotenv
load_dotenv()
REPO_URL = os.getenv("REPO_URL")
REPO_BRANCH = os.getenv("REPO_BRANCH", 'master')
NETBOX_URL = os.getenv("NETBOX_URL")
NETBOX_TOKEN = os.getenv("NETBOX_TOKEN")
IGNORE_SSL_ERRORS = (os.getenv("IGNORE_SSL_ERRORS", "False") == "True")
# optionnally load vendors through a space separated list as env var
VENDORS = os.getenv("VENDORS", "").split()
MANDATORY_ENV_VARS = ["REPO_URL", "NETBOX_URL", "NETBOX_TOKEN"]
for var in MANDATORY_ENV_VARS:
if var not in os.environ:
raise EnvironmentError("Failed because {} is not set.".format(var))