From 95a24a2ad29548de25b9c24397553636c27af8ee Mon Sep 17 00:00:00 2001 From: Di Chen Date: Tue, 31 Aug 2021 11:09:36 +0800 Subject: [PATCH] Set custom timeout for requests [RHELDST-554] (UBIPOP_HTTP_TIMEOUT): New env variable added. (do_request): append "timeout" argument. --- ubipop/_pulp_client.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/ubipop/_pulp_client.py b/ubipop/_pulp_client.py index a635503..de815a4 100644 --- a/ubipop/_pulp_client.py +++ b/ubipop/_pulp_client.py @@ -17,6 +17,7 @@ HTTP_TOTAL_RETRIES = int(os.environ.get("UBIPOP_HTTP_TOTAL_RETRIES", 10)) HTTP_RETRY_BACKOFF = float(os.environ.get("UBIPOP_HTTP_RETRY_BACKOFF", 1)) +HTTP_TIMEOUT = int(os.environ.get("UBIPOP_HTTP_TIMEOUT", 60)) class UnsupportedTypeId(Exception): @@ -77,9 +78,13 @@ def do_request(self, req_type, url, data=None): req_url = urljoin(self.base_url, url) if req_type == "post": - ret = self.local.session.post(req_url, json=data, verify=not self.insecure) + ret = self.local.session.post( + req_url, json=data, verify=not self.insecure, timeout=HTTP_TIMEOUT + ) elif req_type == "get": - ret = self.local.session.get(req_url, verify=not self.insecure) + ret = self.local.session.get( + req_url, verify=not self.insecure, timeout=HTTP_TIMEOUT + ) else: ret = None