diff --git a/esg/esg_db.py b/esg/esg_db.py index 6e4921a..ec6a424 100644 --- a/esg/esg_db.py +++ b/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 diff --git a/esg/esg_impl.py b/esg/esg_impl.py index 48c0011..f0b5ea4 100644 --- a/esg/esg_impl.py +++ b/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