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)