You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.2 KiB
46 lines
1.2 KiB
# -*- 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()
|