NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-30 12:26:25 -06:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Namespace overridden Serialzers for swagger gen

Browse Source
This commit is contained in:
Marco Ceppi 2019-12-16 14:58:20 -05:00
parent 9fa125f1a4
commit 635f6569f8
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
netbox/vapor/api/serializers.py
View File

@ -6,6 +6,7 @@ from dcim.constants import CONNECTION_STATUS_CHOICES, IFACE_TYPE_CHOICES
from extras.api.customfields import CustomFieldModelSerializer from extras.api.customfields import CustomFieldModelSerializer
from dcim.api.nested_serializers import ( from dcim.api.nested_serializers import (
NestedDeviceSerializer, NestedDeviceSerializer,
NestedInterfaceSerializer,
NestedCableSerializer, NestedCableSerializer,
) )
from dcim.api.serializers import ( from dcim.api.serializers import (
@ -31,7 +32,7 @@ def get_serializer_for_model(model, prefix=''):
app_name, prefix, model_name app_name, prefix, model_name
) )
override_serializer_name = 'vapor.api.serializers.{}{}Serializer'.format( override_serializer_name = 'vapor.api.serializers.{}VLAN{}Serializer'.format(
prefix, model_name prefix, model_name
) )
try: try:
@ -47,7 +48,7 @@ def get_serializer_for_model(model, prefix=''):
) )
class NestedVLANSerializer(WritableNestedSerializer): class NestedVaporVLANSerializer(WritableNestedSerializer):
url = serializers.HyperlinkedIdentityField(view_name='ipam-api:vlan-detail') url = serializers.HyperlinkedIdentityField(view_name='ipam-api:vlan-detail')
prefixes = SerializedPKRelatedField( prefixes = SerializedPKRelatedField(
@ -62,16 +63,16 @@ class NestedVLANSerializer(WritableNestedSerializer):
fields = ['id', 'url', 'vid', 'name', 'display_name', 'prefixes'] fields = ['id', 'url', 'vid', 'name', 'display_name', 'prefixes']
class NestedInterfaceSerializer(WritableNestedSerializer): class NestedVLANInterfaceSerializer(WritableNestedSerializer):
device = NestedDeviceSerializer(read_only=True) device = NestedDeviceSerializer(read_only=True)
url = serializers.HyperlinkedIdentityField(view_name='dcim-api:interface-detail') url = serializers.HyperlinkedIdentityField(view_name='dcim-api:interface-detail')
connection_status = ChoiceField(choices=CONNECTION_STATUS_CHOICES, read_only=True) connection_status = ChoiceField(choices=CONNECTION_STATUS_CHOICES, read_only=True)
type = ChoiceField(choices=IFACE_TYPE_CHOICES, required=False) type = ChoiceField(choices=IFACE_TYPE_CHOICES, required=False)
untagged_vlan = NestedVLANSerializer(required=False, allow_null=True) untagged_vlan = NestedVaporVLANSerializer(required=False, allow_null=True)
tagged_vlans = SerializedPKRelatedField( tagged_vlans = SerializedPKRelatedField(
queryset=VLAN.objects.all(), queryset=VLAN.objects.all(),
serializer=NestedVLANSerializer, serializer=NestedVaporVLANSerializer,
required=False, required=False,
many=True, many=True,
) )
@ -129,10 +130,10 @@ class InterfaceSerializer(TaggitSerializer, ConnectedEndpointSerializer):
form_factor = ChoiceField(choices=IFACE_TYPE_CHOICES, required=False) form_factor = ChoiceField(choices=IFACE_TYPE_CHOICES, required=False)
lag = NestedInterfaceSerializer(required=False, allow_null=True) lag = NestedInterfaceSerializer(required=False, allow_null=True)
mode = ChoiceField(choices=IFACE_MODE_CHOICES, required=False, allow_null=True) mode = ChoiceField(choices=IFACE_MODE_CHOICES, required=False, allow_null=True)
untagged_vlan = NestedVLANSerializer(required=False, allow_null=True) untagged_vlan = NestedVaporVLANSerializer(required=False, allow_null=True)
tagged_vlans = SerializedPKRelatedField( tagged_vlans = SerializedPKRelatedField(
queryset=VLAN.objects.all(), queryset=VLAN.objects.all(),
serializer=NestedVLANSerializer, serializer=NestedVaporVLANSerializer,
required=False, required=False,
many=True many=True
) )