change: add timeout to requests.(get|put)

tu-graz-library · Oct 2, 2022 · cee7b60 · cee7b60
1 parent ca13b80
commit cee7b60
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/invenio_alma/services/alma.py b/invenio_alma/services/alma.py
@@ -132,7 +132,7 @@ def get(self, url):
 
         :return str: response content
         """
-        response = requests.get(url, headers=self.headers)
+        response = requests.get(url, headers=self.headers, timeout=10)
         if response.status_code >= 400:
             raise AlmaAPIError(code=response.status_code, msg=response.text)
         return self.extract_alma_records(response.text)
@@ -155,7 +155,7 @@ def put(self, url, data):
 
         :return str: response content
         """
-        response = requests.put(url, data, headers=self.headers)
+        response = requests.put(url, data, headers=self.headers, timeout=10)
         if response.status_code >= 400:
             raise AlmaRESTError(code=response.status_code, msg=response.text)
         return response.text