Merge branch 'legion' into 'master'

update 修改esg相关数据库

See merge request root/tfse_app_api!5

esg/esg_db.py

@@ -1,6 +1,10 @@
 import pymongo
 
-client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format('root', 'sromitdTW569kC#M', '116.63.130.34', 27018))
+DB_HOST = "116.63.130.34"
+DB_PASS = "UTlC9cCoglD1cI1*"
+DB_USER = "root"
+DB_PORT = "27021"
+client = pymongo.MongoClient('mongodb://{}:{}@{}:{}'.format(DB_USER, DB_PASS, DB_HOST, DB_PORT))
 
 
 def find_rating_records(param):
@@ -11,7 +15,7 @@ def find_rating_records(param):
     Returns:
         record: 查询结果
     """
-    collection = client['评价']['评价记录']
+    collection = client['企业数据']['评价记录']
     data = collection.find(param, {'_id': False})
     record = list(data)
     return record
@@ -25,7 +29,7 @@ def insert_rating_records(data):
     Returns:
         -
     """
-    collection = client['评价']['评价记录']
+    collection = client['企业数据']['评价记录']
     collection.insert_one(data)
 
 
@@ -37,7 +41,7 @@ def find_rating_inputs(param):
     Returns:
         records: 查询结果
     """
-    collection = client['评价']['ESG评价填报']
+    collection = client['ESG评价数据']['填报数据']
     data = collection.find(param, {'_id': False})
     record = list(data)
     return record
@@ -51,7 +55,7 @@ def insert_general_input(data):
     Returns:
         -
     """
-    collection = client['评价']['ESG评价填报']
+    collection = client['ESG评价数据']['填报数据']
     collection.insert_one(data)
 
 
@@ -64,7 +68,7 @@ def update_general_input(rid, data):
     Returns:
         res: desc
     """
-    collection = client['评价']['ESG评价填报']
+    collection = client['ESG评价数据']['填报数据']
     collection.update_one({"评价ID": rid}, {"$set": data})
 
 
@@ -72,7 +76,7 @@ def find_risk_data(rid):
     """
     查找评价所需的风险数据
     """
-    collection = client['评价']['ESG风险数据']
+    collection = client['ESG评价数据']['风险数据']
     data = collection.find({"评价ID": rid}, {'_id': False})
     record = list(data)
     return record
@@ -87,7 +91,7 @@ def update_rating_records(find, data):
     Returns:
         -
     """
-    collection = client['评价']['评价记录']
+    collection = client['企业数据']['评价记录']
     collection.update_one(find, {"$set": data})
 
 
@@ -99,19 +103,6 @@ def insert_process_records(data):
     Returns:
         -
     """
-    collection = client['评价']['ESG评价流程']
+    collection = client['ESG评价数据']['评价流程']
     collection.insert_one(data)
 
-
-def find_industry_by_cid(cid):
-    """
-    根据cid查询企业所属行业
-    Args:
-        cid: 企业ID
-    Returns:
-        industry: 所属行业
-    """
-    collection = client['企业']['公司基本信息']
-    data = collection.find({"企业ID": cid}, {'_id': False})
-    industry = data[0]['工商信息']['行业']
-    return industry