10711 Add Scope to WirelessLAN

This commit is contained in:
Arthur Hanson 2024-10-28 14:12:34 -07:00
parent c91c9e5fe0
commit 8bafef654a
2 changed files with 10 additions and 1 deletions

View File

@ -1,5 +1,6 @@
from rest_framework import serializers
from drf_spectacular.utils import extend_schema_field
from ipam.api.serializers_.vlans import VLANSerializer
from netbox.api.fields import ChoiceField
from netbox.api.serializers import NestedGroupModelSerializer, NetBoxModelSerializer
@ -45,3 +46,11 @@ class WirelessLANSerializer(NetBoxModelSerializer):
'created', 'last_updated',
]
brief_fields = ('id', 'url', 'display', 'ssid', 'description')
@extend_schema_field(serializers.JSONField(allow_null=True))
def get_scope(self, obj):
if obj.scope_id is None:
return None
serializer = get_serializer_for_model(obj.scope)
context = {'request': self.context['request']}
return serializer(obj.scope, nested=True, context=context).data

View File

@ -28,7 +28,7 @@ class WirelessLANGroupType(OrganizationalObjectType):
@strawberry_django.type(
models.WirelessLAN,
fields='__all__',
exclude=('scope_type', 'scope_id', '_location', '_region', '_site', '_sitegroup'),
filters=WirelessLANFilter
)
class WirelessLANType(NetBoxObjectType):