netbox-docker/startup_scripts/070_rack_roles.py

from dcim.models import RackRole
from utilities.forms import COLOR_CHOICES

from startup_script_utils import load_yaml
import sys

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:
    color = params.pop('color')

    for color_tpl in COLOR_CHOICES:
      if color in color_tpl:
        params['color'] = color_tpl[0]

  rack_role, created = RackRole.objects.get_or_create(**params)

  if created:
    print("🎨 Created rack role", rack_role.name)
Add rack role seeds 2018-10-15 08:14:42 -05:00			`from dcim.models import RackRole`
			`from utilities.forms import COLOR_CHOICES`

Add function to load YAML files This commit starts to remove some code redundancy from the startup scripts for easier maintenance. 2020-02-05 08:31:01 -06:00			`from startup_script_utils import load_yaml`
Ignore missing initializers Previously, the startup_scripts would fail if an initializer file was not present. Now they just ignore missing files. 2018-12-19 07:25:58 -06:00			`import sys`

Add function to load YAML files This commit starts to remove some code redundancy from the startup scripts for easier maintenance. 2020-02-05 08:31:01 -06:00			`rack_roles = load_yaml('/opt/netbox/initializers/rack_roles.yml')`
Ignore missing initializers Previously, the startup_scripts would fail if an initializer file was not present. Now they just ignore missing files. 2018-12-19 07:25:58 -06:00
Add function to load YAML files This commit starts to remove some code redundancy from the startup scripts for easier maintenance. 2020-02-05 08:31:01 -06:00			`if rack_roles is None:`
			`sys.exit()`
Add rack role seeds 2018-10-15 08:14:42 -05:00
Add function to load YAML files This commit starts to remove some code redundancy from the startup scripts for easier maintenance. 2020-02-05 08:31:01 -06:00			`for params in rack_roles:`
			`if 'color' in params:`
			`color = params.pop('color')`
Add rack role seeds 2018-10-15 08:14:42 -05:00
Add function to load YAML files This commit starts to remove some code redundancy from the startup scripts for easier maintenance. 2020-02-05 08:31:01 -06:00			`for color_tpl in COLOR_CHOICES:`
			`if color in color_tpl:`
			`params['color'] = color_tpl[0]`
Add rack role seeds 2018-10-15 08:14:42 -05:00
Add function to load YAML files This commit starts to remove some code redundancy from the startup scripts for easier maintenance. 2020-02-05 08:31:01 -06:00			`rack_role, created = RackRole.objects.get_or_create(**params)`
Add rack role seeds 2018-10-15 08:14:42 -05:00
Add function to load YAML files This commit starts to remove some code redundancy from the startup scripts for easier maintenance. 2020-02-05 08:31:01 -06:00			`if created:`
			`print("🎨 Created rack role", rack_role.name)`