from apps.utils.viewsets import GenericViewSet
from rest_framework.decorators import action
from apps.em.models import Equipment
from apps.wpm.modes import Mlog
from django.utils import timezone

class AnaViewSet(GenericViewSet):
    perms_map = {}
    @action(methods=['post'], detail=False, perms_map={'post': '*'})
    def 设备最后生产日志(self, request):
        data = request.data
        now = timezone.now()
        mgroup_name = data["mgroup_name"]
        # 生产设备
        equip_qs = Equipment.objects.filter(mgroup__name=mgroup_name, type = 10)
        # 设备最后生产日志
        mlog_qs = Mlog.objects.filter(equip__in=equip_qs)
        mlog_qs = mlog_qs | Mlog.objects.filter(work_start_time__lte=now, work_end_time__isnull=True)
        mlog_qs = mlog_qs | Mlog.objects.filter(work_start_time__lte=now, work_end_time__gte=now)