From b9ac0207fc421497ec56c0913b10f5e6caba2044 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: Thu, 4 Aug 2022 17:47:26 +0800
Subject: [PATCH] feature: add tasks to plan

---
 services/plan_service.py |  7 ++-----
 views/plan_view.py       | 12 ++++++++++++
 2 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/services/plan_service.py b/services/plan_service.py
index 632ccb5..5de9253 100644
--- a/services/plan_service.py
+++ b/services/plan_service.py
@@ -70,12 +70,9 @@ async def add_tasks(data, plan_id):
     if not plan:
         return ERROR_UN_EXISTED_PLAN, False
     tasks = ','.join(str(x) for x in data['tasks'])
-    if not plan.tasks:
-        plan.tasks = tasks
-    else:
-        plan.tasks = f'{plan.tasks},{tasks}'
+    plan.tasks = tasks
     await plan.update()
-    await Task.batch_update({'plan_id': plan_id}, Task.id.in_(data['tasks']))
+    await Task.batch_update({'plan_id': plan_id, 'plan_title': plan.plan_title}, Task.id.in_(data['tasks']))
     return plan.to_dict(), True
 
 
diff --git a/views/plan_view.py b/views/plan_view.py
index 4988348..b92f28e 100644
--- a/views/plan_view.py
+++ b/views/plan_view.py
@@ -61,6 +61,18 @@ async def get_plan_task_list(_, plan_id, user_infos):
     return rsp(data=result)
 
 
+@bp.post('/<plan_id>/tasks')
+@read_auth
+async def add_task_view(request, plan_id, user_infos):
+    response, ok = check_args('tasks', request.json)
+    if not ok:
+        return response
+    result, ok = await add_tasks(request.json, plan_id)
+    if not ok:
+        return rsp(code=500, msg=result)
+    return rsp(data=result)
+
+
 @bp.post('/review/<plan_id>')
 @read_auth
 async def review(request, plan_id, user_infos):
-- 
Gitee