from extras.constants import CF_TYPE_TEXT, CF_TYPE_INTEGER, CF_TYPE_BOOLEAN, CF_TYPE_DATE, CF_TYPE_URL, CF_TYPE_SELECT from extras.models import CustomField, CustomFieldChoice from ruamel.yaml import YAML text_to_fields = { 'boolean': CF_TYPE_BOOLEAN, 'date': CF_TYPE_DATE, 'integer': CF_TYPE_INTEGER, 'selection': CF_TYPE_SELECT, 'text': CF_TYPE_TEXT, 'url': CF_TYPE_URL, } 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) with open('/opt/netbox/initializers/custom_fields.yml', 'r') as stream: yaml = YAML(typ='safe') customfields = yaml.load(stream) if customfields is not None: 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): custom_field.default = cf_details['default'] if cf_details.get('description', 0): custom_field.description = cf_details['description'] # If no filter_logic is specified then it will default to 'Loose' if cf_details.get('filter_logic', 0): for choice_id, choice_text in CF_FILTER_CHOICES: if choice_text.lower() == cf_details['filter_logic']: custom_field.filter_logic = choice_id if cf_details.get('label', 0): 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)) if cf_details.get('required', 0): custom_field.required = cf_details['required'] if cf_details.get('type', 0): custom_field.type = text_to_fields[cf_details['type']] if cf_details.get('weight', 0): custom_field.weight = cf_details['weight'] custom_field.save() for choice_details in cf_details.get('choices', []): choice = CustomFieldChoice.objects.create( field=custom_field, value=choice_details['value']) if choice_details.get('weight', 0): choice.weight = choice_details['weight'] choice.save() print("🔧 Created custom field", cf_name)