mirror of
https://github.com/netbox-community/netbox.git
synced 2025-07-18 21:16:27 -06:00
1508e3a770
Made DeviceRoles hierarchical, had to also change the filtersets for Device, ConfigContext and VirtualMachine to use the TreeNodeMultipleChoiceFilter. Note: The model was changed to use NestedGroupModel, a side-effect of this is it also adds comments field, but I thought that was better then doing a one-off just for DeviceRole and having to define the fields, validators, etc.. - keeps everything DRY / consistent. * 18981 Make Device Roles Hierarchical * 18981 forms, serializer * 18981 fix tests * 18981 fix tests * 18981 fix tests * 18981 fix tests * 18981 fix tests * 18981 fix migration merge * 18981 fix tests * 18981 fix filtersets * 18981 fix tests * 18981 comments * 18981 review changes
45 lines
1.8 KiB
Python
45 lines
1.8 KiB
Python
from dcim.models import DeviceRole, InventoryItemRole
|
|
from extras.api.serializers_.configtemplates import ConfigTemplateSerializer
|
|
from netbox.api.fields import RelatedObjectCountField
|
|
from netbox.api.serializers import NestedGroupModelSerializer, NetBoxModelSerializer
|
|
from .nested import NestedDeviceRoleSerializer
|
|
|
|
__all__ = (
|
|
'DeviceRoleSerializer',
|
|
'InventoryItemRoleSerializer',
|
|
)
|
|
|
|
|
|
class DeviceRoleSerializer(NestedGroupModelSerializer):
|
|
parent = NestedDeviceRoleSerializer(required=False, allow_null=True, default=None)
|
|
config_template = ConfigTemplateSerializer(nested=True, required=False, allow_null=True, default=None)
|
|
|
|
# Related object counts
|
|
device_count = RelatedObjectCountField('devices')
|
|
virtualmachine_count = RelatedObjectCountField('virtual_machines')
|
|
|
|
class Meta:
|
|
model = DeviceRole
|
|
fields = [
|
|
'id', 'url', 'display_url', 'display', 'name', 'slug', 'color', 'vm_role', 'config_template', 'parent',
|
|
'description', 'tags', 'custom_fields', 'created', 'last_updated', 'device_count', 'virtualmachine_count',
|
|
'comments', '_depth',
|
|
]
|
|
brief_fields = (
|
|
'id', 'url', 'display', 'name', 'slug', 'description', 'device_count', 'virtualmachine_count', '_depth'
|
|
)
|
|
|
|
|
|
class InventoryItemRoleSerializer(NetBoxModelSerializer):
|
|
|
|
# Related object counts
|
|
inventoryitem_count = RelatedObjectCountField('inventory_items')
|
|
|
|
class Meta:
|
|
model = InventoryItemRole
|
|
fields = [
|
|
'id', 'url', 'display_url', 'display', 'name', 'slug', 'color', 'description', 'tags', 'custom_fields',
|
|
'created', 'last_updated', 'inventoryitem_count',
|
|
]
|
|
brief_fields = ('id', 'url', 'display', 'name', 'slug', 'description', 'inventoryitem_count')
|