diff --git a/Context/common.py b/Context/common.py index e11d49a..9d1cd08 100644 --- a/Context/common.py +++ b/Context/common.py @@ -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) diff --git a/ThirdPartyApi/Tianyancha/Apis.py b/ThirdPartyApi/Tianyancha/Apis.py index 617e164..fc9ba0b 100644 --- a/ThirdPartyApi/Tianyancha/Apis.py +++ b/ThirdPartyApi/Tianyancha/Apis.py @@ -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'] }) diff --git a/ThirdPartyApi/Yujingtong/Apis.py b/ThirdPartyApi/Yujingtong/Apis.py index f578f9a..cf268af 100644 --- a/ThirdPartyApi/Yujingtong/Apis.py +++ b/ThirdPartyApi/Yujingtong/Apis.py @@ -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 })