15308555518/ tunasync-scripts
1
0
Fork 0
mirror of https://github.com/tuna/tunasync-scripts.git synced 2025-12-25 16:32:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

github-release: add timeout when reading from HTTP connection [ci skip] 

Signed-off-by: Shengqi Chen <harry-chen@outlook.com>
This commit is contained in:
Shengqi Chen 2025-09-13 15:34:11 +08:00
parent 6d38d8f625
commit 8abc8d3291
No known key found for this signature in database
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
github-release.py
View File

@ -4,7 +4,6 @@ import json
import logging
import os
import tempfile
import traceback
from datetime import datetime
from pathlib import Path
@ -45,6 +44,7 @@ def github_get(*args, **kwargs) -> requests.Response:
headers["Authorization"] = "token {}".format(os.environ["GITHUB_TOKEN"])
headers["User-Agent"] = UA
kwargs["headers"] = headers
kwargs["timeout"] = TIMEOUT_OPTION
return requests.get(*args, **kwargs)
@ -63,7 +63,8 @@ def do_download(
delete=False,
) as f:
tmp_dst_file = Path(f.name)
for chunk in r.iter_content(chunk_size=1024**2):
# download in 1MB chunks
for chunk in r.iter_content(chunk_size=1 << 20):
if chunk: # filter out keep-alive new chunks
f.write(chunk)
# f.flush()