mirror of https://github.com/jumpserver/jumpserver
85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from rest_framework import serializers
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from common.drf.fields import LabeledChoiceField
|
|
from orgs.models import Organization
|
|
from orgs.mixins.serializers import OrgResourceModelSerializerMixin
|
|
from tickets.models import Ticket, TicketFlow
|
|
from tickets.const import TicketType, TicketStatus, TicketState
|
|
|
|
__all__ = [
|
|
'TicketApplySerializer', 'TicketApproveSerializer', 'TicketSerializer',
|
|
]
|
|
|
|
|
|
class TicketSerializer(OrgResourceModelSerializerMixin):
|
|
type = LabeledChoiceField(choices=TicketType.choices, read_only=True, label=_('Type'))
|
|
status = LabeledChoiceField(choices=TicketStatus.choices, read_only=True, label=_('Status'))
|
|
state = LabeledChoiceField(choices=TicketState.choices, read_only=True, label=_("State"))
|
|
|
|
class Meta:
|
|
model = Ticket
|
|
fields_mini = ['id', 'title']
|
|
fields_small = fields_mini + ['org_id', 'comment']
|
|
read_only_fields = [
|
|
'serial_num', 'process_map', 'approval_step', 'type', 'state', 'applicant',
|
|
'status', 'date_created', 'date_updated', 'org_name', 'rel_snapshot'
|
|
]
|
|
fields = fields_small + read_only_fields
|
|
extra_kwargs = {
|
|
'type': {'required': True}
|
|
}
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.set_type_choices()
|
|
|
|
def set_type_choices(self):
|
|
tp = self.fields.get('type')
|
|
if not tp:
|
|
return
|
|
choices = tp._choices
|
|
choices.pop(TicketType.general, None)
|
|
tp._choices = choices
|
|
|
|
@classmethod
|
|
def setup_eager_loading(cls, queryset):
|
|
queryset = queryset.prefetch_related('ticket_steps')
|
|
return queryset
|
|
|
|
|
|
class TicketApproveSerializer(TicketSerializer):
|
|
class Meta(TicketSerializer.Meta):
|
|
fields = TicketSerializer.Meta.fields
|
|
read_only_fields = fields
|
|
|
|
|
|
class TicketApplySerializer(TicketSerializer):
|
|
org_id = serializers.CharField(required=True, max_length=36, allow_blank=True, label=_("Organization"))
|
|
|
|
@staticmethod
|
|
def validate_org_id(org_id):
|
|
org = Organization.get_instance(org_id)
|
|
if not org:
|
|
error = _('The organization `{}` does not exist'.format(org_id))
|
|
raise serializers.ValidationError(error)
|
|
return org_id
|
|
|
|
def validate(self, attrs):
|
|
if self.instance:
|
|
return attrs
|
|
|
|
ticket_type = attrs.get('type')
|
|
org_id = attrs.get('org_id')
|
|
flow = TicketFlow.get_org_related_flows(org_id=org_id) \
|
|
.filter(type=ticket_type).first()
|
|
|
|
if flow:
|
|
attrs['flow'] = flow
|
|
return attrs
|
|
else:
|
|
error = _('The ticket flow `{}` does not exist'.format(ticket_type))
|
|
raise serializers.ValidationError(error)
|