arista/eos-downloader
1
0
Fork 0
Code Issues Activity

Merging upstream version 0.10.1.

Browse source 
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
Daniel Baumann 2025-02-05 13:51:38 +01:00
parent 6d84bde19f
commit 19825a5820
Signed by: daniel
GPG key ID: FBB4F0E80A80222F
4 changed files with 26 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

7
eos_downloader/download.py
View file

@ -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: