mirror of https://github.com/jumpserver/jumpserver
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from rest_framework import serializers
|
|
|
|
from common.serializers import AdaptedBulkListSerializer
|
|
from common.mixins.serializers import BulkSerializerMixin
|
|
from .base import TicketSerializer
|
|
from ..models import LoginConfirmTicket
|
|
|
|
|
|
__all__ = ['LoginConfirmTicketSerializer', 'LoginConfirmTicketActionSerializer']
|
|
|
|
|
|
class LoginConfirmTicketSerializer(BulkSerializerMixin, serializers.ModelSerializer):
|
|
class Meta:
|
|
list_serializer_class = AdaptedBulkListSerializer
|
|
model = LoginConfirmTicket
|
|
fields = TicketSerializer.Meta.fields + [
|
|
'ip', 'city', 'action'
|
|
]
|
|
read_only_fields = TicketSerializer.Meta.read_only_fields
|
|
|
|
def create(self, validated_data):
|
|
validated_data.pop('action')
|
|
return super().create(validated_data)
|
|
|
|
def update(self, instance, validated_data):
|
|
action = validated_data.get("action")
|
|
user = self.context["request"].user
|
|
|
|
if action and user not in instance.assignees.all():
|
|
error = {"action": "Only assignees can update"}
|
|
raise serializers.ValidationError(error)
|
|
if instance.status == instance.STATUS_CLOSED:
|
|
validated_data.pop('action')
|
|
instance = super().update(instance, validated_data)
|
|
if not instance.status == instance.STATUS_CLOSED:
|
|
instance.perform_action(action, user)
|
|
return instance
|
|
|
|
|
|
class LoginConfirmTicketActionSerializer(serializers.ModelSerializer):
|
|
comment = serializers.CharField(allow_blank=True)
|
|
|
|
class Meta:
|
|
model = LoginConfirmTicket
|
|
fields = ['action']
|
|
|
|
def update(self, instance, validated_data):
|
|
pass
|
|
|
|
def create(self, validated_data):
|
|
pass
|