mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
7abb2b2ab5
* Introduce RelatedObjectCountField * Introduce get_annotations_for_serializer() and enable dynamic annotations * Add RelatedObjectCountFields to serializers; remove static annotations from querysets * Remove annotations cleanup logic from BriefModeMixin * Annotate type for RelatedObjectCountField * Remove redundant field on TagSerializer * Add missing reverse relationship for power feeds to rack * Refactor RelatedObjectCountField to take a single relationship name
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
from drf_spectacular.utils import extend_schema_serializer
|
|
from rest_framework import serializers
|
|
|
|
from netbox.api.fields import RelatedObjectCountField
|
|
from netbox.api.serializers import WritableNestedSerializer
|
|
from virtualization.models import *
|
|
|
|
__all__ = [
|
|
'NestedClusterGroupSerializer',
|
|
'NestedClusterSerializer',
|
|
'NestedClusterTypeSerializer',
|
|
'NestedVirtualDiskSerializer',
|
|
'NestedVMInterfaceSerializer',
|
|
'NestedVirtualMachineSerializer',
|
|
]
|
|
|
|
#
|
|
# Clusters
|
|
#
|
|
|
|
|
|
@extend_schema_serializer(
|
|
exclude_fields=('cluster_count',),
|
|
)
|
|
class NestedClusterTypeSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(view_name='virtualization-api:clustertype-detail')
|
|
cluster_count = RelatedObjectCountField('clusters')
|
|
|
|
class Meta:
|
|
model = ClusterType
|
|
fields = ['id', 'url', 'display', 'name', 'slug', 'cluster_count']
|
|
|
|
|
|
@extend_schema_serializer(
|
|
exclude_fields=('cluster_count',),
|
|
)
|
|
class NestedClusterGroupSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(view_name='virtualization-api:clustergroup-detail')
|
|
cluster_count = RelatedObjectCountField('clusters')
|
|
|
|
class Meta:
|
|
model = ClusterGroup
|
|
fields = ['id', 'url', 'display', 'name', 'slug', 'cluster_count']
|
|
|
|
|
|
@extend_schema_serializer(
|
|
exclude_fields=('virtualmachine_count',),
|
|
)
|
|
class NestedClusterSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(view_name='virtualization-api:cluster-detail')
|
|
virtualmachine_count = RelatedObjectCountField('virtual_machines')
|
|
|
|
class Meta:
|
|
model = Cluster
|
|
fields = ['id', 'url', 'display', 'name', 'virtualmachine_count']
|
|
|
|
|
|
#
|
|
# Virtual machines
|
|
#
|
|
|
|
class NestedVirtualMachineSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(view_name='virtualization-api:virtualmachine-detail')
|
|
|
|
class Meta:
|
|
model = VirtualMachine
|
|
fields = ['id', 'url', 'display', 'name']
|
|
|
|
|
|
class NestedVMInterfaceSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(view_name='virtualization-api:vminterface-detail')
|
|
virtual_machine = NestedVirtualMachineSerializer(read_only=True)
|
|
|
|
class Meta:
|
|
model = VMInterface
|
|
fields = ['id', 'url', 'display', 'virtual_machine', 'name']
|
|
|
|
|
|
class NestedVirtualDiskSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(view_name='virtualization-api:virtualdisk-detail')
|
|
virtual_machine = NestedVirtualMachineSerializer(read_only=True)
|
|
|
|
class Meta:
|
|
model = VirtualDisk
|
|
fields = ['id', 'url', 'display', 'virtual_machine', 'name', 'size']
|