aboutsummaryrefslogtreecommitdiff
path: root/uploader
diff options
context:
space:
mode:
authorFrederick Muriuki Muriithi2025-07-14 13:46:47 -0500
committerFrederick Muriuki Muriithi2025-07-14 13:46:47 -0500
commit6f9a91f1cb92530ad50c608c1f59ff38acb1b1f5 (patch)
treeb23e77448d70658d819ae2f864c40612796d41ac /uploader
parent58480d532a73997474692b5df4c186a60ef61208 (diff)
downloadgn-uploader-6f9a91f1cb92530ad50c608c1f59ff38acb1b1f5.tar.gz
Provide timeouts for requests calls.
Diffstat (limited to 'uploader')
-rw-r--r--uploader/oauth2/client.py3
-rw-r--r--uploader/publications/pubmed.py3
2 files changed, 4 insertions, 2 deletions
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)