Preallocate space on disk before downloading files

FlashpointProject · Oct 2, 2019 · 820743c · 820743c
1 parent ab83427
commit 820743c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/update.py b/update.py
@@ -27,6 +27,9 @@ def download_file(session, url, dest):
     with session.get(url, stream=True, timeout=10) as r:
         r.raise_for_status()
         with open(dest, 'wb') as f:
+            if 'Content-Length' in r.headers:
+                length = int(r.headers['Content-Length'])
+                f.truncate(length)
             shutil.copyfileobj(r.raw, f)
 
 # Fix for "read-only" files on Windows