From a151baaeddb312b0434d41a4b9ca011981960619 Mon Sep 17 00:00:00 2001 From: caoqianming Date: Fri, 3 Jan 2025 15:21:16 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E6=B8=85=E6=B4=97?= =?UTF-8?q?=E6=95=B0=E6=8D=AE=E7=9A=84=E6=8E=A5=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/develop/views.py | 27 +++++++++++++++++++++++++-- 1 file changed, 25 insertions(+), 2 deletions(-) diff --git a/apps/develop/views.py b/apps/develop/views.py index 1411fce7..8e347588 100755 --- a/apps/develop/views.py +++ b/apps/develop/views.py @@ -19,10 +19,13 @@ from apps.opm.models import Opl from apps.third.dahua import dhClient from apps.third.speaker import spClient from apps.third.models import TDevice +from apps.utils.permission import RbacPermission from apps.utils.sms import send_sms from apps.utils.speech import generate_voice from apps.utils.tools import get_info_from_id import importlib +from rest_framework_simplejwt.authentication import JWTAuthentication +from rest_framework.authentication import BasicAuthentication, SessionAuthentication from apps.utils.viewsets import CustomGenericViewSet from apps.utils.wx import wxClient @@ -234,7 +237,7 @@ class TestViewSet(CustomGenericViewSet): authentication_classes = () permission_classes = () - @action(methods=['post'], detail=False, serializer_class=SendSmsSerializer) + @action(methods=['post'], detail=False, serializer_class=SendSmsSerializer, authentication_classes=()) def send_sms(self, request, pk=None): """发送短信测试 @@ -561,7 +564,9 @@ class TestViewSet(CustomGenericViewSet): # tr.save() return Response() - @action(methods=['post'], detail=False, serializer_class=CleanDataSerializer, permission_classes=[]) + @action(methods=['post'], detail=False, serializer_class=CleanDataSerializer, + authentication_classes=[JWTAuthentication, BasicAuthentication, SessionAuthentication], + permission_classes=[IsAdminUser]) def clean_data(self, request, pk=None): """清空数据 @@ -608,6 +613,24 @@ class TestViewSet(CustomGenericViewSet): from django_celery_results.models import TaskResult TaskResult.objects.filter(task_name__in=['apps.vm.tasks.close_visit_by_leave_time', 'apps.rpm.tasks.close_rpj_by_leave_time']).delete() + if 'mes' in datas: + from apps.inm.models import MaterialBatch, MIO + from apps.wpmw.models import Wpr + from apps.wpm.models import WMaterial, Mlog + from apps.pum.models import PuOrder + from apps.sam.models import Order + from apps.pm.models import Utask, Mtask + from apps.wpm.models import Handover + MaterialBatch.objects.all().delete() + MIO.objects.all().delete() + Wpr.objects.all().delete() + WMaterial.objects.all().delete() + Utask.objects.all().delete() + Mtask.objects.all().delete() + Mlog.objects.all().delete() + PuOrder.objects.get_queryset(all=True).delete() + Order.objects.get_queryset(all=True).delete() + Handover.objects.all().delete() return Response() @action(methods=['post'], detail=False, serializer_class=Serializer, permission_classes=[])