初始化企业数据
This commit is contained in:
parent
d0f88fc933
commit
3a6ab2d99c
|
@ -6,15 +6,24 @@ from company.scripts import *
|
|||
company_route = Blueprint('company', __name__)
|
||||
|
||||
|
||||
@company_route.route('/basic_info', methods=['POST'])
|
||||
@company_route.route('/init_company_data', methods=['POST'])
|
||||
@verify_token
|
||||
def basic_info():
|
||||
"""
|
||||
基础信息清晰
|
||||
初始化企业数据
|
||||
包括 天眼查数据库、企业基本信息
|
||||
Parameters:
|
||||
company_name 企业名称
|
||||
Returns:
|
||||
res: desc
|
||||
"""
|
||||
cid = request.json['cid']
|
||||
company_name = request.json['company_name']
|
||||
basic_info_etl(company_name)
|
||||
|
||||
# 拉取企业数据
|
||||
info1 = drag_company_data_request(company_name)
|
||||
if info1 != "企业数据拉取成功":
|
||||
return {"info": "数据拉取失败"}, 210
|
||||
|
||||
# 清洗基本信息数据
|
||||
basic_info_etl(cid, company_name)
|
||||
|
|
|
@ -1,9 +1,32 @@
|
|||
def basic_info_etl(param):
|
||||
import json
|
||||
import requests
|
||||
|
||||
|
||||
def drag_company_data_request(company_name):
|
||||
"""
|
||||
从天眼查接口获取企业数据
|
||||
Parameters:
|
||||
company_name: 企业名称
|
||||
Returns:
|
||||
-
|
||||
"""
|
||||
url = "http://139.9.249.34:51009/api/tyc/drag_data"
|
||||
headers = {'token': "uzdq51N4!I0%HY4sCaQ!aeCSIDIVIdAM"}
|
||||
data = {"企业名称": company_name}
|
||||
res = requests.post(url=url, headers=headers, data=json.dumps(data))
|
||||
if res.status_code == 200:
|
||||
return "企业数据拉取成功"
|
||||
else:
|
||||
return "企业数据拉取失败"
|
||||
|
||||
|
||||
def basic_info_etl(cid, company_name):
|
||||
"""
|
||||
根据企业名称,查询天眼查数据库
|
||||
将数据按规定格式存储到股交企业数据库中
|
||||
Parameters:
|
||||
param: desc
|
||||
cid: 企业ID
|
||||
company_name: 企业名称
|
||||
Returns:
|
||||
res: desc
|
||||
-
|
||||
"""
|
Loading…
Reference in New Issue