dataplatform/Utils/TycApi/Configure.py

73 lines
1.8 KiB
Python

import time
from Utils.DataBase.MongoHelperUtils import get_tyc_mongodb
token = "32737167-cb63-4ce9-9397-d66169488f51"
headers = {'Authorization': token}
# 数据库参数
DB_GSBJ = '公司背景'
DB_SFFX = '司法风险'
DB_JYFX = '经营风险'
DB_TYFX = '天眼风险'
DB_JYZK = '经营状况'
DB_ZSCQ = '知识产权'
DB_GSFZ = '公司发展'
def judege_api_code(code):
"""
判断状态码
Parameters:
code: 传入状态码,返回状态码对应状态说明
Returns:
状态码说明信息
"""
if code == 0:
return '请求成功'
elif code == 300000:
return '无数据'
elif code == 300001:
return '请求失败'
elif code == 300002:
return '账号失效'
elif code == 300003:
return '账号过期'
elif code == 300004:
return '访问频率过快'
elif code == 300005:
return '无权限访问此api'
elif code == 300006:
return '余额不足'
elif code == 300007:
return '剩余次数不足'
elif code == 300008:
return '缺少必要参数'
elif code == 300009:
return '账号信息有误'
elif code == 300010:
return 'URL不存在'
elif code == 300011:
return '此IP无权限访问此api'
elif code == 300012:
return '报告生成中'
def conserve_data(db, name, key, value):
"""
保存天眼查api接口数据
传入数据库名称,企业名称,集合名称,需要保存的数据
"""
tyc_mongodb = get_tyc_mongodb()
insert_data = dict()
insert_data['企业名称'] = name
insert_data['录入时间'] = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
insert_data[key] = value
# 数据库操作
tyc_mongodb.upsert_data(
db,
key,
{"企业名称": name},
insert_data
)