tfse-app-api-v0.2/esg/esg_impl.py

272 lines
9.9 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

from user.user_db import FIND_USER_INFO
from user.user_impl import check_verified
from esg.esg_utils import *
from esg.esg_db import *
def rating_records_by_cid(cid):
"""
查询一家企业的所有评价记录
Parameters:
cid: 企业ID
Returns:
records: 企业的评价记录
"""
report_link = '/file/get_company_report?file_id={}'
certification_link = "/file/get_certification?file_id={}"
records = find_rating_records({"企业ID": cid, "评价项目": "ESG评价"})
for record in records:
if records is not None:
record['报告'] = report_link.format(record['报告fid']) if record['报告fid'] is not None else None
record['证书'] = certification_link.format(record['证书fid']) if record['证书fid'] is not None else None
record.pop('企业ID')
record.pop('报告fid')
record.pop('证书fid')
return records
def start_esg_rating_script(cid):
"""
开始进行esg评价的数据填报
Parameters:
cid: 企业ID
Returns:
record 填报信息
"""
# 检查本年度填报状态
def check_input_data():
"""
input_data: 已存在的填报数据
根据企业ID+当前年度,查询是否存在填报数据
若存在填报数据,则返回填报数据
若不存在填报数据则返回False
"""
exist_input_form = find_rating_inputs({"企业ID": cid, "评价年度": get_current_year()})
return exist_input_form if exist_input_form else False
# 存在填报数据情况
def case_input_exist(param):
"""
param: dict 填报数据
根据评价ID查询评价进行状态
若评价已完成,返回‘本年度已评价’的提示,且返回空对象
若评价进行中,返回‘继续填报’的提示,且返回填报数据
"""
rating_records = find_rating_records({"评价ID": param[0]["评价ID"]})
param[0].pop('评价ID')
if rating_records:
rating_status = rating_records[0]['进行状态']
else:
rating_status = None
return ("本年度已评价", {}) if rating_status == '完成' else ("继续填报", param)
# 不存在填报数据情况
def case_input_not_exist():
# new_rid: 新生成填报ID
new_rid = gen_new_rid()
# report_dates: 生成报告期
report_dates = make_3_report_dates()
# company_name: 企业名称
company_name = FIND_USER_INFO({"企业ID": cid})[0]['企业名称']
def prepare_input_template():
"""
根据模板生成新的填报数据
input_template: 填报数据模板
"""
input_template = get_data_template('esg_input')
input_template['评价ID'] = new_rid
input_template['企业ID'] = cid
input_template['企业名称'] = company_name
input_template['评价年度'] = get_current_year()
input_template['公司当年收入(万元)'] = dict()
for y in range(3):
input_template['公司当年收入(万元)'][report_dates[y]] = None
for sheets in ['环境问卷', '社会问卷', '治理问卷']:
for i in range(3):
input_template[sheets]['近三年公司数据'][i]['年份'] = report_dates[i]
insert_general_input(input_template)
def prepare_rating_record_template():
"""
根据模板生成新的评价记录数据
rating_result_template: 评价记录模板
"""
rating_record_template = get_data_template('rating')
rating_record_template['评价ID'] = new_rid
rating_record_template['企业ID'] = cid
rating_record_template['评价项目'] = "ESG评价"
rating_record_template['评价方式'] = "企业申报"
rating_record_template['进行状态'] = "进行"
insert_rating_records(rating_record_template)
def prepare_process_template():
"""
根据数据模板生成新的评价流程数据
process_template: 评价流程
"""
process_template = get_data_template('esg_process')
process_template['评价ID'] = new_rid
process_template['企业ID'] = cid
insert_process_records(process_template)
# 评价流程
prepare_rating_record_template()
# 评价记录
prepare_process_template()
# 填报数据
prepare_input_template()
# 风险数据
prepare_risk_data(cid, new_rid, company_name)
return "开始填报", check_input_data()
# 主方法
def main_job():
# 检查是否认证
if not check_verified(cid):
return "该企业未认证", {}
# 检查是否有填报记录
input_data = check_input_data()
if input_data:
return case_input_exist(input_data)
else:
return case_input_not_exist()
return main_job()
def save_general_input_form(cid, input_form):
"""
保存综合信用评价填报
Parameters:
cid: 企业ID
input_form: 填报表单
Returns:
res: desc
"""
# 保存填报数据
rid = find_rating_records({"企业ID": cid, "进行状态": "进行", "评价项目": "ESG评价"})[0]['评价ID']
update_general_input(rid, input_form)
update_process_status(rid, {"评价流程.数据填报": 1})
def exec_esg_rating_scripts(cid):
"""
开始esg评价流程
Parameters:
cid: 企业ID
Returns:
res: desc
"""
# prepare
record = find_rating_records({"企业ID": cid, "进行状态": "进行", "评价项目": "ESG评价"})
if not record:
return False, "没有正在进行的评价"
# Parameters
rid = record[0]["评价ID"]
inputs = find_rating_inputs({"评价ID": rid})[0]
risks = {"风险数据": find_risk_data(rid)[0]}
risks["风险数据"].pop('企业ID')
risks["风险数据"].pop('评价ID')
risks["风险数据"].pop('更新日期')
rating_inputs = {**inputs, **risks}
# 打分模型
def esg_model():
url = "http://api.fecribd.com/tfse_rating/esg/esg_model"
token = "X0gSlC!YE8jmr2jJr&ilcjS83j!tsoh5"
headers = {'token': token, "Content-Type": "application/json;charset=UTF-8"}
data = json.dumps({"input_data": rating_inputs})
res = requests.post(url, headers=headers, data=data)
if res.status_code == 200:
update_data = {
"评价结果": json.loads(res.text)['result'],
"评价时间": time.strftime("%Y-%m-%d", time.localtime())
}
update_rating_records({"评价ID": rid}, update_data)
update_process_status(rid, {"评价流程.模型打分": 1})
return True
else:
update_process_status(rid, {"评价流程.模型打分": -1})
return False
# 生成报告
def gen_report():
url = "http://api.fecribd.com/tfse_rating/report/generate_esg_report"
token = "X0gSlC!YE8jmr2jJr&ilcjS83j!tsoh5"
headers = {'token': token, "Content-Type": "application/json;charset=UTF-8"}
data = {
"company": rating_inputs['企业名称'],
"cid": rating_inputs['企业ID'],
"rid": rating_inputs['评价ID']
}
res = requests.post(url, headers=headers, data=json.dumps(data))
if res.status_code == 200:
update_data = {
"报告fid": json.loads(res.text)['result']['FileID']
}
update_rating_records({"评价ID": rid}, update_data)
update_process_status(rid, {"评价流程.报告生成": 1})
return True
else:
update_process_status(rid, {"评价流程.报告生成": -1})
return False
# 生成证书
def gen_certified():
url = "http://api.fecribd.com/tfse_rating/certificate/generate_certificate"
token = "X0gSlC!YE8jmr2jJr&ilcjS83j!tsoh5"
headers = {'token': token, "Content-Type": "application/json;charset=UTF-8"}
data = {
"company": rating_inputs['企业名称'],
"cid": rating_inputs['企业ID'],
"rid": rating_inputs['评价ID'],
"type": "esg"
}
res = requests.post(url, headers=headers, data=json.dumps(data))
if res.status_code == 200:
update_data = {
"证书fid": json.loads(res.text)['result']['FileID']
}
update_rating_records({"评价ID": rid}, update_data)
update_process_status(rid, {"评价流程.证书生成": 1})
return True
else:
update_process_status(rid, {"评价流程.证书生成": -1})
return False
# 生成企业主页数据
def esg_index():
url = "http://api.fecribd.com/etl_tfse/company/esg_index_etl"
token = "dmfd7FshT!5Wng9^gcCcQV7T6FBnVgl4"
headers = {'token': token, "Content-Type": "application/json;charset=UTF-8"}
data = {
"rid": rating_inputs['评价ID']
}
res = requests.post(url, headers=headers, data=json.dumps(data))
if res.status_code == 200:
update_process_status(rid, {"评价流程.主页ESG数据生成": 1})
return True
else:
update_process_status(rid, {"评价流程.主页ESG数据生成": -1})
return False
if not esg_model():
return False, "打分失败"
if not gen_report():
return False, "生成报告失败"
if not gen_certified():
return False, "生成证书失败"
if not esg_index():
return False, "准备主页数据失败"
update_rating_records({"评价ID": rid}, {"进行状态": "完成"})
return True, "执行成功"