esg问卷，所属行业自动填入

esg/esg_db.py

@@ -101,3 +101,17 @@ def insert_process_records(data):
     """
     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

esg/esg_impl.py

@@ -38,6 +38,18 @@ def get_current_year():
     return current_year
 
 
+def get_industry_type(cid):
+    """
+    获取行业类型
+    Parameters:
+        cid 企业ID
+    Returns:
+        industry_type 行业类型
+    """
+
+    return 0
+
+
 def gen_new_rid():
     """
     生成新的评价ID，如果该ID存在，则重新生成
@@ -147,6 +159,7 @@ def start_esg_rating_script(cid):
         new_input['企业ID'] = cid
         new_input['企业名称'] = company_name
         new_input['评价年度'] = get_current_year()
+        new_input['所属行业'] = get_industry_type(cid)
         new_input['公司当年收入（万元）'] = dict()
         for y in range(3):
             new_input['公司当年收入（万元）'][report_dates[y]] = None