netbox-docker/startup_scripts/46_device_roles.py

23 lines
630 B
Python
Raw Normal View History

2018-10-15 15:15:23 +02:00
from dcim.models import DeviceRole
from ruamel.yaml import YAML
from utilities.forms import COLOR_CHOICES
with open('/opt/netbox/initializers/device_roles.yml', 'r') as stream:
yaml=YAML(typ='safe')
device_roles = yaml.load(stream)
if device_roles is not None:
2018-10-16 11:05:28 +02:00
for params in device_roles:
2018-10-15 15:15:23 +02:00
2018-10-16 11:05:28 +02:00
if 'color' in params:
color = params.pop('color')
2018-10-15 15:15:23 +02:00
2018-10-16 11:05:28 +02:00
for color_tpl in COLOR_CHOICES:
if color in color_tpl:
params['color'] = color_tpl[0]
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)