Merge branch 'wsc' into 'master'
Wsc See merge request root/tfse_app_api!2
This commit is contained in:
commit
748a5f06ca
|
@ -83,3 +83,15 @@ def make_verify_code():
|
|||
ch = chr(random.randrange(ord('0'), ord('9') + 1))
|
||||
code += ch
|
||||
return code
|
||||
|
||||
|
||||
def sub_dict(param1, param2):
|
||||
"""
|
||||
获取字典的子集
|
||||
Parameters:
|
||||
param1: 原字典
|
||||
param2: 子集字段
|
||||
Returns:
|
||||
子集
|
||||
"""
|
||||
return dict((key, value) for key, value in param1.items() if key in param2)
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
from company.scripts import find_company_data_scripts, find_financial_index_detail_scripts
|
||||
from company.scripts import find_company_data_scripts, find_financial_index_detail_scripts, find_head_info_scripts
|
||||
|
||||
|
||||
class CompanyIndex:
|
||||
|
@ -21,3 +21,7 @@ class CompanyIndex:
|
|||
return '查询成功', result
|
||||
else:
|
||||
return '无数据', {}
|
||||
|
||||
def head_info_obj(self):
|
||||
result = find_head_info_scripts(self.cid)
|
||||
return "查询成功", result
|
||||
|
|
|
@ -23,4 +23,13 @@ def company_page(**kwargs):
|
|||
info, result = company_index.get_company_data_obj()
|
||||
else:
|
||||
info, result = company_index.get_index_detail_obj()
|
||||
return {"info": info, "result": result}
|
||||
return {"info": info, "result": result}, 200
|
||||
|
||||
|
||||
@company_route.route('/head_info', methods=['GET'])
|
||||
@verify_token
|
||||
def head_info(**kwargs):
|
||||
company_index = CompanyIndex()
|
||||
company_index.cid = kwargs['cid']
|
||||
info, result = company_index.head_info_obj()
|
||||
return {"info": info, "result": result}, 200
|
||||
|
|
|
@ -1,6 +1,7 @@
|
|||
import json
|
||||
import pandas as pd
|
||||
|
||||
from common.scripts import sub_dict
|
||||
from company.db import find_data_in_company
|
||||
|
||||
|
||||
|
@ -8,6 +9,23 @@ def find_company_data_scripts(table, cid):
|
|||
return find_data_in_company(table, {"企业ID": cid})
|
||||
|
||||
|
||||
def find_head_info_scripts(cid):
|
||||
data = find_data_in_company('综合评价分析', {"企业ID": cid})
|
||||
if data:
|
||||
return sub_dict(data[0], ['企业名称', '综合信用等级'])
|
||||
else:
|
||||
company = find_data_in_company('公司基本信息', {"企业ID": cid})[0]
|
||||
result = {
|
||||
"企业名称": company['企业名称'],
|
||||
"综合信用等级": {
|
||||
"评价时间": "N/A",
|
||||
"信用等级": "N/A",
|
||||
"信用评分": "N/A"
|
||||
},
|
||||
}
|
||||
return result
|
||||
|
||||
|
||||
def find_financial_index_detail_scripts(table, cid):
|
||||
df = pd.DataFrame(find_data_in_company(table, {"企业ID": cid})).sort_values('年报期', ascending=False)
|
||||
cols = [
|
||||
|
|
|
@ -216,21 +216,26 @@ def exec_general_rating_scripts(cid):
|
|||
risks["风险数据"].pop('评价ID')
|
||||
rating_inputs = {**inputs, **risks}
|
||||
|
||||
# main
|
||||
url = "http://139.9.249.34:51012/tfse_rating/rating/general_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()),
|
||||
"进行状态": "完成",
|
||||
"报告fid": "xxx",
|
||||
"证书fid": "xxx"
|
||||
}
|
||||
update_rating_records({"评价ID": rid}, update_data)
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
# 打分模型
|
||||
def general_model():
|
||||
url = "http://139.9.249.34:51012/tfse_rating/rating/general_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()),
|
||||
"进行状态": "完成",
|
||||
"报告fid": None,
|
||||
"证书fid": None
|
||||
}
|
||||
update_rating_records({"评价ID": rid}, update_data)
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
# 生成报告
|
||||
def gen_report():
|
||||
pass
|
||||
|
|
Loading…
Reference in New Issue