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

# -*- coding: utf-8 -*-
#
from rest_framework import serializers

__all__ = ['AccountFieldsSerializerMixin']


class AccountFieldsSerializerMixin(serializers.ModelSerializer):
    class Meta:
        fields_mini = [
            'id', 'name', 'username', 'privileged', 'ip',
            'asset_name', 'platform', 'version'
        ]
        fields_write_only = ['password', 'private_key', 'public_key', 'passphrase']
        fields_other = ['date_created', 'date_updated', 'comment']
        fields_small = fields_mini + fields_write_only + fields_other
        fields_fk = ['asset']
        fields = fields_small + fields_fk
        extra_kwargs = {
            'private_key': {'write_only': True},
            'public_key': {'write_only': True},
            'token': {'write_only': True},
            'password': {'write_only': True},
        }

    def validate_name(self, value):
        if not value:
            return self.initial_data.get('username')
        return ''