jumpserver/apps/tickets/api/relation.py

from rest_framework.mixins import CreateModelMixin
from rest_framework import views
from rest_framework.response import Response
from rest_framework import status

from common.drf.api import JMSGenericViewSet
from tickets.models import TicketSession
from tickets.serializers import TicketSessionRelationSerializer
from terminal.serializers import SessionSerializer
from orgs.utils import tmp_to_root_org


class TicketSessionRelationViewSet(CreateModelMixin, JMSGenericViewSet):
    queryset = TicketSession.objects.all()
    serializer_class = TicketSessionRelationSerializer


# Todo: 放到上面的 ViewSet 中
class TicketSessionApi(views.APIView):
    perm_model = TicketSession
    rbac_perms = {
        '*': ['tickets.view_ticket']
    }

    def get(self, request, *args, **kwargs):
        with tmp_to_root_org():
            tid = self.kwargs['ticket_id']
            ticket_session = TicketSession.objects.filter(ticket=tid).first()
            if not ticket_session:
                return Response(status=status.HTTP_404_NOT_FOUND)

            session = ticket_session.session
            serializer = SessionSerializer(session)
            return Response(serializer.data)