Cleanup & Reorg startup scripts

This commit is contained in:
ryanmerolle 2022-01-30 00:42:50 -05:00
parent ff20e4f49c
commit c93e0e5a6b
41 changed files with 0 additions and 57 deletions

View file

@ -0,0 +1,24 @@
import sys
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:
color = params.pop("color")
for color_tpl in ColorChoices:
if color in color_tpl:
params["color"] = color_tpl[0]
device_role, created = DeviceRole.objects.get_or_create(**params)
if created:
print("🎨 Created device role", device_role.name)