[Update] 完善 作业中心/任务列表 (#4105)

This commit is contained in:
xinwen
2020-06-15 19:42:36 +08:00
committed by GitHub
parent 5bea782b9f
commit c5a9a85818
3 changed files with 56 additions and 17 deletions

View File

@@ -4,14 +4,17 @@
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, AdHocExecution
from ..serializers import TaskSerializer, AdHocSerializer, \
AdHocExecutionSerializer, TaskDetailSerializer
from ..serializers import (
TaskSerializer,
AdHocSerializer,
AdHocExecutionSerializer,
TaskDetailSerializer,
AdHocDetailSerializer,
)
from ..tasks import run_ansible_task
__all__ = [
@@ -53,6 +56,11 @@ class AdHocViewSet(viewsets.ModelViewSet):
serializer_class = AdHocSerializer
permission_classes = (IsOrgAdmin,)
def get_serializer_class(self):
if self.action == 'retrieve':
return AdHocDetailSerializer
return super().get_serializer_class()
def get_queryset(self):
task_id = self.request.query_params.get('task')
if task_id: