mirror of https://github.com/jumpserver/jumpserver
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
|
|
import logging
|
|
from django.shortcuts import get_object_or_404
|
|
from rest_framework import viewsets, generics
|
|
from rest_framework.views import Response
|
|
from rest_framework import status
|
|
|
|
from common.permissions import IsAppUser, IsOrgAdminOrAppUser, IsSuperUser
|
|
from ..models import Terminal, Status, Session
|
|
from .. import serializers
|
|
from ..utils import TypedComponentsStatusMetricsUtil
|
|
|
|
logger = logging.getLogger(__file__)
|
|
|
|
|
|
__all__ = [
|
|
'StatusViewSet',
|
|
'ComponentsMetricsAPIView',
|
|
]
|
|
|
|
|
|
class StatusViewSet(viewsets.ModelViewSet):
|
|
queryset = Status.objects.all()
|
|
serializer_class = serializers.StatusSerializer
|
|
permission_classes = (IsOrgAdminOrAppUser,)
|
|
session_serializer_class = serializers.SessionSerializer
|
|
task_serializer_class = serializers.TaskSerializer
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
serializer = self.get_serializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
self.handle_sessions()
|
|
self.perform_create(serializer)
|
|
tasks = self.request.user.terminal.task_set.filter(is_finished=False)
|
|
serializer = self.task_serializer_class(tasks, many=True)
|
|
return Response(serializer.data, status=201)
|
|
|
|
def handle_sessions(self):
|
|
session_ids = self.request.data.get('sessions', [])
|
|
Session.set_sessions_active(session_ids)
|
|
|
|
def get_queryset(self):
|
|
terminal_id = self.kwargs.get("terminal", None)
|
|
if terminal_id:
|
|
terminal = get_object_or_404(Terminal, id=terminal_id)
|
|
return terminal.status_set.all()
|
|
return super().get_queryset()
|
|
|
|
def perform_create(self, serializer):
|
|
serializer.validated_data.pop('sessions', None)
|
|
serializer.validated_data["terminal"] = self.request.user.terminal
|
|
return super().perform_create(serializer)
|
|
|
|
def get_permissions(self):
|
|
if self.action == "create":
|
|
self.permission_classes = (IsAppUser,)
|
|
return super().get_permissions()
|
|
|
|
|
|
class ComponentsMetricsAPIView(generics.GenericAPIView):
|
|
""" 返回汇总组件指标数据 """
|
|
permission_classes = (IsSuperUser,)
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
util = TypedComponentsStatusMetricsUtil()
|
|
metrics = util.get_metrics()
|
|
return Response(metrics, status=status.HTTP_200_OK)
|