From 5e5e73df8b9f19d22c6c96abca0d4417eff3d9c9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E6=80=9D=E5=B7=9D?= <sichanwong0527@gmail.com>
Date: Tue, 22 Mar 2022 15:16:48 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AF=84=E4=BB=B7=E6=B5=81=E7=A8=8B=20?=
 =?UTF-8?q?=E5=8D=87=E7=BA=A7=E5=88=B0v0.21?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 company/db.py      | 15 +++++++++++++++
 company/routes.py  |  4 ++--
 company/scripts.py |  6 +++---
 3 files changed, 20 insertions(+), 5 deletions(-)

diff --git a/company/db.py b/company/db.py
index bf670fa..9d9e3eb 100644
--- a/company/db.py
+++ b/company/db.py
@@ -95,3 +95,18 @@ def UPSERT_SINGLE_DATA(param1, param2, param3, param4):
     """
     collection = client_tfse_021[param1][param2]
     collection.update_one(param3, {"$set": param4}, upsert=True)
+
+
+def UPDATE_SINGLE_DATA(param1, param2, param3, param4):
+    """
+    插入单条数据
+    Parameters:
+        param1: str 数据库
+        param2: str 数据表
+        param3: dict 查询条件
+        param4: dict 更新或新插入的数据
+    """
+    collection = client_tfse_021[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 c7aaec7..03d6919 100644
--- a/company/routes.py
+++ b/company/routes.py
@@ -87,8 +87,8 @@ def update_process():
     rid = request.json['rid']
     renew_data = request.json['renew_data']
     types = request.json['types']
-
-    info = update_process_scripts(types, rid, renew_data)
+    db = "综信评价数据" if types == "综合评价流程" else "ESG评价数据"
+    info = update_process_scripts(db, 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 9a917e3..387d2db 100644
--- a/company/scripts.py
+++ b/company/scripts.py
@@ -7,7 +7,7 @@ import pandas as pd
 
 from common.scripts import read_json_file, sub_dict, df_iterrows
 from company.db import find_data_in_tyc, insert_data_to_tfse, find_data_in_tfse, conserve_data_to_tfse, update_data, \
-    UPSERT_SINGLE_DATA
+    UPSERT_SINGLE_DATA, UPDATE_SINGLE_DATA
 from rating.scripts.risk_detail import associate_risk_detail, change_log_detail
 
 
@@ -820,11 +820,11 @@ def update_process_scripts(param1, param2, param3):
     """
     更新流程数据
     Parameters:
-        param1 dict 数据集名称
+        param1 dict 数据库名称
         param2 str 评价ID
         param3 dict 需要更新的数据
     Returns:
         result bool 更新结果
     """
-    info = update_data('评价', param1, {"评价ID": param2}, param3)
+    info = UPDATE_SINGLE_DATA(param1, "评价流程", {"评价ID": param2}, param3)
     return info