diff options
| author | Frederick Muriuki Muriithi | 2025-07-14 13:46:47 -0500 |
|---|---|---|
| committer | Frederick Muriuki Muriithi | 2025-07-14 13:46:47 -0500 |
| commit | 6f9a91f1cb92530ad50c608c1f59ff38acb1b1f5 (patch) | |
| tree | b23e77448d70658d819ae2f864c40612796d41ac | |
| parent | 58480d532a73997474692b5df4c186a60ef61208 (diff) | |
| download | gn-uploader-6f9a91f1cb92530ad50c608c1f59ff38acb1b1f5.tar.gz | |
Provide timeouts for requests calls.
| -rw-r--r-- | scripts/worker.py | 2 | ||||
| -rw-r--r-- | uploader/oauth2/client.py | 3 | ||||
| -rw-r--r-- | uploader/publications/pubmed.py | 3 |
3 files changed, 5 insertions, 3 deletions
diff --git a/scripts/worker.py b/scripts/worker.py index 91b0332..3165fe7 100644 --- a/scripts/worker.py +++ b/scripts/worker.py @@ -79,7 +79,7 @@ def main(): fqjobid = jobs.job_key(args.redisprefix, args.jobid) rconn.hset(fqjobid, "stderr", f"No such job. '{args.job_id}'.") rconn.expire(name=jobs.job_key(args.redisprefix, args.job_id), - time=timedelta(seconds=(2 * 60 * 60))) + time=timedelta(seconds=2 * 60 * 60)) print(f"No such job. '{args.job_id}'.", file=sys.stderr) return 2 return 3 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) |