144 lines
5.5 KiB
Python
144 lines
5.5 KiB
Python
|
import datetime
|
|||
|
|
import asyncio
|
|||
|
|
import httpx
|
|||
|
|
from loguru import logger
|
|||
|
|
from utils.decorators import handle_exceptions_unified
|
|||
|
|
|
|||
|
|
|
|||
|
|
class BrowserApi:
|
|||
|
|
"""
|
|||
|
|
浏览器接口
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
self.local_url = 'http://127.0.0.1:54345'
|
|||
|
|
self.headers = {'Content-Type': 'application/json'}
|
|||
|
|
# 使用异步 HTTP 客户端，启用连接池和超时设置
|
|||
|
|
self.client = httpx.AsyncClient(
|
|||
|
|
base_url=self.local_url,
|
|||
|
|
headers=self.headers,
|
|||
|
|
timeout=httpx.Timeout(30.0, connect=10.0), # 总超时30秒，连接超时10秒
|
|||
|
|
limits=httpx.Limits(max_keepalive_connections=50, max_connections=100), # 连接池配置
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
async def __aenter__(self):
|
|||
|
|
"""异步上下文管理器入口"""
|
|||
|
|
return self
|
|||
|
|
|
|||
|
|
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
|||
|
|
"""异步上下文管理器出口，关闭客户端"""
|
|||
|
|
await self.aclose()
|
|||
|
|
|
|||
|
|
async def aclose(self):
|
|||
|
|
"""关闭 HTTP 客户端"""
|
|||
|
|
if self.client:
|
|||
|
|
await self.client.aclose()
|
|||
|
|
|
|||
|
|
# 打开指纹浏览器
|
|||
|
|
@handle_exceptions_unified()
|
|||
|
|
async def open_browser(self, id: str, jc: int = 0):
|
|||
|
|
"""
|
|||
|
|
打开指纹浏览器（异步优化版本）
|
|||
|
|
:param jc: 计次
|
|||
|
|
:param id: 浏览器id
|
|||
|
|
:return:http, pid
|
|||
|
|
"""
|
|||
|
|
if jc > 3:
|
|||
|
|
return None, None
|
|||
|
|
url = '/browser/open'
|
|||
|
|
data = {
|
|||
|
|
'id': id
|
|||
|
|
}
|
|||
|
|
try:
|
|||
|
|
res = await self.client.post(url, json=data)
|
|||
|
|
res.raise_for_status() # 检查 HTTP 状态码
|
|||
|
|
res_data = res.json()
|
|||
|
|
logger.info(f'打开指纹浏览器: {res_data}')
|
|||
|
|
if not res_data.get('success'):
|
|||
|
|
logger.error(f'打开指纹浏览器失败: {res_data}')
|
|||
|
|
return await self.open_browser(id, jc + 1)
|
|||
|
|
data = res_data.get('data')
|
|||
|
|
http = data.get('http')
|
|||
|
|
pid = data.get('pid')
|
|||
|
|
logger.info(f'打开指纹浏览器成功: {http}, {pid}')
|
|||
|
|
return http, pid
|
|||
|
|
except httpx.TimeoutException as e:
|
|||
|
|
logger.error(f'打开指纹浏览器超时: {e}')
|
|||
|
|
if jc < 3:
|
|||
|
|
return await self.open_browser(id, jc + 1)
|
|||
|
|
return None, None
|
|||
|
|
except httpx.RequestError as e:
|
|||
|
|
logger.error(f'打开指纹浏览器请求错误: {e}')
|
|||
|
|
if jc < 3:
|
|||
|
|
return await self.open_browser(id, jc + 1)
|
|||
|
|
return None, None
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f'打开指纹浏览器异常: {e}')
|
|||
|
|
if jc < 3:
|
|||
|
|
return await self.open_browser(id, jc + 1)
|
|||
|
|
return None, None
|
|||
|
|
|
|||
|
|
# 关闭指纹浏览器
|
|||
|
|
@handle_exceptions_unified()
|
|||
|
|
async def close_browser(self, id: str, jc: int = 0):
|
|||
|
|
"""
|
|||
|
|
关闭指纹浏览器（异步优化版本）
|
|||
|
|
:param jc: 计次
|
|||
|
|
:param id: 浏览器id
|
|||
|
|
:return:
|
|||
|
|
"""
|
|||
|
|
if jc > 3:
|
|||
|
|
return None
|
|||
|
|
url = '/browser/close'
|
|||
|
|
data = {
|
|||
|
|
'id': id
|
|||
|
|
}
|
|||
|
|
try:
|
|||
|
|
res = await self.client.post(url, json=data)
|
|||
|
|
res.raise_for_status() # 检查 HTTP 状态码
|
|||
|
|
res_data = res.json()
|
|||
|
|
logger.info(f'关闭指纹浏览器: {res_data}')
|
|||
|
|
if not res_data.get('success'):
|
|||
|
|
msg = res_data.get('msg', '')
|
|||
|
|
# 如果浏览器正在打开中，等待后重试（不是真正的错误）
|
|||
|
|
if '正在打开中' in msg or 'opening' in msg.lower():
|
|||
|
|
if jc < 3:
|
|||
|
|
# 等待 1-3 秒后重试（根据重试次数递增等待时间）
|
|||
|
|
wait_time = (jc + 1) * 1.0 # 第1次重试等1秒，第2次等2秒，第3次等3秒
|
|||
|
|
logger.info(f'浏览器正在打开中，等待 {wait_time} 秒后重试关闭: browser_id={id}')
|
|||
|
|
await asyncio.sleep(wait_time)
|
|||
|
|
return await self.close_browser(id, jc + 1)
|
|||
|
|
else:
|
|||
|
|
# 超过重试次数，记录警告但不作为错误
|
|||
|
|
logger.warning(f'关闭指纹浏览器失败（浏览器正在打开中，已重试3次）: browser_id={id}')
|
|||
|
|
return None
|
|||
|
|
else:
|
|||
|
|
# 其他错误，记录为错误并重试
|
|||
|
|
logger.error(f'关闭指纹浏览器失败: {res_data}')
|
|||
|
|
if jc < 3:
|
|||
|
|
await asyncio.sleep(0.5) # 短暂等待后重试
|
|||
|
|
return await self.close_browser(id, jc + 1)
|
|||
|
|
return None
|
|||
|
|
logger.info(f'关闭指纹浏览器成功: browser_id={id}')
|
|||
|
|
return True
|
|||
|
|
except httpx.TimeoutException as e:
|
|||
|
|
logger.error(f'关闭指纹浏览器超时: {e}')
|
|||
|
|
if jc < 3:
|
|||
|
|
await asyncio.sleep(1.0)
|
|||
|
|
return await self.close_browser(id, jc + 1)
|
|||
|
|
return None
|
|||
|
|
except httpx.RequestError as e:
|
|||
|
|
logger.error(f'关闭指纹浏览器请求错误: {e}')
|
|||
|
|
if jc < 3:
|
|||
|
|
await asyncio.sleep(1.0)
|
|||
|
|
return await self.close_browser(id, jc + 1)
|
|||
|
|
return None
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f'关闭指纹浏览器异常: {e}')
|
|||
|
|
if jc < 3:
|
|||
|
|
await asyncio.sleep(1.0)
|
|||
|
|
return await self.close_browser(id, jc + 1)
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
browser_api = BrowserApi()
|