from django.db import models from apps.qm.models import FtestDefect, FtestItem from apps.utils.models import BaseModel from apps.mtm.models import Material from rest_framework.exceptions import ParseError from apps.wpm.models import WmStateOption from apps.utils.tools import update_dict # Create your models here. class Wpr(BaseModel): """ TN: 动态产品 """ number = models.CharField("编号", max_length=50, null=True, blank=True) version = models.PositiveIntegerField("版本号", default=9999) state = models.PositiveSmallIntegerField('状态', default=10, choices=WmStateOption.choices) material = models.ForeignKey(Material, verbose_name="当前物料形态", on_delete=models.CASCADE) defects = models.ManyToManyField("qm.defect", through="wpmw.wprdefect", verbose_name="缺陷项") mb = models.ForeignKey("inm.materialbatch", verbose_name="仓库物料", on_delete=models.CASCADE, null=True, blank=True) wm = models.ForeignKey("wpm.wmaterial", verbose_name="车间物料", on_delete=models.CASCADE, null=True, blank=True) oinfo = models.JSONField(verbose_name="其他信息", default=dict, blank=True) wpr_from = models.ForeignKey("self", verbose_name="来源于", on_delete=models.CASCADE, null=True, blank=True) @classmethod def change_or_new(cls, wpr=None, number=None, mb=None, wm=None, old_mb=None, old_wm=None, ftest=None, wpr_from=None, batch_from=None): if wpr is None and number is None: raise ParseError("id和number不能同时为空") if mb and wm: raise ParseError("所属仓库批次和车间批次不可同时存在") if wpr: ins:Wpr = wpr if wpr and mb is None and wm is None and ins.version <= 0: # 相当于撤回操作 wpr.delete() return elif number: try: ins_x = cls.objects.get(number=number) if ins_x.wm is None and ins_x.mb is None: if ins_x.version > 1: # 说明被复用了 if wpr_from is None: wpr_from = ins_x else: raise ParseError(f"该物料编号{number}-已存在不可使用") else: raise ParseError(f"该物料编号{number}-已存在不可使用") except cls.DoesNotExist: ins = cls(number=number) ins.version = -1 ins.oinfo = {} if batch_from: # 尝试从批号追踪来源 try: ins_from = cls.objects.get(number=number) if ins_from.wm is None and ins_from.mb is None: if ins_from.version > 1: # 说明被复用了 wpr_from = ins_from else: raise ParseError(f"该物料编号{number}-已存在不可使用") except cls.DoesNotExist: pass if old_mb and ins.mb != old_mb: raise ParseError(f"请检查-{ins.number}-所属仓库批次") if old_wm and ins.wm != old_wm: raise ParseError(f"请检查-{ins.number}-所属车间批次") ins.wpr_from = wpr_from ins.mb = mb if mb: ins.material = mb.material ins.wm = wm if wm: ins.material = wm.material ins.state = wm.state WprDefect.objects.filter(wpr=ins).delete() if wm.defect: WprDefect.objects.create(wpr=ins, defect=wm.defect, is_main=True) ins.version = ins.version + 1 ins.save() if ftest: # 通过检验变更其缺陷项 WprDefect.objects.filter(wpr=ins).delete() for ftestdefect in FtestDefect.objects.filter(ftest=ftest, has=True): WprDefect.objects.create(wpr=ins, defect=ftestdefect.defect, is_main=ftestdefect.is_main) # 携带某些检测信息 oinfo = {} for ftestitem in FtestItem.objects.filter(ftest=ftest, addto_wpr=True): if ftestitem.test_val_json: oinfo[ftestitem.id] = {"name": ftestitem.testitem.name, "val": ftestitem.test_val_json} ins.oinfo = update_dict(ins.oinfo, oinfo) ins.save() # if ins.mb and ins.wm: # raise ParseError("所属仓库批次和车间批次不可同时存在2") return ins @classmethod def clear(cls, number_list): cls.objects.filter(number__in=number_list).update(mb=None, wm=None) @classmethod def get_qs_by_mb(cls, mb): return cls.objects.filter(mb=mb, wm=None) @classmethod def get_qs_by_wm(cls, wm): return cls.objects.filter(wm=wm, mb=None) @property def wprdefect(self): return WprDefect.objects.filter(wpr=self) class WprDefect(BaseModel): """TN:产物缺陷项关联表""" wpr = models.ForeignKey(Wpr, verbose_name="关联产物", on_delete=models.CASCADE) defect = models.ForeignKey("qm.defect", verbose_name="关联缺陷项", on_delete=models.CASCADE) is_main = models.BooleanField("是否主要缺陷", default=False)