perf: 修改 serializer

pull/8873/head
ibuler 2022-09-13 10:41:49 +08:00
parent 6bf7f7cb48
commit 3e794ec41a
2 changed files with 5 additions and 4 deletions

View File

@ -2,10 +2,10 @@ from rest_framework.decorators import action
from rest_framework.response import Response from rest_framework.response import Response
from common.drf.api import JMSModelViewSet from common.drf.api import JMSModelViewSet
from common.drf.serializers import GroupedChoiceSerailizer from common.drf.serializers import GroupedChoiceSerializer
from assets.models import Platform from assets.models import Platform
from assets.serializers import PlatformSerializer, PlatformOpsMethodSerializer from assets.serializers import PlatformSerializer, PlatformOpsMethodSerializer
from assets.const import AllTypes, Category from assets.const import AllTypes
from assets.playbooks import filter_platform_methods from assets.playbooks import filter_platform_methods
@ -16,7 +16,7 @@ class AssetPlatformViewSet(JMSModelViewSet):
queryset = Platform.objects.all() queryset = Platform.objects.all()
serializer_classes = { serializer_classes = {
'default': PlatformSerializer, 'default': PlatformSerializer,
'categories': GroupedChoiceSerailizer 'categories': GroupedChoiceSerializer
} }
filterset_fields = ['name', 'category', 'type'] filterset_fields = ['name', 'category', 'type']
search_fields = ['name'] search_fields = ['name']

View File

@ -14,6 +14,7 @@ __all__ = [
'MethodSerializer', 'EmptySerializer', 'BulkModelSerializer', 'MethodSerializer', 'EmptySerializer', 'BulkModelSerializer',
'AdaptedBulkListSerializer', 'CeleryTaskSerializer', 'AdaptedBulkListSerializer', 'CeleryTaskSerializer',
'SecretReadableMixin', 'JMSWritableNestedModelSerializer', 'SecretReadableMixin', 'JMSWritableNestedModelSerializer',
'GroupedChoiceSerializer',
] ]
@ -90,7 +91,7 @@ class ChoiceSerializer(serializers.Serializer):
value = serializers.CharField(label=_("Value")) value = serializers.CharField(label=_("Value"))
class GroupedChoiceSerailizer(ChoiceSerializer): class GroupedChoiceSerializer(ChoiceSerializer):
children = ChoiceSerializer(many=True, label=_("Children")) children = ChoiceSerializer(many=True, label=_("Children"))