from django.shortcuts import render
from rest_framework.generics import CreateAPIView, GenericAPIView
from rest_framework.utils.field_mapping import get_relation_kwargs
from rest_framework.viewsets import GenericViewSet, ModelViewSet
from apps.pm.serializers import SubProductionPlanUpdateSerializer

from apps.system.mixins import CreateUpdateModelAMixin, OptimizationMixin
from rest_framework.decorators import action

from apps.wpm.serializers import PickSerializer
from rest_framework.response import Response
# Create your views here.
class WMaterialViewSet(CreateUpdateModelAMixin, GenericViewSet):
    """
    领料
    """
    perms_map={'*':'*'}
    @action(methods=['post'], detail=True, perms_map={'post':'*'}, serializer_class=PickSerializer)
    def pick(self, request, pk=None):
        """
        领料
        """
        serializer= PickSerializer(data=request.data, context={'request': request})
        serializer.is_valid(raise_exception=True)
        serializer.save()
        return Response()