From 67d8ede2a47481ffbf2dc82fd0599b58df26fcd8 Mon Sep 17 00:00:00 2001 From: caoqianming Date: Wed, 17 Jan 2024 16:56:07 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E8=AE=A1=E7=AE=97count=5Fsampling?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/qm/views.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/apps/qm/views.py b/apps/qm/views.py index 264cf9b3..a0bf76ae 100644 --- a/apps/qm/views.py +++ b/apps/qm/views.py @@ -65,6 +65,22 @@ class FtestViewSet(CustomModelViewSet): select_related_fields = ['test_user', 'check_user', 'ftest_work'] filterset_fields = ['type', 'ftest_work'] + def perform_create(self, serializer): + ins = super().perform_create(serializer) + if ins.ftest_work: + ins.ftest_work.count_sampling = Ftest.objects.filter( + ftest_work=ins.ftest_work).count() + ins.ftest_work.save() + return ins + + def perform_destroy(self, instance): + ftest_work = instance.ftest_work + instance.delete() + if ftest_work: + instance.ftest_work.count_sampling = Ftest.objects.filter( + ftest_work=instance.ftest_work).count() + instance.ftest_work.save() + class PtestViewSet(CustomModelViewSet): """