from tabnanny import check
from apps.em.models import ECheckRecord, Equipment
from dateutil.relativedelta import relativedelta
from django.utils import timezone
class EmService:

    @classmethod
    def update_check_date(cls, equip:Equipment):
        # 根据校准检定记录变更下次日期
        check = ECheckRecord.objects.filter(equipment=equip).order_by('id').last()
        if check:
            if equip.cycle:
                equip.check_date = check.check_date
                if equip.next_check_date:
                    equip.next_check_date = equip.next_check_date + relativedelta(months=equip.cycle)
                else:
                    equip.next_check_date = timezone.now() + relativedelta(months=equip.cycle)
                equip.save()