Use black as formatter for python files

.editorconfig

@@ -7,5 +7,5 @@ charset = utf-8
 indent_style = space
 indent_size = 2
 
-[configuration/**.py]
+[*.py]
 indent_size = 4

configuration/configuration.py

@@ -5,9 +5,8 @@
 ####
 
 import re
-
-from os.path import dirname, abspath, join
 from os import environ
+from os.path import abspath, dirname, join
 
 # For reference see https://netbox.readthedocs.io/en/stable/configuration/
 # Based on https://github.com/netbox-community/netbox/blob/master/netbox/netbox/configuration.example.py

configuration/ldap/ldap_config.py

@@ -1,9 +1,10 @@
-import ldap
-
-from django_auth_ldap.config import LDAPSearch
 from importlib import import_module
 from os import environ
 
+import ldap
+from django_auth_ldap.config import LDAPSearch
+
+
 # Read secret from file
 def _read_secret(secret_name, default=None):
     try:
@@ -47,9 +48,11 @@ LDAP_IGNORE_CERT_ERRORS = environ.get('LDAP_IGNORE_CERT_ERRORS', 'False').lower(
 
 AUTH_LDAP_USER_SEARCH_BASEDN = environ.get('AUTH_LDAP_USER_SEARCH_BASEDN', '')
 AUTH_LDAP_USER_SEARCH_ATTR = environ.get('AUTH_LDAP_USER_SEARCH_ATTR', 'sAMAccountName')
-AUTH_LDAP_USER_SEARCH = LDAPSearch(AUTH_LDAP_USER_SEARCH_BASEDN,
+AUTH_LDAP_USER_SEARCH = LDAPSearch(
+    AUTH_LDAP_USER_SEARCH_BASEDN,
     ldap.SCOPE_SUBTREE,
-                                   "(" + AUTH_LDAP_USER_SEARCH_ATTR + "=%(user)s)")
+    "(" + AUTH_LDAP_USER_SEARCH_ATTR + "=%(user)s)"
+)
 
 # This search ought to return all groups to which the user belongs. django_auth_ldap uses this to determine group
 # heirarchy.

docker/configuration.docker.py

@@ -4,10 +4,10 @@
 #
 # They can be imported by other code (see `ldap_config.py` for an example).
 
-from os.path import abspath, isfile
-from os import scandir
 import importlib.util
 import sys
+from os import scandir
+from os.path import abspath, isfile
 
 
 def _filename(f):
@@ -15,7 +15,7 @@ def _filename(f):
 
 
 def _import(module_name, path, loaded_configurations):
-  spec = importlib.util.spec_from_file_location('', path)
+    spec = importlib.util.spec_from_file_location("", path)
     module = importlib.util.module_from_spec(spec)
     spec.loader.exec_module(module)
     sys.modules[module_name] = module
@@ -28,9 +28,9 @@ def _import(module_name, path, loaded_configurations):
 def read_configurations(config_module, config_dir, main_config):
     loaded_configurations = []
 
-  main_config_path = abspath(f'{config_dir}/{main_config}.py')
+    main_config_path = abspath(f"{config_dir}/{main_config}.py")
     if isfile(main_config_path):
-    _import(f'{config_module}.{main_config}', main_config_path, loaded_configurations)
+        _import(f"{config_module}.{main_config}", main_config_path, loaded_configurations)
     else:
         print(f"⚠️ Main configuration '{main_config_path}' not found.")
 
@@ -39,13 +39,13 @@ def read_configurations(config_module, config_dir, main_config):
             if not f.is_file():
                 continue
 
-      if f.name.startswith('__'):
+            if f.name.startswith("__"):
                 continue
 
-      if not f.name.endswith('.py'):
+            if not f.name.endswith(".py"):
                 continue
 
-      if f.name == f'{config_dir}.py':
+            if f.name == f"{config_dir}.py":
                 continue
 
             module_name = f"{config_module}.{f.name[:-len('.py')]}".replace(".", "_")
@@ -66,9 +66,10 @@ def read_configurations(config_module, config_dir, main_config):
 
 
 _loaded_configurations = read_configurations(
-  config_dir = '/etc/netbox/config/',
+    config_dir="/etc/netbox/config/",
-  config_module = 'netbox.configuration',
+    config_module="netbox.configuration",
-  main_config = 'configuration')
+    main_config="configuration",
+)
 
 
 def __getattr__(name):

docker/ldap_config.docker.py

@@ -1,9 +1,10 @@
 from .configuration import read_configurations
 
 _loaded_configurations = read_configurations(
-  config_dir = '/etc/netbox/config/ldap/',
+    config_dir="/etc/netbox/config/ldap/",
-  config_module = 'netbox.configuration.ldap',
+    config_module="netbox.configuration.ldap",
-  main_config = 'ldap_config')
+    main_config="ldap_config",
+)
 
 
 def __getattr__(name):
@@ -14,6 +15,7 @@ def __getattr__(name):
             pass
     raise AttributeError
 
+
 def __dir__():
     names = []
     for config in _loaded_configurations:

pyproject.toml

@@ -0,0 +1,21 @@
+[tool.black]
+line-length = 100
+target-version = ['py38']
+include = '\.pyi?$'
+exclude = '''
+
+(
+  /(
+      \.eggs         # exclude a few common directories in the
+    | \.git          # root of the project
+    | \.venv
+    | \.netbox
+    | \.vscode
+    | configuration
+  )/
+)
+'''
+
+[tool.isort]
+profile = "black"
+multi_line_output = 3

startup_scripts/000_users.py

@@ -4,20 +4,21 @@ from django.contrib.auth.models import User
 from startup_script_utils import load_yaml, set_permissions
 from users.models import Token
 
-users = load_yaml('/opt/netbox/initializers/users.yml')
+users = load_yaml("/opt/netbox/initializers/users.yml")
 if users is None:
     sys.exit()
 
 for username, user_details in users.items():
     if not User.objects.filter(username=username):
         user = User.objects.create_user(
-      username = username,
+            username=username,
-      password = user_details.get('password', 0) or User.objects.make_random_password())
+            password=user_details.get("password", 0) or User.objects.make_random_password(),
+        )
 
-    print("👤 Created user",username)
+        print("👤 Created user", username)
 
-    if user_details.get('api_token', 0):
+        if user_details.get("api_token", 0):
-      Token.objects.create(user=user, key=user_details['api_token'])
+            Token.objects.create(user=user, key=user_details["api_token"])
 
-    yaml_permissions = user_details.get('permissions', [])
+        yaml_permissions = user_details.get("permissions", [])
         set_permissions(user.user_permissions, yaml_permissions)

startup_scripts/010_groups.py

@@ -3,7 +3,7 @@ import sys
 from django.contrib.auth.models import Group, User
 from startup_script_utils import load_yaml, set_permissions
 
-groups = load_yaml('/opt/netbox/initializers/groups.yml')
+groups = load_yaml("/opt/netbox/initializers/groups.yml")
 if groups is None:
     sys.exit()
 
@@ -13,11 +13,11 @@ for groupname, group_details in groups.items():
     if created:
         print("👥 Created group", groupname)
 
-  for username in group_details.get('users', []):
+    for username in group_details.get("users", []):
         user = User.objects.get(username=username)
 
         if user:
             user.groups.add(group)
 
-  yaml_permissions = group_details.get('permissions', [])
+    yaml_permissions = group_details.get("permissions", [])
     set_permissions(group.permissions, yaml_permissions)

startup_scripts/020_custom_fields.py

@@ -3,8 +3,10 @@ import sys
 from extras.models import CustomField
 from startup_script_utils import load_yaml
 
+
 def get_class_for_class_path(class_path):
     import importlib
+
     from django.contrib.contenttypes.models import ContentType
 
     module_name, class_name = class_path.rsplit(".", 1)
@@ -12,44 +14,48 @@ def get_class_for_class_path(class_path):
     clazz = getattr(module, class_name)
     return ContentType.objects.get_for_model(clazz)
 
-customfields = load_yaml('/opt/netbox/initializers/custom_fields.yml')
+
+customfields = load_yaml("/opt/netbox/initializers/custom_fields.yml")
 
 if customfields is None:
     sys.exit()
 
 for cf_name, cf_details in customfields.items():
-  custom_field, created = CustomField.objects.get_or_create(name = cf_name)
+    custom_field, created = CustomField.objects.get_or_create(name=cf_name)
 
     if created:
-    if cf_details.get('default', False):
+        if cf_details.get("default", False):
-      custom_field.default = cf_details['default']
+            custom_field.default = cf_details["default"]
 
-    if cf_details.get('description', False):
+        if cf_details.get("description", False):
-      custom_field.description = cf_details['description']
+            custom_field.description = cf_details["description"]
 
-    if cf_details.get('label', False):
+        if cf_details.get("label", False):
-      custom_field.label = cf_details['label']
+            custom_field.label = cf_details["label"]
 
-    for object_type in cf_details.get('on_objects', []):
+        for object_type in cf_details.get("on_objects", []):
             custom_field.content_types.add(get_class_for_class_path(object_type))
 
-    if cf_details.get('required', False):
+        if cf_details.get("required", False):
-      custom_field.required = cf_details['required']
+            custom_field.required = cf_details["required"]
 
-    if cf_details.get('type', False):
+        if cf_details.get("type", False):
-      custom_field.type = cf_details['type']
+            custom_field.type = cf_details["type"]
 
-    if cf_details.get('weight', -1) >= 0:
+        if cf_details.get("weight", -1) >= 0:
-      custom_field.weight = cf_details['weight']
+            custom_field.weight = cf_details["weight"]
 
-    if cf_details.get('choices', False):
+        if cf_details.get("choices", False):
             custom_field.choices = []
 
-      for choice_detail in cf_details.get('choices', []):
+            for choice_detail in cf_details.get("choices", []):
-        if isinstance(choice_detail, dict) and 'value' in choice_detail:
+                if isinstance(choice_detail, dict) and "value" in choice_detail:
                     # legacy mode
-          print(f"⚠️ Please migrate the choice '{choice_detail['value']}' of '{cf_name}' to the new format, as 'weight' is no longer supported!")
+                    print(
-          custom_field.choices.append(choice_detail['value'])
+                        f"⚠️ Please migrate the choice '{choice_detail['value']}' of '{cf_name}'"
+                        + " to the new format, as 'weight' is no longer supported!"
+                    )
+                    custom_field.choices.append(choice_detail["value"])
                 else:
                     custom_field.choices.append(choice_detail)
 

startup_scripts/020_tags.py

@@ -1,21 +1,21 @@
-from extras.models import Tag
-from utilities.choices import ColorChoices
-
-from startup_script_utils import load_yaml
 import sys
 
-tags = load_yaml('/opt/netbox/initializers/tags.yml')
+from extras.models import Tag
+from startup_script_utils import load_yaml
+from utilities.choices import ColorChoices
+
+tags = load_yaml("/opt/netbox/initializers/tags.yml")
 
 if tags is None:
     sys.exit()
 
 for params in tags:
-  if 'color' in params:
+    if "color" in params:
-    color = params.pop('color')
+        color = params.pop("color")
 
         for color_tpl in ColorChoices:
             if color in color_tpl:
-        params['color'] = color_tpl[0]
+                params["color"] = color_tpl[0]
 
     tag, created = Tag.objects.get_or_create(**params)
 

startup_scripts/030_regions.py

@@ -1,22 +1,21 @@
-from dcim.models import Region
-from startup_script_utils import load_yaml
 import sys
 
-regions = load_yaml('/opt/netbox/initializers/regions.yml')
+from dcim.models import Region
+from startup_script_utils import load_yaml
+
+regions = load_yaml("/opt/netbox/initializers/regions.yml")
 
 if regions is None:
     sys.exit()
 
-optional_assocs = {
+optional_assocs = {"parent": (Region, "name")}
-  'parent': (Region, 'name')
-}
 
 for params in regions:
 
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/040_sites.py

@@ -1,18 +1,15 @@
 import sys
 
 from dcim.models import Region, Site
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from tenancy.models import Tenant
 
-sites = load_yaml('/opt/netbox/initializers/sites.yml')
+sites = load_yaml("/opt/netbox/initializers/sites.yml")
 
 if sites is None:
     sys.exit()
 
-optional_assocs = {
+optional_assocs = {"region": (Region, "name"), "tenant": (Tenant, "name")}
-  'region': (Region, 'name'),
-  'tenant': (Tenant, 'name')
-}
 
 for params in sites:
     custom_field_data = pop_custom_fields(params)
@@ -20,7 +17,7 @@ for params in sites:
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/050_manufacturers.py

@@ -1,8 +1,9 @@
-from dcim.models import Manufacturer
-from startup_script_utils import load_yaml
 import sys
 
-manufacturers = load_yaml('/opt/netbox/initializers/manufacturers.yml')
+from dcim.models import Manufacturer
+from startup_script_utils import load_yaml
+
+manufacturers = load_yaml("/opt/netbox/initializers/manufacturers.yml")
 
 if manufacturers is None:
     sys.exit()

startup_scripts/060_device_types.py

@@ -1,36 +1,31 @@
 import sys
 
 from dcim.models import DeviceType, Manufacturer, Region
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from tenancy.models import Tenant
 
-device_types = load_yaml('/opt/netbox/initializers/device_types.yml')
+device_types = load_yaml("/opt/netbox/initializers/device_types.yml")
 
 if device_types is None:
     sys.exit()
 
-required_assocs = {
+required_assocs = {"manufacturer": (Manufacturer, "name")}
-  'manufacturer': (Manufacturer, 'name')
-}
 
-optional_assocs = {
+optional_assocs = {"region": (Region, "name"), "tenant": (Tenant, "name")}
-  'region': (Region, 'name'),
-  'tenant': (Tenant, 'name')
-}
 
 for params in device_types:
     custom_field_data = pop_custom_fields(params)
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
 
         params[assoc] = model.objects.get(**query)
 
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/070_rack_roles.py

@@ -1,21 +1,21 @@
-from dcim.models import RackRole
-from utilities.choices import ColorChoices
-
-from startup_script_utils import load_yaml
 import sys
 
-rack_roles = load_yaml('/opt/netbox/initializers/rack_roles.yml')
+from dcim.models import RackRole
+from startup_script_utils import load_yaml
+from utilities.choices import ColorChoices
+
+rack_roles = load_yaml("/opt/netbox/initializers/rack_roles.yml")
 
 if rack_roles is None:
     sys.exit()
 
 for params in rack_roles:
-  if 'color' in params:
+    if "color" in params:
-    color = params.pop('color')
+        color = params.pop("color")
 
         for color_tpl in ColorChoices:
             if color in color_tpl:
-        params['color'] = color_tpl[0]
+                params["color"] = color_tpl[0]
 
     rack_role, created = RackRole.objects.get_or_create(**params)
 

startup_scripts/075_rack_groups.py

@@ -1,25 +1,23 @@
-from dcim.models import Site,RackGroup
-from startup_script_utils import load_yaml
 import sys
 
-rack_groups = load_yaml('/opt/netbox/initializers/rack_groups.yml')
+from dcim.models import RackGroup, Site
+from startup_script_utils import load_yaml
+
+rack_groups = load_yaml("/opt/netbox/initializers/rack_groups.yml")
 
 if rack_groups is None:
     sys.exit()
 
-required_assocs = {
+required_assocs = {"site": (Site, "name")}
-  'site': (Site, 'name')
-}
 
 for params in rack_groups:
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
         params[assoc] = model.objects.get(**query)
 
     rack_group, created = RackGroup.objects.get_or_create(**params)
 
     if created:
         print("🎨 Created rack group", rack_group.name)
-

startup_scripts/080_racks.py

@@ -1,22 +1,20 @@
 import sys
 
-from dcim.models import Site, RackRole, Rack, RackGroup
+from dcim.models import Rack, RackGroup, RackRole, Site
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from tenancy.models import Tenant
 
-racks = load_yaml('/opt/netbox/initializers/racks.yml')
+racks = load_yaml("/opt/netbox/initializers/racks.yml")
 
 if racks is None:
     sys.exit()
 
-required_assocs = {
+required_assocs = {"site": (Site, "name")}
-  'site': (Site, 'name')
-}
 
 optional_assocs = {
-  'role': (RackRole, 'name'),
+    "role": (RackRole, "name"),
-  'tenant': (Tenant, 'name'),
+    "tenant": (Tenant, "name"),
-  'group': (RackGroup, 'name')
+    "group": (RackGroup, "name"),
 }
 
 for params in racks:
@@ -24,14 +22,14 @@ for params in racks:
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
 
         params[assoc] = model.objects.get(**query)
 
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/090_device_roles.py

@@ -1,22 +1,22 @@
-from dcim.models import DeviceRole
-from utilities.choices import ColorChoices
-
-from startup_script_utils import load_yaml
 import sys
 
-device_roles = load_yaml('/opt/netbox/initializers/device_roles.yml')
+from dcim.models import DeviceRole
+from startup_script_utils import load_yaml
+from utilities.choices import ColorChoices
+
+device_roles = load_yaml("/opt/netbox/initializers/device_roles.yml")
 
 if device_roles is None:
     sys.exit()
 
 for params in device_roles:
 
-  if 'color' in params:
+    if "color" in params:
-    color = params.pop('color')
+        color = params.pop("color")
 
         for color_tpl in ColorChoices:
             if color in color_tpl:
-        params['color'] = color_tpl[0]
+                params["color"] = color_tpl[0]
 
     device_role, created = DeviceRole.objects.get_or_create(**params)
 

startup_scripts/100_platforms.py

@@ -1,14 +1,15 @@
-from dcim.models import Manufacturer, Platform
-from startup_script_utils import load_yaml
 import sys
 
-platforms = load_yaml('/opt/netbox/initializers/platforms.yml')
+from dcim.models import Manufacturer, Platform
+from startup_script_utils import load_yaml
+
+platforms = load_yaml("/opt/netbox/initializers/platforms.yml")
 
 if platforms is None:
     sys.exit()
 
 optional_assocs = {
-  'manufacturer': (Manufacturer, 'name'),
+    "manufacturer": (Manufacturer, "name"),
 }
 
 for params in platforms:
@@ -16,7 +17,7 @@ for params in platforms:
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/110_tenant_groups.py

@@ -1,8 +1,9 @@
-from tenancy.models import TenantGroup
-from startup_script_utils import load_yaml
 import sys
 
-tenant_groups = load_yaml('/opt/netbox/initializers/tenant_groups.yml')
+from startup_script_utils import load_yaml
+from tenancy.models import TenantGroup
+
+tenant_groups = load_yaml("/opt/netbox/initializers/tenant_groups.yml")
 
 if tenant_groups is None:
     sys.exit()

startup_scripts/120_tenants.py

@@ -1,16 +1,14 @@
 import sys
 
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from tenancy.models import Tenant, TenantGroup
 
-tenants = load_yaml('/opt/netbox/initializers/tenants.yml')
+tenants = load_yaml("/opt/netbox/initializers/tenants.yml")
 
 if tenants is None:
     sys.exit()
 
-optional_assocs = {
+optional_assocs = {"group": (TenantGroup, "name")}
-  'group': (TenantGroup, 'name')
-}
 
 for params in tenants:
     custom_field_data = pop_custom_fields(params)
@@ -18,7 +16,7 @@ for params in tenants:
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/130_cluster_types.py

@@ -1,8 +1,9 @@
-from virtualization.models import ClusterType
-from startup_script_utils import load_yaml
 import sys
 
-cluster_types = load_yaml('/opt/netbox/initializers/cluster_types.yml')
+from startup_script_utils import load_yaml
+from virtualization.models import ClusterType
+
+cluster_types = load_yaml("/opt/netbox/initializers/cluster_types.yml")
 
 if cluster_types is None:
     sys.exit()

startup_scripts/135_cluster_groups.py

@@ -1,8 +1,9 @@
-from virtualization.models import ClusterGroup
-from startup_script_utils import load_yaml
 import sys
 
-cluster_groups = load_yaml('/opt/netbox/initializers/cluster_groups.yml')
+from startup_script_utils import load_yaml
+from virtualization.models import ClusterGroup
+
+cluster_groups = load_yaml("/opt/netbox/initializers/cluster_groups.yml")
 
 if cluster_groups is None:
     sys.exit()

startup_scripts/135_clusters.py

@@ -1,23 +1,21 @@
 import sys
 
 from dcim.models import Site
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
-from virtualization.models import Cluster, ClusterType, ClusterGroup
 from tenancy.models import Tenant
+from virtualization.models import Cluster, ClusterGroup, ClusterType
 
-clusters = load_yaml('/opt/netbox/initializers/clusters.yml')
+clusters = load_yaml("/opt/netbox/initializers/clusters.yml")
 
 if clusters is None:
     sys.exit()
 
-required_assocs = {
+required_assocs = {"type": (ClusterType, "name")}
-  'type': (ClusterType, 'name')
-}
 
 optional_assocs = {
-  'site': (Site, 'name'),
+    "site": (Site, "name"),
-  'group': (ClusterGroup, 'name'),
+    "group": (ClusterGroup, "name"),
-  'tenant': (Tenant, 'name')
+    "tenant": (Tenant, "name"),
 }
 
 for params in clusters:
@@ -25,14 +23,14 @@ for params in clusters:
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
 
         params[assoc] = model.objects.get(**query)
 
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/140_clusters.py

@@ -1,23 +1,21 @@
 import sys
 
 from dcim.models import Site
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
-from virtualization.models import Cluster, ClusterType, ClusterGroup
 from tenancy.models import Tenant
+from virtualization.models import Cluster, ClusterGroup, ClusterType
 
-clusters = load_yaml('/opt/netbox/initializers/clusters.yml')
+clusters = load_yaml("/opt/netbox/initializers/clusters.yml")
 
 if clusters is None:
     sys.exit()
 
-required_assocs = {
+required_assocs = {"type": (ClusterType, "name")}
-  'type': (ClusterType, 'name')
-}
 
 optional_assocs = {
-  'site': (Site, 'name'),
+    "site": (Site, "name"),
-  'group': (ClusterGroup, 'name'),
+    "group": (ClusterGroup, "name"),
-  'tenant': (Tenant, 'name')
+    "tenant": (Tenant, "name"),
 }
 
 for params in clusters:
@@ -25,14 +23,14 @@ for params in clusters:
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
 
         params[assoc] = model.objects.get(**query)
 
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/140_devices.py

@@ -1,45 +1,45 @@
 import sys
 
-from dcim.models import Site, Rack, DeviceRole, DeviceType, Device, Platform
+from dcim.models import Device, DeviceRole, DeviceType, Platform, Rack, Site
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from tenancy.models import Tenant
 from virtualization.models import Cluster
 
-devices = load_yaml('/opt/netbox/initializers/devices.yml')
+devices = load_yaml("/opt/netbox/initializers/devices.yml")
 
 if devices is None:
     sys.exit()
 
 required_assocs = {
-  'device_role': (DeviceRole, 'name'),
+    "device_role": (DeviceRole, "name"),
-  'device_type': (DeviceType, 'model'),
+    "device_type": (DeviceType, "model"),
-  'site': (Site, 'name')
+    "site": (Site, "name"),
 }
 
 optional_assocs = {
-  'tenant': (Tenant, 'name'),
+    "tenant": (Tenant, "name"),
-  'platform': (Platform, 'name'),
+    "platform": (Platform, "name"),
-  'rack': (Rack, 'name'),
+    "rack": (Rack, "name"),
-  'cluster': (Cluster, 'name')
+    "cluster": (Cluster, "name"),
 }
 
 for params in devices:
     custom_field_data = pop_custom_fields(params)
 
     # primary ips are handled later in `270_primary_ips.py`
-  params.pop('primary_ip4', None)
+    params.pop("primary_ip4", None)
-  params.pop('primary_ip6', None)
+    params.pop("primary_ip6", None)
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
 
         params[assoc] = model.objects.get(**query)
 
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/145_devices.py

@@ -1,45 +1,45 @@
 import sys
 
-from dcim.models import Site, Rack, DeviceRole, DeviceType, Device, Platform
+from dcim.models import Device, DeviceRole, DeviceType, Platform, Rack, Site
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from tenancy.models import Tenant
 from virtualization.models import Cluster
 
-devices = load_yaml('/opt/netbox/initializers/devices.yml')
+devices = load_yaml("/opt/netbox/initializers/devices.yml")
 
 if devices is None:
     sys.exit()
 
 required_assocs = {
-  'device_role': (DeviceRole, 'name'),
+    "device_role": (DeviceRole, "name"),
-  'device_type': (DeviceType, 'model'),
+    "device_type": (DeviceType, "model"),
-  'site': (Site, 'name')
+    "site": (Site, "name"),
 }
 
 optional_assocs = {
-  'tenant': (Tenant, 'name'),
+    "tenant": (Tenant, "name"),
-  'platform': (Platform, 'name'),
+    "platform": (Platform, "name"),
-  'rack': (Rack, 'name'),
+    "rack": (Rack, "name"),
-  'cluster': (Cluster, 'name')
+    "cluster": (Cluster, "name"),
 }
 
 for params in devices:
     custom_field_data = pop_custom_fields(params)
 
     # primary ips are handled later in `270_primary_ips.py`
-  params.pop('primary_ip4', None)
+    params.pop("primary_ip4", None)
-  params.pop('primary_ip6', None)
+    params.pop("primary_ip6", None)
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
 
         params[assoc] = model.objects.get(**query)
 
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/150_rirs.py

@@ -1,8 +1,9 @@
-from ipam.models import RIR
-from startup_script_utils import load_yaml
 import sys
 
-rirs = load_yaml('/opt/netbox/initializers/rirs.yml')
+from ipam.models import RIR
+from startup_script_utils import load_yaml
+
+rirs = load_yaml("/opt/netbox/initializers/rirs.yml")
 
 if rirs is None:
     sys.exit()

startup_scripts/160_aggregates.py

@@ -1,38 +1,36 @@
 import sys
 
-from ipam.models import Aggregate, RIR
+from ipam.models import RIR, Aggregate
 from netaddr import IPNetwork
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from tenancy.models import Tenant
 
-aggregates = load_yaml('/opt/netbox/initializers/aggregates.yml')
+aggregates = load_yaml("/opt/netbox/initializers/aggregates.yml")
 
 if aggregates is None:
     sys.exit()
 
-required_assocs = {
+required_assocs = {"rir": (RIR, "name")}
-  'rir': (RIR, 'name')
-}
 
 optional_assocs = {
-  'tenant': (Tenant, 'name'),
+    "tenant": (Tenant, "name"),
 }
 
 for params in aggregates:
     custom_field_data = pop_custom_fields(params)
 
-  params['prefix'] = IPNetwork(params['prefix'])
+    params["prefix"] = IPNetwork(params["prefix"])
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
 
         params[assoc] = model.objects.get(**query)
 
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/165_cluster_groups.py

@@ -1,8 +1,9 @@
-from virtualization.models import ClusterGroup
-from startup_script_utils import load_yaml
 import sys
 
-cluster_groups = load_yaml('/opt/netbox/initializers/cluster_groups.yml')
+from startup_script_utils import load_yaml
+from virtualization.models import ClusterGroup
+
+cluster_groups = load_yaml("/opt/netbox/initializers/cluster_groups.yml")
 
 if cluster_groups is None:
     sys.exit()

startup_scripts/175_route_targets.py

@@ -1,17 +1,15 @@
 import sys
 
 from ipam.models import RouteTarget
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from tenancy.models import Tenant
 
-route_targets = load_yaml('/opt/netbox/initializers/route_targets.yml')
+route_targets = load_yaml("/opt/netbox/initializers/route_targets.yml")
 
 if route_targets is None:
     sys.exit()
 
-optional_assocs = {
+optional_assocs = {"tenant": (Tenant, "name")}
-  'tenant': (Tenant, 'name')
-}
 
 for params in route_targets:
     custom_field_data = pop_custom_fields(params)
@@ -19,7 +17,7 @@ for params in route_targets:
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/180_vrfs.py

@@ -1,17 +1,15 @@
 import sys
 
 from ipam.models import VRF
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from tenancy.models import Tenant
 
-vrfs = load_yaml('/opt/netbox/initializers/vrfs.yml')
+vrfs = load_yaml("/opt/netbox/initializers/vrfs.yml")
 
 if vrfs is None:
     sys.exit()
 
-optional_assocs = {
+optional_assocs = {"tenant": (Tenant, "name")}
-  'tenant': (Tenant, 'name')
-}
 
 for params in vrfs:
     custom_field_data = pop_custom_fields(params)
@@ -19,7 +17,7 @@ for params in vrfs:
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/190_prefix_vlan_roles.py

@@ -1,8 +1,9 @@
-from ipam.models import Role
-from startup_script_utils import load_yaml
 import sys
 
-roles = load_yaml('/opt/netbox/initializers/prefix_vlan_roles.yml')
+from ipam.models import Role
+from startup_script_utils import load_yaml
+
+roles = load_yaml("/opt/netbox/initializers/prefix_vlan_roles.yml")
 
 if roles is None:
     sys.exit()

startup_scripts/200_vlan_groups.py

@@ -2,16 +2,14 @@ import sys
 
 from dcim.models import Site
 from ipam.models import VLANGroup
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 
-vlan_groups = load_yaml('/opt/netbox/initializers/vlan_groups.yml')
+vlan_groups = load_yaml("/opt/netbox/initializers/vlan_groups.yml")
 
 if vlan_groups is None:
     sys.exit()
 
-optional_assocs = {
+optional_assocs = {"site": (Site, "name")}
-  'site': (Site, 'name')
-}
 
 for params in vlan_groups:
     custom_field_data = pop_custom_fields(params)
@@ -19,7 +17,7 @@ for params in vlan_groups:
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/210_vlans.py

@@ -1,21 +1,21 @@
 import sys
 
 from dcim.models import Site
-from ipam.models import VLAN, VLANGroup, Role
+from ipam.models import VLAN, Role, VLANGroup
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from tenancy.models import Tenant, TenantGroup
 
-vlans = load_yaml('/opt/netbox/initializers/vlans.yml')
+vlans = load_yaml("/opt/netbox/initializers/vlans.yml")
 
 if vlans is None:
     sys.exit()
 
 optional_assocs = {
-  'site': (Site, 'name'),
+    "site": (Site, "name"),
-  'tenant': (Tenant, 'name'),
+    "tenant": (Tenant, "name"),
-  'tenant_group': (TenantGroup, 'name'),
+    "tenant_group": (TenantGroup, "name"),
-  'group': (VLANGroup, 'name'),
+    "group": (VLANGroup, "name"),
-  'role': (Role, 'name')
+    "role": (Role, "name"),
 }
 
 for params in vlans:
@@ -24,7 +24,7 @@ for params in vlans:
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/220_prefixes.py

@@ -1,34 +1,34 @@
 import sys
 
 from dcim.models import Site
-from ipam.models import Prefix, VLAN, Role, VRF
+from ipam.models import VLAN, VRF, Prefix, Role
 from netaddr import IPNetwork
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from tenancy.models import Tenant, TenantGroup
 
-prefixes = load_yaml('/opt/netbox/initializers/prefixes.yml')
+prefixes = load_yaml("/opt/netbox/initializers/prefixes.yml")
 
 if prefixes is None:
     sys.exit()
 
 optional_assocs = {
-  'site': (Site, 'name'),
+    "site": (Site, "name"),
-  'tenant': (Tenant, 'name'),
+    "tenant": (Tenant, "name"),
-  'tenant_group': (TenantGroup, 'name'),
+    "tenant_group": (TenantGroup, "name"),
-  'vlan': (VLAN, 'name'),
+    "vlan": (VLAN, "name"),
-  'role': (Role, 'name'),
+    "role": (Role, "name"),
-  'vrf': (VRF, 'name')
+    "vrf": (VRF, "name"),
 }
 
 for params in prefixes:
     custom_field_data = pop_custom_fields(params)
 
-  params['prefix'] = IPNetwork(params['prefix'])
+    params["prefix"] = IPNetwork(params["prefix"])
 
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
             params[assoc] = model.objects.get(**query)
 
     prefix, created = Prefix.objects.get_or_create(**params)

startup_scripts/230_virtual_machines.py

@@ -1,42 +1,40 @@
 import sys
 
-from dcim.models import Platform, DeviceRole
+from dcim.models import DeviceRole, Platform
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from tenancy.models import Tenant
 from virtualization.models import Cluster, VirtualMachine
 
-virtual_machines = load_yaml('/opt/netbox/initializers/virtual_machines.yml')
+virtual_machines = load_yaml("/opt/netbox/initializers/virtual_machines.yml")
 
 if virtual_machines is None:
     sys.exit()
 
-required_assocs = {
+required_assocs = {"cluster": (Cluster, "name")}
-  'cluster': (Cluster, 'name')
-}
 
 optional_assocs = {
-  'tenant': (Tenant, 'name'),
+    "tenant": (Tenant, "name"),
-  'platform': (Platform, 'name'),
+    "platform": (Platform, "name"),
-  'role': (DeviceRole, 'name')
+    "role": (DeviceRole, "name"),
 }
 
 for params in virtual_machines:
     custom_field_data = pop_custom_fields(params)
 
     # primary ips are handled later in `270_primary_ips.py`
-  params.pop('primary_ip4', None)
+    params.pop("primary_ip4", None)
-  params.pop('primary_ip6', None)
+    params.pop("primary_ip6", None)
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
 
         params[assoc] = model.objects.get(**query)
 
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/240_virtualization_interfaces.py

@@ -1,23 +1,21 @@
 import sys
 
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from virtualization.models import VirtualMachine, VMInterface
 
-interfaces = load_yaml('/opt/netbox/initializers/virtualization_interfaces.yml')
+interfaces = load_yaml("/opt/netbox/initializers/virtualization_interfaces.yml")
 
 if interfaces is None:
     sys.exit()
 
-required_assocs = {
+required_assocs = {"virtual_machine": (VirtualMachine, "name")}
-  'virtual_machine': (VirtualMachine, 'name')
-}
 
 for params in interfaces:
     custom_field_data = pop_custom_fields(params)
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
 
         params[assoc] = model.objects.get(**query)
 

startup_scripts/250_dcim_interfaces.py

@@ -1,23 +1,21 @@
 import sys
 
-from dcim.models import Interface, Device
+from dcim.models import Device, Interface
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 
-interfaces = load_yaml('/opt/netbox/initializers/dcim_interfaces.yml')
+interfaces = load_yaml("/opt/netbox/initializers/dcim_interfaces.yml")
 
 if interfaces is None:
     sys.exit()
 
-required_assocs = {
+required_assocs = {"device": (Device, "name")}
-  'device': (Device, 'name')
-}
 
 for params in interfaces:
     custom_field_data = pop_custom_fields(params)
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
 
         params[assoc] = model.objects.get(**query)
 

startup_scripts/260_ip_addresses.py

@@ -5,30 +5,32 @@ from django.contrib.contenttypes.models import ContentType
 from django.db.models import Q
 from ipam.models import VRF, IPAddress
 from netaddr import IPNetwork
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
 from tenancy.models import Tenant
 from virtualization.models import VirtualMachine, VMInterface
 
-ip_addresses = load_yaml('/opt/netbox/initializers/ip_addresses.yml')
+ip_addresses = load_yaml("/opt/netbox/initializers/ip_addresses.yml")
 
 if ip_addresses is None:
     sys.exit()
 
 optional_assocs = {
-  'tenant': (Tenant, 'name'),
+    "tenant": (Tenant, "name"),
-  'vrf': (VRF, 'name'),
+    "vrf": (VRF, "name"),
-  'interface': (None, None)
+    "interface": (None, None),
 }
 
-vm_interface_ct = ContentType.objects.filter(Q(app_label='virtualization', model='vminterface')).first()
+vm_interface_ct = ContentType.objects.filter(
-interface_ct = ContentType.objects.filter(Q(app_label='dcim', model='interface')).first()
+    Q(app_label="virtualization", model="vminterface")
+).first()
+interface_ct = ContentType.objects.filter(Q(app_label="dcim", model="interface")).first()
 
 for params in ip_addresses:
     custom_field_data = pop_custom_fields(params)
 
-  vm = params.pop('virtual_machine', None)
+    vm = params.pop("virtual_machine", None)
-  device = params.pop('device', None)
+    device = params.pop("device", None)
-  params['address'] = IPNetwork(params['address'])
+    params["address"] = IPNetwork(params["address"])
 
     if vm and device:
         print("IP Address can only specify one of the following: virtual_machine or device.")
@@ -37,19 +39,19 @@ for params in ip_addresses:
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      if assoc == 'interface':
+            if assoc == "interface":
                 if vm:
                     vm_id = VirtualMachine.objects.get(name=vm).id
-              query = { 'name': params.pop(assoc), "virtual_machine_id": vm_id }
+                    query = {"name": params.pop(assoc), "virtual_machine_id": vm_id}
-              params['assigned_object_type'] = vm_interface_ct
+                    params["assigned_object_type"] = vm_interface_ct
-              params['assigned_object_id'] = VMInterface.objects.get(**query).id
+                    params["assigned_object_id"] = VMInterface.objects.get(**query).id
                 elif device:
                     dev_id = Device.objects.get(name=device).id
-              query = { 'name': params.pop(assoc), "device_id": dev_id }
+                    query = {"name": params.pop(assoc), "device_id": dev_id}
-              params['assigned_object_type'] = interface_ct
+                    params["assigned_object_type"] = interface_ct
-              params['assigned_object_id'] = Interface.objects.get(**query).id
+                    params["assigned_object_id"] = Interface.objects.get(**query).id
             else:
-          query = { field: params.pop(assoc) }
+                query = {field: params.pop(assoc)}
                 params[assoc] = model.objects.get(**query)
 
     ip_address, created = IPAddress.objects.get_or_create(**params)

startup_scripts/270_primary_ips.py

@@ -1,19 +1,21 @@
+import sys
+
 from dcim.models import Device
 from ipam.models import IPAddress
-from virtualization.models import VirtualMachine
 from startup_script_utils import load_yaml
-import sys
+from virtualization.models import VirtualMachine
+
 
 def link_primary_ip(assets, asset_model):
     for params in assets:
-    primary_ip_fields = set(params) & {'primary_ip4', 'primary_ip6'}
+        primary_ip_fields = set(params) & {"primary_ip4", "primary_ip6"}
         if not primary_ip_fields:
             continue
 
         for assoc, details in optional_assocs.items():
             if assoc in params:
                 model, field = details
-        query = { field: params.pop(assoc) }
+                query = {field: params.pop(assoc)}
 
                 try:
                     params[assoc] = model.objects.get(**query)
@@ -21,19 +23,20 @@ def link_primary_ip(assets, asset_model):
                     primary_ip_fields -= {assoc}
                     print(f"⚠️ IP Address '{query[field]}' not found")
 
-    asset = asset_model.objects.get(name=params['name'])
+        asset = asset_model.objects.get(name=params["name"])
         for field in primary_ip_fields:
             if getattr(asset, field) != params[field]:
                 setattr(asset, field, params[field])
                 print(f"🔗 Define primary IP '{params[field].address}' on '{asset.name}'")
         asset.save()
 
-devices = load_yaml('/opt/netbox/initializers/devices.yml')
+
-virtual_machines = load_yaml('/opt/netbox/initializers/virtual_machines.yml')
+devices = load_yaml("/opt/netbox/initializers/devices.yml")
+virtual_machines = load_yaml("/opt/netbox/initializers/virtual_machines.yml")
 
 optional_assocs = {
-    'primary_ip4': (IPAddress, 'address'),
+    "primary_ip4": (IPAddress, "address"),
-    'primary_ip6': (IPAddress, 'address')
+    "primary_ip6": (IPAddress, "address"),
 }
 
 if devices is None and virtual_machines is None:

startup_scripts/280_custom_links.py

@@ -1,10 +1,10 @@
+import sys
+
 from django.contrib.contenttypes.models import ContentType
 from extras.models import CustomLink
 from startup_script_utils import load_yaml
-import sys
 
-
+custom_links = load_yaml("/opt/netbox/initializers/custom_links.yml")
-custom_links = load_yaml('/opt/netbox/initializers/custom_links.yml')
 
 if custom_links is None:
     sys.exit()
@@ -18,10 +18,14 @@ def get_content_type_id(content_type):
 
 
 for link in custom_links:
-  content_type = link.pop('content_type')
+    content_type = link.pop("content_type")
-  link['content_type_id'] = get_content_type_id(content_type)
+    link["content_type_id"] = get_content_type_id(content_type)
-  if link['content_type_id'] is None:
+    if link["content_type_id"] is None:
-    print("⚠️ Unable to create Custom Link '{0}': The content_type '{1}' is unknown".format(link.name, content_type))
+        print(
+            "⚠️ Unable to create Custom Link '{0}': The content_type '{1}' is unknown".format(
+                link.name, content_type
+            )
+        )
         continue
 
     custom_link, created = CustomLink.objects.get_or_create(**link)

startup_scripts/280_providers.py

@@ -1,8 +1,9 @@
-from circuits.models import Provider
-from startup_script_utils import *
 import sys
 
-providers = load_yaml('/opt/netbox/initializers/providers.yml')
+from circuits.models import Provider
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
+
+providers = load_yaml("/opt/netbox/initializers/providers.yml")
 
 if providers is None:
     sys.exit()

startup_scripts/290_circuit_types.py

@@ -1,8 +1,9 @@
-from circuits.models import CircuitType
-from startup_script_utils import *
 import sys
 
-circuit_types = load_yaml('/opt/netbox/initializers/circuit_types.yml')
+from circuits.models import CircuitType
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
+
+circuit_types = load_yaml("/opt/netbox/initializers/circuit_types.yml")
 
 if circuit_types is None:
     sys.exit()

startup_scripts/290_webhooks.py

@@ -1,10 +1,10 @@
+import sys
+
 from django.contrib.contenttypes.models import ContentType
 from extras.models import Webhook
 from startup_script_utils import load_yaml
-import sys
 
-
+webhooks = load_yaml("/opt/netbox/initializers/webhooks.yml")
-webhooks = load_yaml('/opt/netbox/initializers/webhooks.yml')
 
 if webhooks is None:
     sys.exit()
@@ -19,10 +19,10 @@ def get_content_type_id(hook_name, content_type):
 
 
 for hook in webhooks:
-  obj_types = hook.pop('object_types')
+    obj_types = hook.pop("object_types")
 
     try:
-    obj_type_ids = [get_content_type_id(hook['name'], obj) for obj in obj_types]
+        obj_type_ids = [get_content_type_id(hook["name"], obj) for obj in obj_types]
     except ContentType.DoesNotExist:
         continue
 

startup_scripts/300_circuits.py

@@ -1,35 +1,31 @@
-from circuits.models import Circuit, Provider, CircuitType
-from tenancy.models import Tenant
-from startup_script_utils import *
 import sys
 
-circuits = load_yaml('/opt/netbox/initializers/circuits.yml')
+from circuits.models import Circuit, CircuitType, Provider
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
+from tenancy.models import Tenant
+
+circuits = load_yaml("/opt/netbox/initializers/circuits.yml")
 
 if circuits is None:
     sys.exit()
 
-required_assocs = {
+required_assocs = {"provider": (Provider, "name"), "type": (CircuitType, "name")}
-  'provider': (Provider, 'name'),
-  'type': (CircuitType, 'name')
-}
 
-optional_assocs = {
+optional_assocs = {"tenant": (Tenant, "name")}
-  'tenant': (Tenant, 'name')
-}
 
 for params in circuits:
     custom_field_data = pop_custom_fields(params)
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
 
         params[assoc] = model.objects.get(**query)
 
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/310_secret_roles.py

@@ -1,8 +1,9 @@
-from secrets.models import SecretRole
-from startup_script_utils import load_yaml
 import sys
+from secrets.models import SecretRole
 
-secret_roles = load_yaml('/opt/netbox/initializers/secret_roles.yml')
+from startup_script_utils import load_yaml
+
+secret_roles = load_yaml("/opt/netbox/initializers/secret_roles.yml")
 
 if secret_roles is None:
     sys.exit()

startup_scripts/320_services.py

@@ -1,17 +1,18 @@
-from ipam.models import Service
-from dcim.models import Device
-from virtualization.models import VirtualMachine
-from startup_script_utils import load_yaml
 import sys
 
-services = load_yaml('/opt/netbox/initializers/services.yml')
+from dcim.models import Device
+from ipam.models import Service
+from startup_script_utils import load_yaml
+from virtualization.models import VirtualMachine
+
+services = load_yaml("/opt/netbox/initializers/services.yml")
 
 if services is None:
     sys.exit()
 
 optional_assocs = {
-  'device': (Device, 'name'),
+    "device": (Device, "name"),
-  'virtual_machine': (VirtualMachine, 'name')
+    "virtual_machine": (VirtualMachine, "name"),
 }
 
 for params in services:
@@ -19,7 +20,7 @@ for params in services:
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/330_power_panels.py

@@ -1,35 +1,30 @@
 import sys
 
-from dcim.models import Site, RackGroup, PowerPanel
+from dcim.models import PowerPanel, RackGroup, Site
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
-from tenancy.models import Tenant
 
-power_panels = load_yaml('/opt/netbox/initializers/power_panels.yml')
+power_panels = load_yaml("/opt/netbox/initializers/power_panels.yml")
 
 if power_panels is None:
     sys.exit()
 
-required_assocs = {
+required_assocs = {"site": (Site, "name")}
-  'site': (Site, 'name')
-}
 
-optional_assocs = {
+optional_assocs = {"rack_group": (RackGroup, "name")}
-  'rack_group': (RackGroup, 'name')
-}
 
 for params in power_panels:
     custom_field_data = pop_custom_fields(params)
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
 
         params[assoc] = model.objects.get(**query)
 
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/340_power_feeds.py

@@ -1,35 +1,30 @@
 import sys
 
-from dcim.models import Rack, PowerPanel, PowerFeed
+from dcim.models import PowerFeed, PowerPanel, Rack
-from startup_script_utils import *
+from startup_script_utils import load_yaml, pop_custom_fields, set_custom_fields_values
-from tenancy.models import Tenant
 
-power_feeds = load_yaml('/opt/netbox/initializers/power_feeds.yml')
+power_feeds = load_yaml("/opt/netbox/initializers/power_feeds.yml")
 
 if power_feeds is None:
     sys.exit()
 
-required_assocs = {
+required_assocs = {"power_panel": (PowerPanel, "name")}
-  'power_panel': (PowerPanel, 'name')
-}
 
-optional_assocs = {
+optional_assocs = {"rack": (Rack, "name")}
-  'rack': (Rack, 'name')
-}
 
 for params in power_feeds:
     custom_field_data = pop_custom_fields(params)
 
     for assoc, details in required_assocs.items():
         model, field = details
-    query = { field: params.pop(assoc) }
+        query = {field: params.pop(assoc)}
 
         params[assoc] = model.objects.get(**query)
 
     for assoc, details in optional_assocs.items():
         if assoc in params:
             model, field = details
-      query = { field: params.pop(assoc) }
+            query = {field: params.pop(assoc)}
 
             params[assoc] = model.objects.get(**query)
 

startup_scripts/__main__.py

@@ -2,22 +2,24 @@
 
 import runpy
 from os import scandir
-from os.path import dirname, abspath
+from os.path import abspath, dirname
 
 this_dir = dirname(abspath(__file__))
 
+
 def filename(f):
     return f.name
 
+
 with scandir(this_dir) as it:
-  for f in sorted(it, key = filename):
+    for f in sorted(it, key=filename):
         if not f.is_file():
             continue
 
-    if f.name.startswith('__'):
+        if f.name.startswith("__"):
             continue
 
-    if not f.name.endswith('.py'):
+        if not f.name.endswith(".py"):
             continue
 
         print(f"▶️  Running the startup script {f.path}")

startup_scripts/startup_script_utils/__init__.py

@@ -1,3 +1,3 @@
+from .custom_fields import pop_custom_fields, set_custom_fields_values
 from .load_yaml import load_yaml
 from .permissions import set_permissions
-from .custom_fields import set_custom_fields_values, pop_custom_fields

startup_scripts/startup_script_utils/custom_fields.py

@@ -5,11 +5,12 @@ def set_custom_fields_values(entity, custom_field_data):
     entity.custom_field_data = custom_field_data
     return entity.save()
 
+
 def pop_custom_fields(params):
-  if 'custom_field_data' in params:
+    if "custom_field_data" in params:
-    return params.pop('custom_field_data')
+        return params.pop("custom_field_data")
-  elif 'custom_fields' in params:
+    elif "custom_fields" in params:
         print("⚠️ Please rename 'custom_fields' to 'custom_field_data'!")
-    return params.pop('custom_fields')
+        return params.pop("custom_fields")
 
     return None

startup_scripts/startup_script_utils/load_yaml.py

@@ -1,4 +1,5 @@
 from pathlib import Path
+
 from ruamel.yaml import YAML
 
 

startup_scripts/startup_script_utils/permissions.py

@@ -9,7 +9,11 @@ def set_permissions(subject, permission_filters):
         if "*" in permission_filter:
             permission_filter_regex = "^" + permission_filter.replace("*", ".*") + "$"
             permissions = Permission.objects.filter(codename__iregex=permission_filter_regex)
-      print("  ⚿ Granting", permissions.count(), "permissions matching '" + permission_filter + "'")
+            print(
+                "  ⚿ Granting",
+                permissions.count(),
+                "permissions matching '" + permission_filter + "'",
+            )
         else:
             permissions = Permission.objects.filter(codename=permission_filter)
             print("  ⚿ Granting permission", permission_filter)