netbox-docker/startup_scripts/090_device_roles.py

25 lines
607 B
Python
Raw Normal View History

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()
2018-10-15 15:15:23 +02:00
for params in device_roles:
2018-10-15 15:15:23 +02:00
if "color" in params:
color = params.pop("color")
2018-10-15 15:15:23 +02:00
for color_tpl in ColorChoices:
if color in color_tpl:
params["color"] = color_tpl[0]
2018-10-16 11:05:28 +02:00
device_role, created = DeviceRole.objects.get_or_create(**params)
2018-10-15 15:15:23 +02:00
if created:
print("🎨 Created device role", device_role.name)