65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
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
|
|
|
|
module_name, class_name = class_path.rsplit(".", 1)
|
|
module = importlib.import_module(module_name)
|
|
clazz = getattr(module, class_name)
|
|
return ContentType.objects.get_for_model(clazz)
|
|
|
|
|
|
customfields = load_yaml("/opt/netbox/initializers/custom_fields.yml")
|
|
|
|
if customfields is None:
|
|
sys.exit()
|
|
|
|
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", False):
|
|
custom_field.default = cf_details["default"]
|
|
|
|
if cf_details.get("description", False):
|
|
custom_field.description = cf_details["description"]
|
|
|
|
if cf_details.get("label", False):
|
|
custom_field.label = cf_details["label"]
|
|
|
|
for object_type in cf_details.get("on_objects", []):
|
|
custom_field.content_types.add(get_class_for_class_path(object_type))
|
|
|
|
if cf_details.get("required", False):
|
|
custom_field.required = cf_details["required"]
|
|
|
|
if cf_details.get("type", False):
|
|
custom_field.type = cf_details["type"]
|
|
|
|
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 cf_details.get("choices", []):
|
|
if isinstance(choice_detail, dict) and "value" in choice_detail:
|
|
# legacy mode
|
|
print(
|
|
f"⚠️ Please migrate the choice '{choice_detail['value']}' of '{cf_name}'"
|
|
+ " to the new format, as 'weight' is no longer supported!"
|
|
)
|
|
custom_field.choices.append(choice_detail["value"])
|
|
else:
|
|
custom_field.choices.append(choice_detail)
|
|
|
|
custom_field.save()
|
|
|
|
print("🔧 Created custom field", cf_name)
|