from rest_framework.views import APIView
from apps.em.models import Equipment
from apps.em.serializers import EquipmentSerializer
from apps.mtm.models import InputMaterial, Step, SubProduction, UsedStep
from apps.system.mixins import CreateUpdateModelAMixin
from apps.pm.serializers import ProductionPlanCreateFromOrderSerializer, ProductionPlanSerializer, ResourceCalListSerializer, ResourceCalSerializer
from rest_framework.mixins import CreateModelMixin, ListModelMixin
from apps.pm.models import ProductionPlan
from rest_framework.viewsets import GenericViewSet, ModelViewSet
from django.shortcuts import render
from apps.sam.models import Order
from rest_framework.exceptions import APIException
from rest_framework.response import Response
from rest_framework.decorators import action
# Create your views here.

def updateOrderPlanedCount(order):
    """
    更新订单已排数量
    """
    planed_count = 0
    plans = ProductionPlan.objects.filter(order=order)
    for i in plans:
        planed_count = planed_count + i.count
    order.planed_count = planed_count
    order.save()

class ProductionPlanViewSet(CreateUpdateModelAMixin, ListModelMixin, CreateModelMixin, GenericViewSet):
    """
    生产计划
    """
    perms_map = {'*': '*'}
    queryset = ProductionPlan.objects.select_related('order', 'order__contract', 'product')
    serializer_class = ProductionPlanSerializer
    search_fields = ['number']
    filterset_fields = []
    ordering_fields = ['id']
    ordering = ['-id']

    def get_serializer_class(self):
        if self.action in ['create']:
            return ProductionPlanCreateFromOrderSerializer
        return ProductionPlanSerializer

    def create(self, request, *args, **kwargs):
        data = request.data
        serializer = self.get_serializer(data=data)
        serializer.is_valid(raise_exception=True)
        if data.get('order', None):
            order = Order.objects.get(pk=data['order'])
            if order.planed_count >= data['count'] or data['count'] > 0:
                pass
            else:
                raise APIException('排产数量错误')
            instance = serializer.save(create_by=request.user, product=order.product)
            updateOrderPlanedCount(instance.order)
        return Response()

class ResourceViewSet(GenericViewSet):
    
    perms_map = {'*': '*'}
    @action(methods=['post'], detail=False, perms_map={'get':'*'}, serializer_class=ResourceCalListSerializer)
    def cal(self, request, pk=None):
        """
        物料消耗计算
        """
        rdata = request.data
        serializer = self.get_serializer(data=rdata)
        serializer.is_valid(raise_exception=True)
        res_d_list = []
        res = []
        for i in rdata:
            materials = InputMaterial.objects.filter(subproduction__product__id=i['id'], 
            subproduction__is_deleted=False, is_deleted=False, material__type__in=[3,4]).order_by('material__number')\
                .values('material__id', 'material__name', 'material__number', 'material__type', 'count', 'material__count')
            l_m = list(materials)
            for m in l_m:
                if m['material__id'] in res_d_list:
                    index = res_d_list.index(m['material__id'])
                    res[index]['count'] = res[index]['count'] + m['count']*i['count']
                else:
                    res_d_list.append(m['material__id'])
                    res.append({'id':m['material__id'], 'name':m['material__name'],
                     'type':m['material__type'], 'number':m['material__number'], 
                     'count':m['count']*i['count'], 'inv_count':m['material__count']})
        return Response(res)

    @action(methods=['post'], detail=False, perms_map={'get':'*'}, serializer_class=ResourceCalListSerializer)
    def cal_equip(self, request, pk=None):
        """
        设备状态查看
        """
        rdata = request.data
        serializer = self.get_serializer(data=rdata)
        serializer.is_valid(raise_exception=True)
        rdata_l = []
        for i in rdata:
            rdata_l.append(i['id'])
        subproductions = SubProduction.objects.filter(product__id__in=rdata_l, is_deleted=False)
        steps = Step.objects.filter(usedsteps__is_deleted=False, usedsteps__subproduction__in=subproductions)
        equips = Equipment.objects.filter(step_equips__in=steps, is_deleted=False)
        serializer = EquipmentSerializer(instance=equips, many=True)
        return Response(serializer.data)