netbox-docker/startup_scripts/130_rack_roles.py

25 lines
683 B
Python
Raw Normal View History

import sys
2018-10-15 15:14:42 +02:00
from dcim.models import RackRole
2022-04-05 08:34:08 +02:00
from startup_script_utils import load_yaml, split_params
from utilities.choices import ColorChoices
rack_roles = load_yaml("/opt/netbox/initializers/rack_roles.yml")
if rack_roles is None:
sys.exit()
2018-10-15 15:14:42 +02:00
for params in rack_roles:
if "color" in params:
color = params.pop("color")
2018-10-15 15:14:42 +02:00
for color_tpl in ColorChoices:
if color in color_tpl:
params["color"] = color_tpl[0]
2018-10-15 15:14:42 +02:00
2022-04-05 08:34:08 +02:00
matching_params, defaults = split_params(params)
rack_role, created = RackRole.objects.get_or_create(**matching_params, defaults=defaults)
2018-10-15 15:14:42 +02:00
if created:
print("🎨 Created rack role", rack_role.name)