perf: 优化ops代码

This commit is contained in:
Aaron3S
2022-12-06 17:30:54 +08:00
parent 6233e2b3de
commit d92b198a12
6 changed files with 40 additions and 28 deletions

View File

@@ -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)