# -*- coding: utf-8 -*-

from .monitor_task_db import MonitorTask
import utils.date_format as date_format
from base.enums import TaskStatus


async def get_today_task():
    """
    获取当天的任务信息
    :return:
    """
    # 当天日期
    today = date_format.gen_today_str()
    task_model = MonitorTask()
    result = await task_model.filter(gather_date=today, status=TaskStatus.WAITING.value).all()
    return result


async def get_task(task_id):
    """
    获取指定id的任务信息
    :param task_id:
    :return:
    """
    task_model = MonitorTask()
    return await task_model.get_or_none(id=task_id)


async def complete(task_id):
    task_model = MonitorTask()
    await task_model.filter(id=task_id).update(status=TaskStatus.COMPLETED.value, update_time=date_format.timestamp())


async def running(task_id):
    task_model = MonitorTask()
    await task_model.filter(id=task_id).update(status=TaskStatus.RUNNING.value, update_time=date_format.timestamp())


async def fail(task_id):
    task_model = MonitorTask()
    await task_model.filter(id=task_id).update(status=TaskStatus.FAIL.value, update_time=date_format.timestamp())


if __name__ == '__main__':
    get_today_task()