mirror of https://github.com/jumpserver/jumpserver
131 lines
4.6 KiB
Python
131 lines
4.6 KiB
Python
from django.conf import settings
|
|
from django.utils.translation import gettext_lazy as _
|
|
from rest_framework import serializers
|
|
|
|
from accounts.models import Account
|
|
from assets.models import Platform
|
|
from assets.serializers import HostSerializer
|
|
from common.const.choices import Status
|
|
from common.serializers.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")),
|
|
)
|
|
|
|
CORE_HOST = serializers.CharField(default=settings.SITE_URL, label=_('API Server'), max_length=1024)
|
|
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 Single Session Per User"))
|
|
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 = {
|
|
**HostSerializer.Meta.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):
|
|
status = LabeledChoiceField(choices=Status.choices, label=_('Status'), default=Status.pending)
|
|
|
|
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
|