From 91f71fef6816cf3191255946c3df262a08f86268 Mon Sep 17 00:00:00 2001 From: Ludovico Magnocavallo Date: Mon, 21 Aug 2023 18:37:54 +0200 Subject: [PATCH] trap requests timeout error (#1607) --- blueprints/cloud-operations/quota-monitoring/src/main.py | 4 +++- .../cloud-operations/quota-monitoring/src/requirements.txt | 1 + 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/blueprints/cloud-operations/quota-monitoring/src/main.py b/blueprints/cloud-operations/quota-monitoring/src/main.py index ecdacbe2..5a845364 100755 --- a/blueprints/cloud-operations/quota-monitoring/src/main.py +++ b/blueprints/cloud-operations/quota-monitoring/src/main.py @@ -29,6 +29,7 @@ import warnings import click import google.auth +import requests.exceptions from google.auth.transport.requests import AuthorizedSession @@ -123,7 +124,8 @@ def fetch(request, delete=False): else: response = HTTP.post(request.url, headers=request.headers, data=json.dumps(request.data)) - except google.auth.exceptions.RefreshError as e: + except (google.auth.exceptions.RefreshError, + requests.exceptions.ReadTimeout) as e: raise SystemExit(e.args[0]) try: rdata = json.loads(response.content) diff --git a/blueprints/cloud-operations/quota-monitoring/src/requirements.txt b/blueprints/cloud-operations/quota-monitoring/src/requirements.txt index f488fac7..408307eb 100644 --- a/blueprints/cloud-operations/quota-monitoring/src/requirements.txt +++ b/blueprints/cloud-operations/quota-monitoring/src/requirements.txt @@ -2,3 +2,4 @@ click functions-framework google-api-core google-cloud-monitoring +requests \ No newline at end of file