mirror of https://github.com/jumpserver/jumpserver
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
|
|
import datetime
|
|
import logging
|
|
from django.utils import timezone
|
|
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 ..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
|
|
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)
|
|
task_serializer = self.get_task_serializer()
|
|
return Response(task_serializer.data, status=201)
|
|
|
|
def handle_sessions(self):
|
|
session_ids = self.request.data.get('sessions', [])
|
|
Session.set_sessions_active(session_ids)
|
|
|
|
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_task_serializer(self):
|
|
critical_time = timezone.now() - datetime.timedelta(minutes=10)
|
|
tasks = self.request.user.terminal.task_set.filter(is_finished=False, date_created__gte=critical_time)
|
|
serializer = self.task_serializer_class(tasks, many=True)
|
|
return serializer
|
|
|
|
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()
|
|
|
|
|
|
class ComponentsMetricsAPIView(generics.GenericAPIView):
|
|
""" 返回汇总组件指标数据 """
|
|
rbac_perms = {
|
|
'GET': 'terminal.view_terminal'
|
|
}
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
util = TypedComponentsStatusMetricsUtil()
|
|
metrics = util.get_metrics()
|
|
return Response(metrics, status=status.HTTP_200_OK)
|