Mascali/netbox-docker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Update Custom Field logic for Netbox v2.10.x

Browse source
This commit is contained in:
Christian Mäder 2020-10-17 21:51:38 +02:00 • committed by Christian Mäder
parent 036f94a450
commit 5c9bea8b50
24 changed files with 168 additions and 268 deletions
Download patch file Download diff file Expand all files Collapse all files

37
startup_scripts/020_custom_fields.py
View file

@ -1,8 +1,8 @@
from extras.models import CustomField, CustomFieldChoice
from startup_script_utils import load_yaml
import sys
from extras.models import CustomField
from startup_script_utils import load_yaml
def get_class_for_class_path(class_path):
import importlib
from django.contrib.contenttypes.models import ContentType
@ -21,34 +21,37 @@ for cf_name, cf_details in customfields.items():
custom_field, created = CustomField.objects.get_or_create(name = cf_name)
if created:
if cf_details.get('default', 0):
if cf_details.get('default', False):
custom_field.default = cf_details['default']
if cf_details.get('description', 0):
if cf_details.get('description', False):
custom_field.description = cf_details['description']
if cf_details.get('label', 0):
if cf_details.get('label', False):
custom_field.label = cf_details['label']
for object_type in cf_details.get('on_objects', []):
custom_field.obj_type.add(get_class_for_class_path(object_type))
custom_field.content_types.add(get_class_for_class_path(object_type))
if cf_details.get('required', 0):
if cf_details.get('required', False):
custom_field.required = cf_details['required']
if cf_details.get('type', 0):
if cf_details.get('type', False):
custom_field.type = cf_details['type']
if cf_details.get('weight', 0):
if cf_details.get('weight', -1) >= 0:
custom_field.weight = cf_details['weight']
if cf_details.get('choices', False):
custom_field.choices = []
for _, choice_detail in enumerate(cf_details.get('choices', [])):
if isinstance(choice_detail, str):
custom_field.choices.append(choice_detail)
else: # legacy mode
print(f"⚠️ Please migrate the 'choices' of '{cf_name}' to the new format, as 'weight' is no longer supported!")
custom_field.choices.append(choice_detail['value'])
custom_field.save()
for idx, choice_details in enumerate(cf_details.get('choices', [])):
choice, _ = CustomFieldChoice.objects.get_or_create(
field=custom_field,
value=choice_details['value'],
defaults={'weight': idx * 10}
)
print("🔧 Created custom field", cf_name)