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

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

Browse Source
This commit is contained in:
AkashiCoin 2024-09-22 17:12:21 +08:00 committed by GitHub
parent 32c11b9919
commit 207f947a1c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
__version__
View File

@ -1 +1 @@
__version__: v0.2.2-f9c7360 __version__: v0.2.2-d539b1f

7
zhenxun/utils/http_utils.py
View File

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