DigitalMgnt
/
urban-investment-research
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bug

This commit is contained in:
wcq 2023-03-29 11:30:23 +08:00
parent fded6ec01e
commit d11d0c4b18
3 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Context/common.py
View File

@ -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)

10
ThirdPartyApi/Tianyancha/Apis.py
View File

@ -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']
})

7
ThirdPartyApi/Yujingtong/Apis.py
View File

@ -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
})