mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-09-10 20:00:10 +00:00
perf: 优化ops代码
This commit is contained in:
@@ -2,6 +2,7 @@ from rest_framework import viewsets
|
||||
from rest_framework_bulk import BulkModelViewSet
|
||||
|
||||
from common.mixins import CommonApiMixin
|
||||
from ops.api.base import SelfBulkModelViewSet
|
||||
from ops.models import Job, JobExecution
|
||||
from ops.serializers.job import JobSerializer, JobExecutionSerializer
|
||||
|
||||
@@ -16,12 +17,13 @@ def set_task_to_serializer_data(serializer, task):
|
||||
setattr(serializer, "_data", data)
|
||||
|
||||
|
||||
class JobViewSet(CommonApiMixin, BulkModelViewSet):
|
||||
class JobViewSet(SelfBulkModelViewSet):
|
||||
serializer_class = JobSerializer
|
||||
permission_classes = ()
|
||||
model = Job
|
||||
|
||||
def get_queryset(self):
|
||||
query_set = Job.objects.filter(creator=self.request.user)
|
||||
query_set = super().get_queryset()
|
||||
if self.action != 'retrieve':
|
||||
return query_set.filter(instant=False)
|
||||
return query_set
|
||||
@@ -45,10 +47,11 @@ class JobViewSet(CommonApiMixin, BulkModelViewSet):
|
||||
set_task_to_serializer_data(serializer, task)
|
||||
|
||||
|
||||
class JobExecutionViewSet(CommonApiMixin, BulkModelViewSet):
|
||||
class JobExecutionViewSet(SelfBulkModelViewSet):
|
||||
serializer_class = JobExecutionSerializer
|
||||
http_method_names = ('get', 'post', 'head', 'options',)
|
||||
permission_classes = ()
|
||||
model = JobExecution
|
||||
|
||||
def perform_create(self, serializer):
|
||||
instance = serializer.save()
|
||||
@@ -56,8 +59,7 @@ class JobExecutionViewSet(CommonApiMixin, BulkModelViewSet):
|
||||
set_task_to_serializer_data(serializer, task)
|
||||
|
||||
def get_queryset(self):
|
||||
query_set = JobExecution.objects.filter(creator=self.request.user)
|
||||
query_set = query_set.filter(creator=self.request.user)
|
||||
query_set = super().get_queryset()
|
||||
job_id = self.request.query_params.get('job_id')
|
||||
if job_id:
|
||||
query_set = query_set.filter(job_id=job_id)
|
||||
|
Reference in New Issue
Block a user