fix: 修复获取用户收藏的资产API失败的问题

pull/9090/head
Jiangjie.Bai 2022-11-16 15:07:35 +08:00
parent 0dec61368c
commit 896b59b1bd
2 changed files with 5 additions and 6 deletions

View File

@ -15,7 +15,7 @@ from ...const import Category, AllTypes
__all__ = [
'AssetSerializer', 'AssetSimpleSerializer', 'MiniAssetSerializer',
'AssetTaskSerializer', 'AssetsTaskSerializer',
'AssetTaskSerializer', 'AssetsTaskSerializer', 'AssetProtocolsSerializer',
]

View File

@ -5,6 +5,7 @@ from django.utils.translation import ugettext_lazy as _
from rest_framework import serializers
from assets.const import Category, AllTypes
from assets.serializers.asset.common import AssetProtocolsSerializer
from assets.models import Node, Asset, Platform, Account
from common.drf.fields import ObjectRelatedField, LabeledChoiceField
from perms.serializers.permission import ActionChoicesField
@ -17,17 +18,15 @@ __all__ = [
class AssetGrantedSerializer(serializers.ModelSerializer):
""" 被授权资产的数据结构 """
platform = serializers.SlugRelatedField(
slug_field='name', queryset=Platform.objects.all(), label=_("Platform")
)
protocols = ObjectRelatedField(read_only=True, many=True)
platform = ObjectRelatedField(required=False, queryset=Platform.objects, label=_('Platform'))
protocols = AssetProtocolsSerializer(many=True, required=False, label=_('Protocols'))
category = LabeledChoiceField(choices=Category.choices, read_only=True, label=_('Category'))
type = LabeledChoiceField(choices=AllTypes.choices(), read_only=True, label=_('Type'))
class Meta:
model = Asset
only_fields = [
"id", "name", "address", "protocols",
"id", "name", "address",
'domain', 'platform',
"comment", "org_id", "is_active",
]