diff --git a/hb_server/apps/mtm/views.py b/hb_server/apps/mtm/views.py index 4b98443..f1510cb 100644 --- a/hb_server/apps/mtm/views.py +++ b/hb_server/apps/mtm/views.py @@ -7,10 +7,11 @@ from apps.mtm.serializers import MaterialSerializer, ProductProcessListSerialize from apps.system.mixins import CreateUpdateModelAMixin, OptimizationMixin from rest_framework.decorators import action from rest_framework.response import Response +from utils.pagination import PageOrNot # Create your views here. -class MaterialViewSet(CreateUpdateModelAMixin, ModelViewSet): +class MaterialViewSet(PageOrNot, CreateUpdateModelAMixin, ModelViewSet): """ 物料表-增删改查 """ @@ -33,7 +34,7 @@ class MaterialViewSet(CreateUpdateModelAMixin, ModelViewSet): return Response(serializer.data) -class ProcessViewSet(CreateUpdateModelAMixin, ModelViewSet): +class ProcessViewSet(PageOrNot, CreateUpdateModelAMixin, ModelViewSet): """ 工序表-增删改查 """ @@ -66,7 +67,7 @@ class StepViewSet(CreateModelMixin, UpdateModelMixin, RetrieveModelMixin, Destro ordering = ['sort'] -class ProductProcessViewSet(CreateModelMixin, UpdateModelMixin, ListModelMixin, DestroyModelMixin, GenericViewSet): +class ProductProcessViewSet(PageOrNot, CreateModelMixin, UpdateModelMixin, ListModelMixin, DestroyModelMixin, GenericViewSet): """ 产品生产工艺流程增删改查 """