update 报告生成逻辑修改

rating/ReportEtl.py

@@ -124,7 +124,3 @@ if __name__ == '__main__':
     db_data = get_tfse_data(info)
     r = CleanModel(info, db_data).storage_data()
     print(r)
-
-    """
-    1.公司背景中的year_03，share_holders
-    """

rating/db.py

@@ -9,7 +9,7 @@ client_tyc = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format(USER, PASSWORD_T
 
 def find_tyc_db(**kwargs):
     """
-    查询数据库记录
+    查询天眼查数据库记录
     Parameters:
         kwargs: 查询条件
     Returns:
@@ -23,7 +23,7 @@ def find_tyc_db(**kwargs):
 
 def find_tfse_db(**kwargs):
     """
-    查询数据库记录
+    查询天府股交数据库记录
     Parameters:
         kwargs: 查询条件
     Returns:
@@ -37,12 +37,12 @@ def find_tfse_db(**kwargs):
 
 def conserve_data(**kwargs):
     """
-    保存数据
+    根据评价ID保存数据
     Parameters:
         kwargs: 查询条件
     Returns:
         保存结果
     """
     collection = client_tfse[kwargs['db']][kwargs['col']]
-    collection.update_one({"企业ID": kwargs['cid']}, {"$set": kwargs['insert_data']}, upsert=True)
+    collection.update_one({"评价ID": kwargs['rid']}, {"$set": kwargs['insert_data']}, upsert=True)
     return '保存成功'