mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-10-31 05:41:59 +00:00
7d5a13de38
perf: 优化status api perf: 优化 status api perf: 修改sesion参数 perf: 修改migrations perf: 优化数据结构 perf: 修改保留日志 perf: 优化之前的一个写法
75 lines
2.6 KiB
Python
75 lines
2.6 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', [])
|
|
# guacamole 上报的 session 是字符串
|
|
# "[53cd3e47-210f-41d8-b3c6-a184f3, 53cd3e47-210f-41d8-b3c6-a184f4]"
|
|
if isinstance(session_ids, str):
|
|
session_ids = session_ids[1:-1].split(',')
|
|
session_ids = [sid.strip() for sid in session_ids if sid.strip()]
|
|
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)
|