mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-01-15 23:23:21 +00:00
55c95c58f6
* [Update] 添加一下model到operate log, [platform,remoteapppermission,changeauthplan,gatherusertask] * [Bugfix] 修改了返回platform的几个位置，修改了command execution的url * [Update] 优化ops task表结构，避免列表页查询几十次sql, 优化了基础的encryptjsonfield * [Update] 修改adhoc 返回的become字段，避免密码泄露 * [Update] 修改变量名称
85 lines
2.5 KiB
Python
85 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
|
|
from django.shortcuts import get_object_or_404
|
|
from rest_framework import viewsets, generics
|
|
from rest_framework.views import Response
|
|
from django.db.models import Count, Q
|
|
|
|
from common.permissions import IsOrgAdmin
|
|
from common.serializers import CeleryTaskSerializer
|
|
from orgs.utils import current_org
|
|
from ..models import Task, AdHoc, AdHocRunHistory
|
|
from ..serializers import TaskSerializer, AdHocSerializer, \
|
|
AdHocRunHistorySerializer
|
|
from ..tasks import run_ansible_task
|
|
|
|
__all__ = [
|
|
'TaskViewSet', 'TaskRun', 'AdHocViewSet', 'AdHocRunHistoryViewSet'
|
|
]
|
|
|
|
|
|
class TaskViewSet(viewsets.ModelViewSet):
|
|
queryset = Task.objects.all()
|
|
filter_fields = ("name",)
|
|
search_fields = filter_fields
|
|
serializer_class = TaskSerializer
|
|
permission_classes = (IsOrgAdmin,)
|
|
|
|
def get_queryset(self):
|
|
queryset = super().get_queryset()
|
|
if current_org.is_real():
|
|
queryset = queryset.filter(created_by=current_org.id)
|
|
else:
|
|
queryset = queryset.filter(created_by='')
|
|
queryset = queryset.select_related('latest_history')
|
|
return queryset
|
|
|
|
|
|
class TaskRun(generics.RetrieveAPIView):
|
|
queryset = Task.objects.all()
|
|
serializer_class = CeleryTaskSerializer
|
|
permission_classes = (IsOrgAdmin,)
|
|
|
|
def retrieve(self, request, *args, **kwargs):
|
|
task = self.get_object()
|
|
t = run_ansible_task.delay(str(task.id))
|
|
return Response({"task": t.id})
|
|
|
|
|
|
class AdHocViewSet(viewsets.ModelViewSet):
|
|
queryset = AdHoc.objects.all()
|
|
serializer_class = AdHocSerializer
|
|
permission_classes = (IsOrgAdmin,)
|
|
|
|
def get_queryset(self):
|
|
task_id = self.request.query_params.get('task')
|
|
if task_id:
|
|
task = get_object_or_404(Task, id=task_id)
|
|
self.queryset = self.queryset.filter(task=task)
|
|
return self.queryset
|
|
|
|
|
|
class AdHocRunHistoryViewSet(viewsets.ModelViewSet):
|
|
queryset = AdHocRunHistory.objects.all()
|
|
serializer_class = AdHocRunHistorySerializer
|
|
permission_classes = (IsOrgAdmin,)
|
|
|
|
def get_queryset(self):
|
|
task_id = self.request.query_params.get('task')
|
|
adhoc_id = self.request.query_params.get('adhoc')
|
|
if task_id:
|
|
task = get_object_or_404(Task, id=task_id)
|
|
adhocs = task.adhoc.all()
|
|
self.queryset = self.queryset.filter(adhoc__in=adhocs)
|
|
|
|
if adhoc_id:
|
|
adhoc = get_object_or_404(AdHoc, id=adhoc_id)
|
|
self.queryset = self.queryset.filter(adhoc=adhoc)
|
|
return self.queryset
|
|
|
|
|
|
|
|
|
|
|