From e50f60da286799fde6b6fe2265623f1827a5f238 Mon Sep 17 00:00:00 2001
From: P3ngSaM <61768364+P3ngSaM@users.noreply.github.com>
Date: Thu, 10 Mar 2022 14:56:06 +0800
Subject: [PATCH] =?UTF-8?q?update=20=E4=BF=AE=E6=94=B9=E6=9B=B4=E6=96=B0?=
 =?UTF-8?q?=E6=B5=81=E7=A8=8B=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 company/db.py      | 2 +-
 company/routes.py  | 3 ++-
 company/scripts.py | 9 +++++----
 3 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/company/db.py b/company/db.py
index 7cecf72..f0273dc 100644
--- a/company/db.py
+++ b/company/db.py
@@ -77,7 +77,7 @@ def update_data(param1, param2, param3, param4):
     Returns:
         -
     """
-    collection = client_admin[param1][param2]
+    collection = client_tfse[param1][param2]
     res = collection.update_one(param3, {"$set": param4})
     info = res.raw_result['updatedExisting']
     return info
diff --git a/company/routes.py b/company/routes.py
index 7b21f39..c7aaec7 100644
--- a/company/routes.py
+++ b/company/routes.py
@@ -86,8 +86,9 @@ def update_process():
     """
     rid = request.json['rid']
     renew_data = request.json['renew_data']
+    types = request.json['types']
 
-    info = update_process_scripts(rid, renew_data)
+    info = update_process_scripts(types, rid, renew_data)
     status = 200 if info else 210
     result = '更新成功' if info else '更新失败'
     return {"info": result}, status
diff --git a/company/scripts.py b/company/scripts.py
index 8f8c084..b5b73b0 100644
--- a/company/scripts.py
+++ b/company/scripts.py
@@ -814,14 +814,15 @@ def calculate_weight(param):
     return weight
 
 
-def update_process_scripts(param1, param2):
+def update_process_scripts(param1, param2, param3):
     """
     更新流程数据
     Parameters:
-        param1 str 评价ID
-        param2 dict 需要更新的数据
+        param1 dict 数据集名称
+        param2 str 评价ID
+        param3 dict 需要更新的数据
     Returns:
         result bool 更新结果
     """
-    info = update_data('看板', '评价流程', {"评价ID": param1}, param2)
+    info = update_data('评价', param1, {"评价ID": param2}, param3)
     return info