jumpserver/apps/assets/serializers/account/template.py

from common.drf.serializers import SecretReadableMixin
from assets.models import AccountTemplate
from .base import BaseAccountSerializer


class AccountTemplateSerializer(BaseAccountSerializer):
    class Meta(BaseAccountSerializer.Meta):
        model = AccountTemplate

    # @classmethod
    # def validate_required(cls, attrs):
    #     # TODO 选择模版后检查一些必填项
    #     required_field_dict = {}
    #     error = _('This field is required.')
    #     for k, v in cls().fields.items():
    #         if v.required and k not in attrs:
    #             required_field_dict[k] = error
    #     if not required_field_dict:
    #         return
    #     raise serializers.ValidationError(required_field_dict)


class AccountTemplateSecretSerializer(SecretReadableMixin, AccountTemplateSerializer):
    class Meta(AccountTemplateSerializer.Meta):
        extra_kwargs = {
            'secret': {'write_only': False},
        }