From 6f9a91f1cb92530ad50c608c1f59ff38acb1b1f5 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Mon, 14 Jul 2025 13:46:47 -0500
Subject: Provide timeouts for requests calls.

---
 uploader/oauth2/client.py       | 3 ++-
 uploader/publications/pubmed.py | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

(limited to 'uploader')

diff --git a/uploader/oauth2/client.py b/uploader/oauth2/client.py
index 12fbf80..b94a044 100644
--- a/uploader/oauth2/client.py
+++ b/uploader/oauth2/client.py
@@ -43,7 +43,8 @@ def __fetch_auth_server_jwks__() -> KeySet:
     return KeySet([
         JsonWebKey.import_key(key)
         for key in requests.get(
-                urljoin(authserver_uri(), "auth/public-jwks")
+                urljoin(authserver_uri(), "auth/public-jwks"),
+                timeout=(9.13, 20)
         ).json()["jwks"]])
 
 
diff --git a/uploader/publications/pubmed.py b/uploader/publications/pubmed.py
index ed9b652..e8b0584 100644
--- a/uploader/publications/pubmed.py
+++ b/uploader/publications/pubmed.py
@@ -88,7 +88,8 @@ def fetch_publications(pubmed_ids: tuple[int, ...]) -> tuple[dict, ...]:
                 "db": "pubmed",
                 "retmode": "xml",
                 "id": ",".join(str(item) for item in pubmed_ids)
-            })
+            },
+            timeout=(9.13, 20))
 
         if response.status_code == 200:
             return __process_pubmed_publication_data__(response.text)
-- 
cgit v1.2.3