Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2026-10-08 21:30:01 +08:00
Code Issues Actions 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
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__version__
+1 -1
View File
@@ -1 +1 @@
__version__: v0.2.2-f9c7360 __version__: v0.2.2-d539b1f
zhenxun/utils/http_utils.py
+5 -2
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)