mirror of https://github.com/jumpserver/jumpserver
58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from rest_framework import serializers
|
|
|
|
from assets.const import Category, AllTypes
|
|
from assets.models import Node, Asset, Platform, Account
|
|
from common.drf.fields import ObjectRelatedField, LabeledChoiceField
|
|
from perms.serializers.permission import ActionChoicesField
|
|
|
|
__all__ = [
|
|
'NodeGrantedSerializer', 'AssetGrantedSerializer',
|
|
'ActionsSerializer', 'AccountsGrantedSerializer'
|
|
]
|
|
|
|
|
|
class AssetGrantedSerializer(serializers.ModelSerializer):
|
|
""" 被授权资产的数据结构 """
|
|
platform = serializers.SlugRelatedField(
|
|
slug_field='name', queryset=Platform.objects.all(), label=_("Platform")
|
|
)
|
|
protocols = ObjectRelatedField(read_only=True, many=True)
|
|
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",
|
|
'domain', 'platform',
|
|
"comment", "org_id", "is_active",
|
|
]
|
|
fields = only_fields + ['category', 'type'] + ['org_name']
|
|
read_only_fields = fields
|
|
|
|
|
|
class NodeGrantedSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = Node
|
|
fields = [
|
|
'id', 'name', 'key', 'value', 'org_id', "assets_amount"
|
|
]
|
|
read_only_fields = fields
|
|
|
|
|
|
class ActionsSerializer(serializers.Serializer):
|
|
actions = ActionChoicesField(read_only=True)
|
|
|
|
|
|
class AccountsGrantedSerializer(serializers.ModelSerializer):
|
|
actions = ActionChoicesField(read_only=True)
|
|
|
|
class Meta:
|
|
model = Account
|
|
fields = ['id', 'name', 'username', 'secret_type', 'has_secret', 'actions']
|
|
read_only_fields = fields
|