From c1bdf05e08f7e8d8ba3d3f392784b291077ee520 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sun, 21 Apr 2024 22:03:28 +0200
Subject: Merging upstream version 0.10.1.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 .github/workflows/pr-management.yml | 10 +++++-----
 eos_downloader/download.py          |  7 ++++++-
 eos_downloader/object_downloader.py | 17 +++++++++++++----
 pyproject.toml                      |  4 ++--
 4 files changed, 26 insertions(+), 12 deletions(-)

diff --git a/.github/workflows/pr-management.yml b/.github/workflows/pr-management.yml
index ec36edc..814bb39 100644
--- a/.github/workflows/pr-management.yml
+++ b/.github/workflows/pr-management.yml
@@ -13,7 +13,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.8", "3.9", "3.10"]
+        python-version: ["3.8", "3.9", "3.10", "3.11"]
     steps:
     - uses: actions/checkout@v4
     - uses: actions/setup-python@v5
@@ -27,7 +27,7 @@ jobs:
     needs: [pre-commit]
     strategy:
       matrix:
-        python-version: ["3.8", "3.9", "3.10"]
+        python-version: ["3.8", "3.9", "3.10", "3.11"]
 
     steps:
       - uses: actions/checkout@v4
@@ -55,7 +55,7 @@ jobs:
 
     strategy:
       matrix:
-        python: ["3.8", "3.9", "3.10"]
+        python: ["3.8", "3.9", "3.10", "3.11"]
 
     steps:
       - uses: actions/checkout@v4
@@ -78,7 +78,7 @@ jobs:
 
     strategy:
       matrix:
-        python: ["3.8", "3.9", "3.10"]
+        python: ["3.8", "3.9", "3.10", "3.11"]
 
     steps:
       - uses: actions/checkout@v4
@@ -101,7 +101,7 @@ jobs:
 
     strategy:
       matrix:
-        python: ["3.8", "3.9", "3.10"]
+        python: ["3.8", "3.9", "3.10", "3.11"]
 
     steps:
       - uses: actions/checkout@v4
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:
diff --git a/eos_downloader/object_downloader.py b/eos_downloader/object_downloader.py
index 30cf9df..c107fee 100644
--- a/eos_downloader/object_downloader.py
+++ b/eos_downloader/object_downloader.py
@@ -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 [
diff --git a/pyproject.toml b/pyproject.toml
index f125def..b06934a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "eos_downloader"
-version = "v0.10.0"
+version = "v0.10.1"
 readme = "README.md"
 authors = [{ name = "Thomas Grimonet", email = "thomas.grimonet@gmail.com" }]
 maintainers = [
@@ -94,7 +94,7 @@ namespaces = false
 # Version
 ################################
 [tool.bumpver]
-current_version = "0.10.0"
+current_version = "0.10.1"
 version_pattern = "MAJOR.MINOR.PATCH"
 commit_message  = "bump: Version {old_version} -> {new_version}"
 commit = true
-- 
cgit v1.2.3