Set timeout to 10 seconds for all external requests

delta10 · Nov 1, 2022 · 3a130a1 · 3a130a1
1 parent 8bd90ab
commit 3a130a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/lib.py b/lib.py
@@ -46,7 +46,7 @@ def send(self, endpoint, additional_headers):
             'SOAPAction': self.SOAP_ACTION
         }
 
-        response = requests.post(endpoint, self.tostring(), headers={ **headers, **additional_headers })
+        response = requests.post(endpoint, self.tostring(), headers={ **headers, **additional_headers }, timeout=10)
         response.raise_for_status()
 
         return etree.fromstring(response.content)

diff --git a/server.py b/server.py
@@ -116,7 +116,7 @@ def buitenbeter():
     }
 
     url = urljoin(app.config['SIGNALEN_ENDPOINT'], '/signals/category/prediction')
-    response = requests.post(url, data=json.dumps(data), headers={ **headers, **app.config['ADDITIONAL_SIGNALEN_HEADERS'] })
+    response = requests.post(url, data=json.dumps(data), headers={ **headers, **app.config['ADDITIONAL_SIGNALEN_HEADERS'] }, timeout=10)
 
     sub_category = urljoin(app.config['SIGNALEN_ENDPOINT'], '/signals/v1/public/terms/categories/overig/sub_categories/overig')
 
@@ -153,7 +153,7 @@ def buitenbeter():
     }
 
     url = urljoin(app.config['SIGNALEN_ENDPOINT'], '/signals/v1/private/signals/')
-    response = requests.post(url, data=json.dumps(data), headers={ **headers, **app.config['ADDITIONAL_SIGNALEN_HEADERS'] })
+    response = requests.post(url, data=json.dumps(data), headers={ **headers, **app.config['ADDITIONAL_SIGNALEN_HEADERS'] }, timeout=10)
 
     signal_data = response.json()
     signal_id = signal_data.get('signal_id')
@@ -182,7 +182,7 @@ def buitenbeter():
 
 
         url = urljoin(app.config['SIGNALEN_ENDPOINT'], f'/signals/v1/public/signals/{signal_id}/attachments/')
-        response = requests.post(url, data=data, files=files, headers=app.config['ADDITIONAL_SIGNALEN_HEADERS'])
+        response = requests.post(url, data=data, files=files, headers=app.config['ADDITIONAL_SIGNALEN_HEADERS'], timeout=10)
         if not response.ok:
             app.logger.info(f'Could not create attachment in Signalen: {response.text}')
             return 'Could not create attachment in Signalen', 400