diff --git a/models/plan_model.py b/models/plan_model.py index 398978ad4ce6f4edb32d916009f819802429511b..65fd9e339d927ee587f4626b8da9f890771cc43a 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'