diff --git a/apps/cm/views.py b/apps/cm/views.py index 25f5db29..2ed04daa 100644 --- a/apps/cm/views.py +++ b/apps/cm/views.py @@ -5,13 +5,14 @@ from apps.cm.serializers import TidSerializer, LabelMatSerializer from apps.inm.models import MaterialBatch from rest_framework.exceptions import ParseError from rest_framework.response import Response -from apps.utils.viewsets import CustomGenericViewSet, RetrieveModelMixin +from apps.utils.viewsets import CustomGenericViewSet, RetrieveModelMixin, CustomListModelMixin # Create your views here. -class LableMatViewSet(RetrieveModelMixin, CustomGenericViewSet): +class LableMatViewSet(CustomListModelMixin, RetrieveModelMixin, CustomGenericViewSet): perms_map = {"post": "*", "get": "*"} serializer_class = LabelMatSerializer queryset = LableMat.objects.all() + select_related_fields = ["material", "material_origin", "supplier"] @action(methods=['post'], detail=False, serializer_class=TidSerializer) def get_from_mb(self, request, pk=None):