diff --git a/models/task_model.py b/models/task_model.py index b7ace122e57165001b6a392acb624d54633fc7ec..42428e22e105b1319c7c375939a6090cfd5b5872 100644 --- a/models/task_model.py +++ b/models/task_model.py @@ -2,6 +2,7 @@ from sqlalchemy import Column, String, Integer, Enum, JSON, update, Boolean from common.enums import Status_EN, Task_Run_Method from models import CommonModel +from services.const import ERROR_UN_EXISTED_TASK class Task(CommonModel): @@ -26,3 +27,12 @@ async def update_manual_task_status(task_ids): status=Status_EN.RUNNING.value) await Task.exec_orm(batch_update_orm) return + + +async def update_task_status(status, task_id): + task = await Task.query_obj_one(Task.id == task_id) + if not task: + return ERROR_UN_EXISTED_TASK, False + task.status = status + await task.update() + return task.to_dict(), True