tfse-etl-api-v0.2/company/scripts.py

64 lines
2.5 KiB
Python

import json
import time
import requests
from common.scripts import file_path, read_json_file
from company.db import find_data_in_tyc
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:
cid: 企业ID
company_name: 企业名称
Returns:
-
"""
basic = read_json_file(file_path('/company/template/基本信息.json'))
basic['企业ID'] = cid
basic['更新日期'] = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
basic_tyc = find_data_in_tyc('公司背景', '基本信息', {"企业名称": company_name})
basic['企业名称'] = basic_tyc[0]['企业名称']
basic['工商信息']['企业状态'] = basic_tyc[0]['基本信息']['regStatus']
basic['工商信息']['法定代表人'] = basic_tyc[0]['基本信息']['legalPersonName']
basic['工商信息']['企业类型'] = basic_tyc[0]['基本信息']['companyOrgType']
basic['工商信息']['纳税人识别号'] = basic_tyc[0]['基本信息']['taxNumber']
basic['工商信息']['经营范围'] = basic_tyc[0]['基本信息']['businessScope']
basic['工商信息']['注册资本'] = basic_tyc[0]['基本信息']['regCapital']
basic['工商信息']['实缴资本'] = basic_tyc[0]['基本信息']['actualCapital']
basic['工商信息']['注册地址'] = basic_tyc[0]['基本信息']['regLocation']
basic['工商信息']['登记机关'] = basic_tyc[0]['基本信息']['regInstitute']
basic['工商信息']['行业'] = basic_tyc[0]['基本信息']['industry']
basic['工商信息']['人员规模'] = basic_tyc[0]['基本信息']['staffNumRange']
basic['工商信息']['参保人数'] = basic_tyc[0]['基本信息']['socialStaffNum']
basic['工商信息']['小微企业'] = "" if basic_tyc[0]['基本信息']['regStatus'] == 1 else ""
pass
if __name__ == '__main__':
basic_info_etl('xxx', '德阳市德润达机械有限公司')