Mirrored_Repos/twitch-dl
1
0
Fork 0
mirror of https://github.com/ihabunek/twitch-dl synced 2024-08-30 18:32:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Download chunks to a temp file first

Browse Source
This commit is contained in:
Ivan Habunek 2022-08-14 11:33:23 +02:00
parent 721d78377e
commit cd445674e5
No known key found for this signature in database
GPG Key ID: CDBD63C43A30BB95
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
twitchdl/http.py
View File

@ -70,7 +70,10 @@ async def download(
progress: Progress,
token_bucket: AnyTokenBucket,
):
with open(target, "wb") as f:
# Download to a temp file first, then copy to target when over to avoid
# getting saving chunks which may persist if canceled or --keep is used
tmp_target = f"{target}.tmp"
with open(tmp_target, "wb") as f:
async with client.stream("GET", source) as response:
size = int(response.headers.get("content-length"))
progress.start(task_id, size)
@ -80,6 +83,7 @@ async def download(
token_bucket.advance(size)
progress.advance(task_id, size)
progress.end(task_id)
os.rename(tmp_target, target)
async def download_with_retries(