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

🐛 fix(http_utils): 流式下载Content-Length错误

Browse Source
This commit is contained in:
AkashiCoin 2024-09-22 15:08:47 +08:00
parent f2c20b84bf
commit d539b1f0b1
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
zhenxun/utils/http_utils.py
View File

@ -286,7 +286,9 @@ class AsyncHttpx:
f"Path: {path.absolute()}"
)
async with aiofiles.open(path, "wb") as wf:
total = int(response.headers["Content-Length"])
total = int(
response.headers.get("Content-Length", 0)
)
with rich.progress.Progress( # type: ignore
rich.progress.TextColumn(path.name), # type: ignore
"[progress.percentage]{task.percentage:>3.0f}%", # type: ignore
@ -295,7 +297,8 @@ class AsyncHttpx:
rich.progress.TransferSpeedColumn(), # type: ignore
) as progress:
download_task = progress.add_task(
"Download", total=total
"Download",
total=total if total else None,
)
async for chunk in response.aiter_bytes():
await wf.write(chunk)