from django.utils import timezone
from typing import List

from django.db.models.expressions import F
from apps.pm.models import SubProductionPlan, SubProductionProgress
from apps.mtm.models import Material, Step, SubprodctionMaterial
from apps.qm.models import TestRecord
from apps.system.models import User
from apps.wpm.models import WProduct, WproductFlow, WprouctTicket
from utils.tools import ranstr
class WpmServies(object):

    @classmethod    
    def get_next_step(cls, subproduction_plan:SubProductionPlan, nowstep:Step):
        """
        获取下一步骤
        """
        stepIds = [i['id'] for i in subproduction_plan.steps]
        pindex = stepIds.index(nowstep.id)
        if pindex + 1 < len(stepIds):
            return Step.objects.get(pk=stepIds[pindex+1]), True
        else:
            return nowstep, False

    @classmethod
    def get_subplans_queryset_from_wproducts(cls, wproducts:List):
        """
        通过半成品列表获取所属子计划
        """
        splans = SubProductionPlan.objects.filter(is_deleted=False, wproduct_subplan__in=wproducts)
        return splans

    @classmethod
    def get_subplans_queyset_from_step(cls, step:Step):
        """
        通过当前操作获取所有正在进行的子计划
        """
        splans = SubProductionPlan.objects.filter(is_deleted=False, 
                subproduction__usedstep_subproduction__step=step, state=SubProductionPlan.SUBPLAN_STATE_WORKING)
        return splans
    
    @classmethod
    def update_wproduct_by_test(cls, test:TestRecord, user:User):
        """
        根据检验结果更新玻璃及相关状态
        """
        is_testok = test.is_testok
        wproduct = test.wproduct
        if is_testok:
            if wproduct.act_state == WProduct.WPR_ACT_STATE_TORETEST: # 复检
                wproduct.act_state = WProduct.WPR_ACT_STATE_DOWAIT

            elif wproduct.act_state == WProduct.WPR_ACT_STATE_TOTEST and wproduct.material.type == Material.MA_TYPE_GOOD: # 成品检验
                wproduct.act_state = WProduct.WPR_ACT_STATE_TOFINALTEST 
            else:
                wproduct.act_state = WProduct.WPR_ACT_STATE_OK
            if wproduct.number is None: # 产生半成品编号
                wproduct.number = 'WP'+ranstr(7)

            # 去除ng_sign
            if wproduct.ng_sign:
                wt = WprouctTicket.objects.order_by('id').last() #取最后的工单
                if wt.step.process == test.step.process:
                    wproduct.ng_sign = None
            # 更新子计划相关进度
            cls.update_subproduction_progress_main(sp=wproduct.subproduction_plan)
        else:# 如果不合格
            wproduct.act_state = WProduct.WPR_ACT_STATE_NOTOK
            # 需要走不合格品审理的工单
        wproduct.update_by = user
        wproduct.update_time = timezone.now()
        wproduct.test = None
        wproduct.save()

    @classmethod
    def update_subproduction_progress_main(cls, sp:SubProductionPlan):
        """
        根据产品变动日志更新生产进度
        """
        objs = WproductFlow.objects.filter(subproduction_plan=sp, is_lastlog=True)
        count_ok = objs.filter(act_state__in=[WProduct.WPR_ACT_STATE_INM, 
                    WProduct.WPR_ACT_STATE_OK, WProduct.WPR_ACT_STATE_SELLED]).count()
        count_notok = objs.filter(act_state__in=[WProduct.WPR_ACT_STATE_NOTOK, WProduct.WPR_ACT_STATE_SCRAP]).count()
        count_real = objs.exclude(act_state__in=[WProduct.WPR_ACT_STATE_TORETEST, 
        WProduct.WPR_ACT_STATE_DOWAIT, WProduct.WPR_ACT_STATE_DOING]).count()
        ins = SubProductionProgress.objects.filter(subproduction_plan=sp, 
                is_main=True, type=SubprodctionMaterial.SUB_MA_TYPE_OUT).first()
        if ins:
            ins.count_ok = count_ok
            ins.count_notok = count_notok
            ins.count_real = count_real
            ins.save()