Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2026-10-05 11:49:59 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 文件下载逻辑修改

Browse Source
This commit is contained in:
AkashiCoin
2024-09-03 17:13:31 +08:00
parent 682d19aa2b
commit ce97dedc5f
3 changed files with 22 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zhenxun/utils/http_utils.py
+16 -14
View File
@@ -196,18 +196,19 @@ class AsyncHttpx:
for _ in range(3):
if not stream:
try:
content = (
await cls.get(
url,
params=params,
headers=headers,
cookies=cookies,
use_proxy=use_proxy,
proxy=proxy,
timeout=timeout,
**kwargs,
)
).content
response = await cls.get(
url,
params=params,
headers=headers,
cookies=cookies,
use_proxy=use_proxy,
proxy=proxy,
timeout=timeout,
follow_redirects=True,
**kwargs,
)
response.raise_for_status()
content = response.content
async with aiofiles.open(path, "wb") as wf:
await wf.write(content)
logger.info(f"下载 {url} 成功.. Path：{path.absolute()}")
@@ -230,8 +231,10 @@ class AsyncHttpx:
headers=headers,
cookies=cookies,
timeout=timeout,
follow_redirects=True,
**kwargs,
) as response:
response.raise_for_status()
logger.info(
f"开始下载 {path.name}.. Path: {path.absolute()}"
)
@@ -347,10 +350,9 @@ class AsyncHttpx:
begin_time = time.time()
response = await client.head(url=url, timeout=6)
response.raise_for_status()
elapsed_time = (time.time() - begin_time) * 1000
content_length = int(response.headers["content-length"])
content_length = int(response.headers.get("content-length", 0))
return {
"url": url,