Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-14 21:52:56 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

🚑 修复 httpx 初始化传参错误 (#1926)

Browse Source
This commit is contained in:
molanp 2025-06-17 03:58:10 +08:00 committed by GitHub
parent 62b0b02466
commit ee9a2a6cb0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
zhenxun/utils/http_utils.py
View File

@ -25,6 +25,8 @@ from zhenxun.services.log import logger
from zhenxun.utils.message import MessageUtils
from zhenxun.utils.user_agent import get_user_agent
CLIENT_KEY = ["use_proxy", "proxy", "verify", "headers"]
def get_async_client(
proxies: dict[str, str] | None = None, verify: bool = False, **kwargs
@ -117,14 +119,11 @@ class AsyncHttpx:
for current_url in urls:
try:
logger.info(f"开始获取 {current_url}..")
async with cls._create_client(**kwargs) as client:
# 从 kwargs 中提取仅 client.get 支持的参数
get_kwargs = {
k: v
for k, v in kwargs.items()
if k not in ["use_proxy", "proxy", "verify", "headers"]
}
response = await client.get(current_url, **get_kwargs)
client_kwargs = {k: v for k, v in kwargs.items() if k in CLIENT_KEY}
for key in CLIENT_KEY:
kwargs.pop(key, None)
async with cls._create_client(**client_kwargs) as client:
response = await client.get(current_url, **kwargs)
if check_status_code and response.status_code != check_status_code:
raise HTTPStatusError(
@ -155,13 +154,11 @@ class AsyncHttpx:
返回:
Response: Response
"""
async with cls._create_client(**kwargs) as client:
head_kwargs = {
k: v
for k, v in kwargs.items()
if k not in ["use_proxy", "proxy", "verify"]
}
return await client.head(url, **head_kwargs)
client_kwargs = {k: v for k, v in kwargs.items() if k in CLIENT_KEY}
for key in CLIENT_KEY:
kwargs.pop(key, None)
async with cls._create_client(**client_kwargs) as client:
return await client.head(url, **kwargs)
@classmethod
async def post(cls, url: str, **kwargs) -> Response:
@ -178,13 +175,11 @@ class AsyncHttpx:
返回:
Response: Response。
"""
async with cls._create_client(**kwargs) as client:
post_kwargs = {
k: v
for k, v in kwargs.items()
if k not in ["use_proxy", "proxy", "verify"]
}
return await client.post(url, **post_kwargs)
client_kwargs = {k: v for k, v in kwargs.items() if k in CLIENT_KEY}
for key in CLIENT_KEY:
kwargs.pop(key, None)
async with cls._create_client(**client_kwargs) as client:
return await client.post(url, **kwargs)
@classmethod
async def get_content(cls, url: str, **kwargs) -> bytes: