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

17
eos_downloader/object_downloader.py
View file

@ -40,7 +40,7 @@ from eos_downloader import (
MSG_TOKEN_EXPIRED,
)
from eos_downloader.data import DATA_MAPPING
from eos_downloader.download import DownloadProgressBar
from eos_downloader.download import DownloadProgressBar, REQUEST_HEADERS
# logger = logging.getLogger(__name__)
@ -263,7 +263,10 @@ class ObjectDownloader:
self.authenticate()
jsonpost = {"sessionCode": self.session_id}
result = requests.post(
ARISTA_SOFTWARE_FOLDER_TREE, data=json.dumps(jsonpost), timeout=self.timeout
ARISTA_SOFTWARE_FOLDER_TREE,
data=json.dumps(jsonpost),
timeout=self.timeout,
headers=REQUEST_HEADERS,
)
try:
folder_tree = result.json()["data"]["xml"]
@ -332,7 +335,10 @@ class ObjectDownloader:
self.authenticate()
jsonpost = {"sessionCode": self.session_id, "filePath": remote_file_path}
result = requests.post(
ARISTA_DOWNLOAD_URL, data=json.dumps(jsonpost), timeout=self.timeout
ARISTA_DOWNLOAD_URL,
data=json.dumps(jsonpost),
timeout=self.timeout,
headers=REQUEST_HEADERS,
)
if "data" in result.json() and "url" in result.json()["data"]:
# logger.debug('URL to download file is: {}', result.json())
@ -421,7 +427,10 @@ class ObjectDownloader:
jsonpost = {"accessToken": credentials}
result = requests.post(
session_code_url, data=json.dumps(jsonpost), timeout=self.timeout
session_code_url,
data=json.dumps(jsonpost),
timeout=self.timeout,
headers=REQUEST_HEADERS,
)
if result.json()["status"]["message"] in [