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): """