bug
This commit is contained in:
parent
fded6ec01e
commit
d11d0c4b18
|
@ -55,6 +55,6 @@ phone_msg_send = PhoneMsgSend(conf['hw_msg']['url'],
|
|||
phone_verify_code = PhoneVerifyCode(redis_pool, phone_msg_send)
|
||||
|
||||
# 天眼查接口
|
||||
tianyancha_api = TianyanchaApi(conf['tianyancha']['token'], mg_db)
|
||||
tianyancha_api = TianyanchaApi(conf['tianyancha']['token'], mongo_connect=mg_db)
|
||||
# 预警通接口
|
||||
yujingtong_api = YujingtongApi(conf['yujingtong']['token'], mg_db)
|
||||
yujingtong_api = YujingtongApi(conf['yujingtong']['token'], mongo_connect=mg_db)
|
||||
|
|
|
@ -35,11 +35,15 @@ class TianyanchaApi:
|
|||
天眼查接口
|
||||
"""
|
||||
|
||||
def __init__(self, token: str, config: TianyanchaConfig = TianyanchaConfig, mongo_connect: MongoConnect = None):
|
||||
def __init__(self, token: str, config: TianyanchaConfig = TianyanchaConfig,
|
||||
mongo_connect: MongoConnect = None,
|
||||
mongo_db_name="天眼查"
|
||||
):
|
||||
self.config = config
|
||||
self.token = token
|
||||
self.rq = rq
|
||||
self.mongo_connect = mongo_connect
|
||||
self.mongo_db_name = mongo_db_name
|
||||
|
||||
def get_data(self, api_url, keyword):
|
||||
url = f"{api_url}?keyword={keyword}"
|
||||
|
@ -47,14 +51,14 @@ class TianyanchaApi:
|
|||
|
||||
def get_data_core(self, full_url):
|
||||
if self.mongo_connect:
|
||||
item = self.mongo_connect.db["天眼查"].find_one({"url": full_url})
|
||||
item = self.mongo_connect.db[self.mongo_db_name].find_one({"url": full_url})
|
||||
return JsDict(item['data'])
|
||||
headers = {'Authorization': self.token}
|
||||
res = self.rq.get(full_url, headers=headers)
|
||||
data = res.json()
|
||||
if data['error_code'] == 0:
|
||||
if self.mongo_connect:
|
||||
self.mongo_connect.db["天眼查"].insert_one({
|
||||
self.mongo_connect.db[self.mongo_db_name].insert_one({
|
||||
"url": full_url,
|
||||
"data": data['result']
|
||||
})
|
||||
|
|
|
@ -54,10 +54,11 @@ class YujingtongApi:
|
|||
预警通接口
|
||||
"""
|
||||
|
||||
def __init__(self, token: str, config: YujingtongConfig = YujingtongConfig, mongo_connect: MongoConnect = None):
|
||||
def __init__(self, token: str, config: YujingtongConfig = YujingtongConfig, mongo_connect: MongoConnect = None,mongo_db_name="预警通"):
|
||||
self.config = config
|
||||
self.token = token
|
||||
self.mongo_connect = mongo_connect
|
||||
self.mongo_db_name= mongo_db_name
|
||||
|
||||
def get_list(self, api_url, keyword):
|
||||
"""
|
||||
|
@ -89,7 +90,7 @@ class YujingtongApi:
|
|||
|
||||
def get_data_core(self, full_url):
|
||||
if self.mongo_connect:
|
||||
item = self.mongo_connect.db["预警通"].find_one({"url": full_url})
|
||||
item = self.mongo_connect.db[self.mongo_db_name].find_one({"url": full_url})
|
||||
if item:
|
||||
return JsDict(item['data'])
|
||||
headers = {'Authorization': self.token}
|
||||
|
@ -98,7 +99,7 @@ class YujingtongApi:
|
|||
if data['returncode'] == 0:
|
||||
main_rating_data = data['data']['data']
|
||||
if self.mongo_connect:
|
||||
self.mongo_connect.db["预警通"].insert_one({
|
||||
self.mongo_connect.db[self.mongo_db_name].insert_one({
|
||||
"url": full_url,
|
||||
"data": data
|
||||
})
|
||||
|
|
Loading…
Reference in New Issue