From 38f77c78f57953b30d47900c91a570b483e895a9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9B=B9=E5=89=8D=E6=98=8E?= <909355014@qq.com> Date: Mon, 12 Sep 2022 19:06:05 +0800 Subject: [PATCH] =?UTF-8?q?=E5=BC=80=E5=90=AF/=E5=81=9C=E6=AD=A2=E4=BD=9C?= =?UTF-8?q?=E4=B8=9A=E7=9B=91=E6=8E=A7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/opm/views.py | 54 ++++++++++++++++++++++++++--------------------- 1 file changed, 30 insertions(+), 24 deletions(-) diff --git a/apps/opm/views.py b/apps/opm/views.py index eadfb871..fab8842b 100644 --- a/apps/opm/views.py +++ b/apps/opm/views.py @@ -27,7 +27,8 @@ class OplCateViewSet(CustomModelViewSet): class OperationViewSet(CustomModelViewSet): - perms_map = {'get': 'operation.view', 'post': 'operation.create', 'put': 'operation.update', 'delete': 'operation.delete'} + perms_map = {'get': 'operation.view', 'post': 'operation.create', + 'put': 'operation.update', 'delete': 'operation.delete'} queryset = Operation.objects.all() create_serializer_class = OperationCreateUpdateSerializer update_serializer_class = OperationCreateUpdateSerializer @@ -73,6 +74,34 @@ class OplViewSet(CustomModelViewSet): raise ParseError('许可证已处理不可编辑') return super().update(request, *args, **kwargs) + @action(methods=['post'], detail=True, perms_map={'post': '*'}, + serializer_class=Serializer, logging_methods=[]) + def start_mtask(self, request, pk=None): + """ + 开启作业监控 + + 开启作业监控 + """ + obj = self.get_object() + if obj.mtask: + # 先关闭 + close_mtask(obj.mtask.task_id) + start_mtask(obj) + return Response() + + @action(methods=['post'], detail=True, perms_map={'post': '*'}, + serializer_class=Serializer, logging_methods=[]) + def stop_mtask(self, request, pk=None): + """ + 停止作业监控 + + 停止作业监控 + """ + obj = self.get_object() + if obj.mtask: + close_mtask(obj.mtask.task_id) + return Response() + class OplWorkerViewSet(CustomModelViewSet): perms_map = {'get': '*', 'post': 'opl.update', 'put': 'opl.update', 'delete': 'opl.update'} @@ -99,29 +128,6 @@ class OplWorkerViewSet(CustomModelViewSet): raise ParseError('许可证已处理不可编辑') return super().update(request, *args, **kwargs) - @action(methods=['post'], detail=True, perms_map={'post': '*'}, - serializer_class=Serializer, logging_methods=[]) - def start_mtask(self, request, pk=None): - """ - 开启作业监控 - """ - obj = self.get_object() - if obj.mtask: - # 先关闭 - close_mtask(obj.mtask.task_id) - start_mtask(obj) - return Response() - - @action(methods=['post'], detail=True, perms_map={'post': '*'}, - serializer_class=Serializer, logging_methods=[]) - def close_mtask(self, request, pk=None): - """ - 停止作业监控 - """ - obj = self.get_object() - if obj.mtask: - close_mtask(obj.mtask.task_id) - return Response() class GasCheckViewSet(CreateModelMixin, ListModelMixin, DestroyModelMixin, CustomGenericViewSet): perms_map = {'get': '*', 'post': 'opl.update', 'delete': 'opl.update'}