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