From fc34ab388e11f8637ceea8bf7cdb1aac5e4dd8fb Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 21 Apr 2024 22:03:11 +0200 Subject: Adding upstream version 0.10.1. Signed-off-by: Daniel Baumann --- eos_downloader/download.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'eos_downloader/download.py') diff --git a/eos_downloader/download.py b/eos_downloader/download.py index 2c9576e..df3c381 100644 --- a/eos_downloader/download.py +++ b/eos_downloader/download.py @@ -26,6 +26,11 @@ from rich.progress import ( console = rich.get_console() done_event = Event() +REQUEST_HEADERS = { + "Content-Type": "application/json", + "User-Agent": "Chrome/123.0.0.0", +} + def handle_sigint(signum: Any, frame: Any) -> None: """Progress bar handler""" @@ -64,7 +69,7 @@ class DownloadProgressBar: self, task_id: TaskID, url: str, path: str, block_size: int = 1024 ) -> bool: """Copy data from a url to a local file.""" - response = requests.get(url, stream=True, timeout=5) + response = requests.get(url, stream=True, timeout=5, headers=REQUEST_HEADERS) # This will break if the response doesn't contain content length self.progress.update(task_id, total=int(response.headers["Content-Length"])) with open(path, "wb") as dest_file: -- cgit v1.2.3