diff --git a/startup_scripts/070_rack_roles.py b/startup_scripts/070_rack_roles.py index 98a5be4..f5aa2ce 100644 --- a/startup_scripts/070_rack_roles.py +++ b/startup_scripts/070_rack_roles.py @@ -1,5 +1,5 @@ from dcim.models import RackRole -from utilities.forms import COLOR_CHOICES +from utilities.choices import ColorChoices from startup_script_utils import load_yaml import sys @@ -13,7 +13,7 @@ for params in rack_roles: if 'color' in params: color = params.pop('color') - for color_tpl in COLOR_CHOICES: + for color_tpl in ColorChoices: if color in color_tpl: params['color'] = color_tpl[0] diff --git a/startup_scripts/090_device_roles.py b/startup_scripts/090_device_roles.py index c75253d..5225cc0 100644 --- a/startup_scripts/090_device_roles.py +++ b/startup_scripts/090_device_roles.py @@ -1,5 +1,6 @@ from dcim.models import DeviceRole -from utilities.forms import COLOR_CHOICES +from utilities.choices import ColorChoices + from startup_script_utils import load_yaml import sys @@ -13,7 +14,7 @@ for params in device_roles: if 'color' in params: color = params.pop('color') - for color_tpl in COLOR_CHOICES: + for color_tpl in ColorChoices: if color in color_tpl: params['color'] = color_tpl[0]