mirror of https://github.com/jumpserver/jumpserver
71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
|
from django.utils.translation import ugettext_lazy as _
|
||
|
|
from rest_framework import serializers
|
||
|
|
|
||
|
|
from perms.serializers.base import ActionsField
|
||
|
|
from perms.models import AssetPermission
|
||
|
|
from orgs.utils import tmp_to_org
|
||
|
|
|
||
|
|
from tickets.models import ApplyAssetTicket
|
||
|
|
from .ticket import TicketApplySerializer
|
||
|
|
from .common import BaseApplyAssetApplicationSerializer
|
||
|
|
|
||
|
|
__all__ = ['ApplyAssetSerializer', 'ApplyAssetDisplaySerializer']
|
||
|
|
|
||
|
|
asset_or_node_help_text = _("Select at least one asset or node")
|
||
|
|
|
||
|
|
|
||
|
|
class ApplyAssetSerializer(BaseApplyAssetApplicationSerializer, TicketApplySerializer):
|
||
|
|
apply_actions = ActionsField(required=True, allow_null=True)
|
||
|
|
permission_model = AssetPermission
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = ApplyAssetTicket
|
||
|
|
writeable_fields = [
|
||
|
|
'id', 'title', 'type', 'apply_nodes', 'apply_assets',
|
||
|
|
'apply_system_users', 'apply_actions', 'apply_actions_display',
|
||
|
|
'apply_date_start', 'apply_date_expired', 'org_id'
|
||
|
|
]
|
||
|
|
fields = TicketApplySerializer.Meta.fields + writeable_fields + ['apply_permission_name']
|
||
|
|
read_only_fields = list(set(fields) - set(writeable_fields))
|
||
|
|
ticket_extra_kwargs = TicketApplySerializer.Meta.extra_kwargs
|
||
|
|
extra_kwargs = {
|
||
|
|
'apply_nodes': {'required': False, 'help_text': asset_or_node_help_text},
|
||
|
|
'apply_assets': {'required': False, 'help_text': asset_or_node_help_text}
|
||
|
|
}
|
||
|
|
extra_kwargs.update(ticket_extra_kwargs)
|
||
|
|
|
||
|
|
def validate(self, attrs):
|
||
|
|
attrs = super().validate(attrs)
|
||
|
|
if not attrs.get('apply_nodes') and not attrs.get('apply_assets'):
|
||
|
|
raise serializers.ValidationError({
|
||
|
|
'apply_nodes': asset_or_node_help_text,
|
||
|
|
'apply_assets': asset_or_node_help_text,
|
||
|
|
})
|
||
|
|
return attrs
|
||
|
|
|
||
|
|
|
||
|
|
class ApplyAssetDisplaySerializer(ApplyAssetSerializer):
|
||
|
|
apply_nodes = serializers.SerializerMethodField()
|
||
|
|
apply_assets = serializers.SerializerMethodField()
|
||
|
|
apply_system_users = serializers.SerializerMethodField()
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = ApplyAssetSerializer.Meta.model
|
||
|
|
fields = ApplyAssetSerializer.Meta.fields
|
||
|
|
read_only_fields = fields
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def get_apply_nodes(instance):
|
||
|
|
with tmp_to_org(instance.org_id):
|
||
|
|
return instance.apply_nodes.values_list('id', flat=True)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def get_apply_assets(instance):
|
||
|
|
with tmp_to_org(instance.org_id):
|
||
|
|
return instance.apply_assets.values_list('id', flat=True)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def get_apply_system_users(instance):
|
||
|
|
with tmp_to_org(instance.org_id):
|
||
|
|
return instance.apply_system_users.values_list('id', flat=True)
|