# Create your tasks here
from __future__ import absolute_import, unicode_literals
from apps.utils.tasks import CustomTask
from celery import shared_task
from django.utils.timezone import localtime
from apps.em.models import Equipment
from apps.em.services import set_eq_rs, get_eq_rs


@shared_task(base=CustomTask)
def check_equipment_offline(seconds=30):
    """监测设备是否掉线(根据测点采集)

    监测设备是否掉线
    """
    equips = Equipment.objects.filter(mp_ep_monitored__is_rep_ep_running_state=True, mp_ep_monitored__enabled=True)
    now = localtime()
    for equip in equips:
        rs = get_eq_rs(equip.id)
        if (now - rs['running_state_timex']).total_seconds() > seconds:
                set_eq_rs(equip.id, now, Equipment.OFFLINE)