From af2ee2b43efce0b21d893eb0fcc40198bbe83974 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E5=AD=A6=E5=B3=B0?= <wb-zxf699644@alibaba-inc.com>
Date: Fri, 8 Jul 2022 13:56:48 +0800
Subject: [PATCH] feature: add func update status for task

---
 models/task_model.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/models/task_model.py b/models/task_model.py
index b7ace12..42428e2 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
-- 
Gitee