mirror of https://github.com/jumpserver/jumpserver
88 lines
3.3 KiB
Python
88 lines
3.3 KiB
Python
from django.db.models import F
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from rest_framework import serializers
|
|
|
|
from orgs.mixins.serializers import BulkOrgResourceModelSerializer
|
|
from common.drf.serializers import SecretReadableMixin
|
|
from assets.models import Account, AccountTemplate
|
|
from assets.serializers.base import AuthValidateMixin
|
|
from .common import AccountFieldsSerializerMixin
|
|
|
|
|
|
class AccountSerializerCreateMixin(serializers.ModelSerializer):
|
|
template = serializers.UUIDField(
|
|
required=False, allow_null=True, write_only=True,
|
|
label=_('Account template')
|
|
)
|
|
push_to_asset = serializers.BooleanField(default=False, label=_("Push to asset"), write_only=True)
|
|
|
|
@staticmethod
|
|
def validate_template(value):
|
|
AccountTemplate.objects.get_or_create()
|
|
model = AccountTemplate
|
|
try:
|
|
return model.objects.get(id=value)
|
|
except AccountTemplate.DoesNotExist:
|
|
raise serializers.ValidationError(_('Account template not found'))
|
|
|
|
@staticmethod
|
|
def replace_attrs(account_template: AccountTemplate, attrs: dict):
|
|
exclude_fields = [
|
|
'_state', 'org_id', 'date_verified', 'id',
|
|
'date_created', 'date_updated', 'created_by'
|
|
]
|
|
template_attrs = {k: v for k, v in account_template.__dict__.items() if k not in exclude_fields}
|
|
for k, v in template_attrs.items():
|
|
attrs.setdefault(k, v)
|
|
|
|
def validate(self, attrs):
|
|
account_template = attrs.pop('template', None)
|
|
if account_template:
|
|
self.replace_attrs(account_template, attrs)
|
|
push_to_asset = attrs.pop('push_to_asset', False)
|
|
return super().validate(attrs)
|
|
|
|
|
|
class AccountSerializer(AuthValidateMixin,
|
|
AccountSerializerCreateMixin,
|
|
AccountFieldsSerializerMixin,
|
|
BulkOrgResourceModelSerializer):
|
|
name = serializers.CharField(max_length=128, read_only=True, label=_("Name"))
|
|
ip = serializers.ReadOnlyField(label=_("IP"))
|
|
asset_name = serializers.ReadOnlyField(label=_("Asset"))
|
|
platform = serializers.ReadOnlyField(label=_("Platform"))
|
|
|
|
class Meta(AccountFieldsSerializerMixin.Meta):
|
|
model = Account
|
|
fields = AccountFieldsSerializerMixin.Meta.fields \
|
|
+ ['template', 'push_to_asset']
|
|
|
|
@classmethod
|
|
def setup_eager_loading(cls, queryset):
|
|
""" Perform necessary eager loading of data. """
|
|
queryset = queryset.prefetch_related('asset') \
|
|
.annotate(ip=F('asset__ip')) \
|
|
.annotate(asset_name=F('asset__name'))
|
|
return queryset
|
|
|
|
|
|
class AccountSecretSerializer(SecretReadableMixin, AccountSerializer):
|
|
class Meta(AccountSerializer.Meta):
|
|
fields_backup = [
|
|
'name', 'ip', 'platform', 'protocols', 'username', 'password',
|
|
'private_key', 'public_key', 'date_created', 'date_updated', 'version'
|
|
]
|
|
extra_kwargs = {
|
|
'password': {'write_only': False},
|
|
'private_key': {'write_only': False},
|
|
'public_key': {'write_only': False},
|
|
}
|
|
|
|
|
|
class AccountTaskSerializer(serializers.Serializer):
|
|
ACTION_CHOICES = (
|
|
('test', 'test'),
|
|
)
|
|
action = serializers.ChoiceField(choices=ACTION_CHOICES, write_only=True)
|
|
task = serializers.CharField(read_only=True)
|