mirror of https://github.com/jumpserver/jumpserver
106 lines
3.3 KiB
Python
106 lines
3.3 KiB
Python
from django.utils.translation import ugettext_lazy as _
|
|
from rest_framework import serializers
|
|
|
|
from acls.models import CommandGroup
|
|
from assets.models import Asset, Account, Platform
|
|
from assets.serializers import PlatformSerializer, AssetProtocolsSerializer
|
|
from authentication.models import ConnectionToken
|
|
from orgs.mixins.serializers import OrgResourceModelSerializerMixin
|
|
from perms.serializers.permission import ActionChoicesField
|
|
from users.models import User
|
|
|
|
__all__ = [
|
|
'ConnectionTokenSecretSerializer',
|
|
]
|
|
|
|
|
|
class _ConnectionTokenUserSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = User
|
|
fields = ['id', 'name', 'username', 'email']
|
|
|
|
|
|
class _ConnectionTokenAssetSerializer(serializers.ModelSerializer):
|
|
protocols = AssetProtocolsSerializer(many=True, required=False, label=_('Protocols'))
|
|
|
|
class Meta:
|
|
model = Asset
|
|
fields = [
|
|
'id', 'name', 'address', 'protocols',
|
|
'category', 'type', 'org_id', 'specific'
|
|
]
|
|
|
|
|
|
class _SimpleAccountSerializer(serializers.ModelSerializer):
|
|
""" Account """
|
|
|
|
class Meta:
|
|
model = Account
|
|
fields = ['name', 'username', 'secret_type', 'secret']
|
|
|
|
|
|
class _ConnectionTokenAccountSerializer(serializers.ModelSerializer):
|
|
""" Account """
|
|
su_from = _SimpleAccountSerializer(required=False, label=_('Su from'))
|
|
|
|
class Meta:
|
|
model = Account
|
|
fields = [
|
|
'name', 'username', 'secret_type', 'secret', 'su_from',
|
|
]
|
|
|
|
|
|
class _ConnectionTokenGatewaySerializer(serializers.ModelSerializer):
|
|
""" Gateway """
|
|
|
|
class Meta:
|
|
model = Asset
|
|
fields = [
|
|
'id', 'address', 'port',
|
|
# 'username', 'password', 'private_key'
|
|
]
|
|
|
|
|
|
class _ConnectionTokenACLCmdGroupSerializer(serializers.ModelSerializer):
|
|
""" ACL command group"""
|
|
|
|
class Meta:
|
|
model = CommandGroup
|
|
fields = [
|
|
'id', 'type', 'content', 'ignore_case', 'pattern'
|
|
]
|
|
|
|
|
|
class _ConnectionTokenPlatformSerializer(PlatformSerializer):
|
|
class Meta(PlatformSerializer.Meta):
|
|
model = Platform
|
|
|
|
def get_field_names(self, declared_fields, info):
|
|
names = super().get_field_names(declared_fields, info)
|
|
names = [n for n in names if n not in ['automation']]
|
|
return names
|
|
|
|
|
|
class ConnectionTokenSecretSerializer(OrgResourceModelSerializerMixin):
|
|
user = _ConnectionTokenUserSerializer(read_only=True)
|
|
asset = _ConnectionTokenAssetSerializer(read_only=True)
|
|
account = _ConnectionTokenAccountSerializer(read_only=True)
|
|
gateway = _ConnectionTokenGatewaySerializer(read_only=True)
|
|
platform = _ConnectionTokenPlatformSerializer(read_only=True)
|
|
acl_command_groups = _ConnectionTokenACLCmdGroupSerializer(read_only=True, many=True)
|
|
actions = ActionChoicesField()
|
|
expire_at = serializers.IntegerField()
|
|
expire_now = serializers.BooleanField(label=_('Expired now'), write_only=True, default=True)
|
|
connect_method = serializers.CharField(label=_('Connect method'), write_only=True, default='ssh')
|
|
|
|
class Meta:
|
|
model = ConnectionToken
|
|
fields = [
|
|
'id', 'value', 'user', 'asset', 'account',
|
|
'platform', 'acl_command_groups', 'protocol',
|
|
'gateway', 'actions', 'expire_at', 'expire_now',
|
|
]
|
|
extra_kwargs = {
|
|
'value': {'read_only': True},
|
|
}
|