From e3df022bdc5bdfa35ca63688acbf21a660d2ea55 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: Mon, 4 Jul 2022 09:34:21 +0800
Subject: [PATCH] feature: add func update status for plan

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

diff --git a/models/plan_model.py b/models/plan_model.py
index 398978a..65fd9e3 100644
--- a/models/plan_model.py
+++ b/models/plan_model.py
@@ -2,6 +2,7 @@ from sqlalchemy import Column, String, Integer, JSON, Boolean
 
 from common.enums import Status_EN
 from models import CommonModel
+from services.const import ERROR_UN_EXISTED_PLAN
 
 
 class Plan(CommonModel):
@@ -19,6 +20,15 @@ class Plan(CommonModel):
     report = Column(Boolean, nullable=False, default=False, comment='测试报告是否已生成')
 
 
+async def update_plan_status(status, plan_id):
+    plan = await Plan.query_obj_one(Plan.id == plan_id)
+    if not plan:
+        return ERROR_UN_EXISTED_PLAN, False
+    plan.status = status
+    await plan.update()
+    return plan.to_dict(), True
+
+
 class PlanReview(CommonModel):
     __tablename__ = 'plan_review'
 
-- 
Gitee