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
127 lines
3.4 KiB
Python
127 lines
3.4 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 vpn import models
|
|
|
|
__all__ = (
|
|
'NestedIKEPolicySerializer',
|
|
'NestedIKEProposalSerializer',
|
|
'NestedIPSecPolicySerializer',
|
|
'NestedIPSecProfileSerializer',
|
|
'NestedIPSecProposalSerializer',
|
|
'NestedL2VPNSerializer',
|
|
'NestedL2VPNTerminationSerializer',
|
|
'NestedTunnelGroupSerializer',
|
|
'NestedTunnelSerializer',
|
|
'NestedTunnelTerminationSerializer',
|
|
)
|
|
|
|
|
|
@extend_schema_serializer(
|
|
exclude_fields=('tunnel_count',),
|
|
)
|
|
class NestedTunnelGroupSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(view_name='vpn-api:tunnelgroup-detail')
|
|
tunnel_count = RelatedObjectCountField('tunnels')
|
|
|
|
class Meta:
|
|
model = models.TunnelGroup
|
|
fields = ['id', 'url', 'display', 'name', 'slug', 'tunnel_count']
|
|
|
|
|
|
class NestedTunnelSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(
|
|
view_name='vpn-api:tunnel-detail'
|
|
)
|
|
|
|
class Meta:
|
|
model = models.Tunnel
|
|
fields = ('id', 'url', 'display', 'name')
|
|
|
|
|
|
class NestedTunnelTerminationSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(
|
|
view_name='vpn-api:tunneltermination-detail'
|
|
)
|
|
|
|
class Meta:
|
|
model = models.TunnelTermination
|
|
fields = ('id', 'url', 'display')
|
|
|
|
|
|
class NestedIKEProposalSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(
|
|
view_name='vpn-api:ikeproposal-detail'
|
|
)
|
|
|
|
class Meta:
|
|
model = models.IKEProposal
|
|
fields = ('id', 'url', 'display', 'name')
|
|
|
|
|
|
class NestedIKEPolicySerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(
|
|
view_name='vpn-api:ikepolicy-detail'
|
|
)
|
|
|
|
class Meta:
|
|
model = models.IKEPolicy
|
|
fields = ('id', 'url', 'display', 'name')
|
|
|
|
|
|
class NestedIPSecProposalSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(
|
|
view_name='vpn-api:ipsecproposal-detail'
|
|
)
|
|
|
|
class Meta:
|
|
model = models.IPSecProposal
|
|
fields = ('id', 'url', 'display', 'name')
|
|
|
|
|
|
class NestedIPSecPolicySerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(
|
|
view_name='vpn-api:ipsecpolicy-detail'
|
|
)
|
|
|
|
class Meta:
|
|
model = models.IPSecPolicy
|
|
fields = ('id', 'url', 'display', 'name')
|
|
|
|
|
|
class NestedIPSecProfileSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(
|
|
view_name='vpn-api:ipsecprofile-detail'
|
|
)
|
|
|
|
class Meta:
|
|
model = models.IPSecProfile
|
|
fields = ('id', 'url', 'display', 'name')
|
|
|
|
|
|
#
|
|
# L2VPN
|
|
#
|
|
|
|
class NestedL2VPNSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(view_name='vpn-api:l2vpn-detail')
|
|
|
|
class Meta:
|
|
model = models.L2VPN
|
|
fields = [
|
|
'id', 'url', 'display', 'identifier', 'name', 'slug', 'type'
|
|
]
|
|
|
|
|
|
class NestedL2VPNTerminationSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(view_name='vpn-api:l2vpntermination-detail')
|
|
l2vpn = NestedL2VPNSerializer()
|
|
|
|
class Meta:
|
|
model = models.L2VPNTermination
|
|
fields = [
|
|
'id', 'url', 'display', 'l2vpn'
|
|
]
|