mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
4474eb68e2
Instead of directly writing the destination file, use the combination of a `.tmp` suffix and `os.replace` to guarantee that the destination file is written whole and not partially.
47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import os
|
|
import sys
|
|
|
|
import requests
|
|
|
|
|
|
class Logger:
|
|
def __init__(self):
|
|
self.verbose = False
|
|
|
|
def log(self, message):
|
|
if self.verbose:
|
|
sys.stderr.write(message + "\n")
|
|
|
|
def log_warning(self, message):
|
|
self.log("warning: " + message)
|
|
|
|
|
|
logger = Logger()
|
|
log = logger.log
|
|
log_warning = logger.log_warning
|
|
|
|
|
|
def download_file(url, local_filename):
|
|
log(f"Downloading {local_filename}")
|
|
try:
|
|
with requests.get(url, stream=True) as request:
|
|
if request.status_code == 200:
|
|
download_name = local_filename + ".tmp"
|
|
with open(download_name, "wb") as debug_file:
|
|
for chunk in request.iter_content(chunk_size=64 * 1024):
|
|
debug_file.write(chunk)
|
|
log("Downloaded")
|
|
os.replace(download_name, local_filename)
|
|
return True
|
|
elif request.status_code == 404:
|
|
log("URL was not found")
|
|
else:
|
|
log(f"URL returned status code {request.status_code}")
|
|
except requests.RequestException as e:
|
|
log(f"Exception while making request: {e}")
|
|
return False
|