mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
122 lines
4.3 KiB
122 lines
4.3 KiB
from django.utils.translation import gettext_lazy as _ |
|
from rest_framework import serializers |
|
|
|
from assets.models import Platform, Account |
|
from assets.serializers import HostSerializer |
|
from common.drf.fields import LabeledChoiceField |
|
from common.validators import ProjectUniqueValidator |
|
from .applet import AppletSerializer |
|
from .. import const |
|
from ..models import AppletHost, AppletHostDeployment |
|
|
|
__all__ = [ |
|
'AppletHostSerializer', 'AppletHostDeploymentSerializer', |
|
'AppletHostAccountSerializer', 'AppletHostAppletReportSerializer', |
|
'AppletHostStartupSerializer', 'AppletHostDeployAppletSerializer' |
|
] |
|
|
|
|
|
class DeployOptionsSerializer(serializers.Serializer): |
|
LICENSE_MODE_CHOICES = ( |
|
(4, _('Per Session')), |
|
(2, _('Per Device')), |
|
) |
|
SESSION_PER_USER = ( |
|
(1, _("Disabled")), |
|
(0, _("Enabled")), |
|
) |
|
RDS_Licensing = serializers.BooleanField(default=False, label=_("RDS Licensing")) |
|
RDS_LicenseServer = serializers.CharField(default='127.0.0.1', label=_('RDS License Server'), max_length=1024) |
|
RDS_LicensingMode = serializers.ChoiceField(choices=LICENSE_MODE_CHOICES, default=4, label=_('RDS Licensing Mode')) |
|
RDS_fSingleSessionPerUser = serializers.ChoiceField(choices=SESSION_PER_USER, default=1, |
|
label=_("RDS fSingleSessionPerUser")) |
|
RDS_MaxDisconnectionTime = serializers.IntegerField(default=60000, label=_("RDS Max Disconnection Time")) |
|
RDS_RemoteAppLogoffTimeLimit = serializers.IntegerField(default=0, label=_("RDS Remote App Logoff Time Limit")) |
|
|
|
|
|
class AppletHostSerializer(HostSerializer): |
|
deploy_options = DeployOptionsSerializer(required=False, label=_("Deploy options")) |
|
load = LabeledChoiceField( |
|
read_only=True, label=_('Load status'), choices=const.ComponentLoad.choices, |
|
) |
|
|
|
class Meta(HostSerializer.Meta): |
|
model = AppletHost |
|
fields = HostSerializer.Meta.fields + [ |
|
'load', 'date_synced', 'deploy_options' |
|
] |
|
extra_kwargs = { |
|
'date_synced': {'read_only': True} |
|
} |
|
|
|
def __init__(self, *args, data=None, **kwargs): |
|
if data: |
|
self.set_initial_data(data) |
|
kwargs['data'] = data |
|
super().__init__(*args, **kwargs) |
|
|
|
@staticmethod |
|
def set_initial_data(data): |
|
platform = Platform.objects.get(name='RemoteAppHost') |
|
data.update({ |
|
'platform': platform.id, |
|
'nodes_display': [ |
|
'RemoteAppHosts' |
|
] |
|
}) |
|
|
|
def get_validators(self): |
|
validators = super().get_validators() |
|
# 不知道为啥没有继承过来 |
|
uniq_validator = ProjectUniqueValidator( |
|
queryset=AppletHost.objects.all(), |
|
fields=('org_id', 'name') |
|
) |
|
validators.append(uniq_validator) |
|
return validators |
|
|
|
|
|
class HostAppletSerializer(AppletSerializer): |
|
publication = serializers.SerializerMethodField() |
|
|
|
class Meta(AppletSerializer.Meta): |
|
fields = AppletSerializer.Meta.fields + ['publication'] |
|
|
|
|
|
class AppletHostDeploymentSerializer(serializers.ModelSerializer): |
|
class Meta: |
|
model = AppletHostDeployment |
|
fields_mini = ['id', 'host', 'status', 'task'] |
|
read_only_fields = [ |
|
'status', 'date_created', 'date_updated', |
|
'date_start', 'date_finished' |
|
] |
|
fields = fields_mini + ['comment'] + read_only_fields |
|
|
|
|
|
class AppletHostDeployAppletSerializer(AppletHostDeploymentSerializer): |
|
applet_id = serializers.UUIDField(write_only=True, allow_null=True, required=False) |
|
|
|
class Meta(AppletHostDeploymentSerializer.Meta): |
|
fields = AppletHostDeploymentSerializer.Meta.fields + ['applet_id'] |
|
|
|
def create(self, validated_data): |
|
applet_id = validated_data.pop('applet_id', None) |
|
deployment = super().create(validated_data) |
|
return deployment |
|
|
|
|
|
class AppletHostAccountSerializer(serializers.ModelSerializer): |
|
class Meta: |
|
model = Account |
|
fields = ['id', 'username', 'secret', 'is_active', 'date_updated'] |
|
|
|
|
|
class AppletHostAppletReportSerializer(serializers.Serializer): |
|
id = serializers.UUIDField(read_only=True) |
|
name = serializers.CharField() |
|
version = serializers.CharField() |
|
|
|
|
|
class AppletHostStartupSerializer(serializers.Serializer): |
|
pass
|
|
|