fix: 修改代理获取方法以使用区域代码并更新全局代理字典

This commit is contained in:
晓丰 2025-05-17 12:43:46 +08:00
parent fe3a8bea4c
commit 8239dde04c
2 changed files with 3 additions and 3 deletions

3
DB.py
View File

@ -252,12 +252,11 @@ class DBVidcon:
self.cursor.close() self.cursor.close()
self.conn.close() self.conn.close()
def get_proxy(self, region_name: str) -> str: def get_proxy(self, region_code: str) -> str:
""" """
Redis 队列 proxy_queue:<region_code> 弹出一个代理并返回 Redis 队列 proxy_queue:<region_code> 弹出一个代理并返回
如果队列为空阻塞 如果队列为空阻塞
""" """
region_code = self.get_proxy_parameter(region_name)
proxy = "" proxy = ""
while True: while True:
key = f"proxy_queue:{region_code}" key = f"proxy_queue:{region_code}"

View File

@ -694,8 +694,9 @@ def integrate_data():
else: else:
for index, (payload, kitem) in enumerate(keywords): for index, (payload, kitem) in enumerate(keywords):
try: try:
proxiesdict = db.get_proxy_agent_dict()
global Gproxies global Gproxies
Gproxies = kitem['rn'] Gproxies = proxiesdict[kitem['rn']]
v_list = get_searchInfo(kitem['keyword']) v_list = get_searchInfo(kitem['keyword'])
if not v_list: if not v_list: