mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
5d582a2eec
Rework how the cache folder for debug information is handled. This commit introduces an additional environment variable, `REVNG_CACHE_DIR` that allows overriding where semi-disposable files will be stored.
51 lines
1.2 KiB
Python
51 lines
1.2 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import requests
|
|
from xdg import xdg_cache_home
|
|
|
|
|
|
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 cache_directory() -> Path:
|
|
if "REVNG_CACHE_DIR" in os.environ:
|
|
return Path(os.environ["REVNG_CACHE_DIR"])
|
|
else:
|
|
return xdg_cache_home() / "revng"
|
|
|
|
|
|
def download_file(url, local_filename):
|
|
log(f"Downloading {local_filename}")
|
|
with requests.get(url, stream=True) as request:
|
|
if request.status_code == 200:
|
|
with open(local_filename, "wb") as debug_file:
|
|
for chunk in request.iter_content(chunk_size=64 * 1024):
|
|
debug_file.write(chunk)
|
|
log("Downloaded")
|
|
return True
|
|
elif request.status_code == 404:
|
|
log("URL was not found")
|
|
else:
|
|
log(f"URL returned status code {request.status_code}")
|
|
return False
|