From 108df5b7997c08baa37b4fa61dff32af912b181a Mon Sep 17 00:00:00 2001
From: "Ronald Portier (Therp BV)" <ronald@therp.nl>
Date: Fri, 20 Jan 2023 14:42:30 +0100
Subject: [PATCH] [IMP] *_online_ponto: add timeout to request get and post

---
 .../models/ponto_interface.py                       | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/account_statement_import_online_ponto/models/ponto_interface.py b/account_statement_import_online_ponto/models/ponto_interface.py
index 7be235ee..f116d6af 100644
--- a/account_statement_import_online_ponto/models/ponto_interface.py
+++ b/account_statement_import_online_ponto/models/ponto_interface.py
@@ -40,6 +40,7 @@ class PontoInterface(models.AbstractModel):
             url,
             params={"grant_type": "client_credentials"},
             headers=login_headers,
+            timeout=60,
         )
         data = self._get_response_data(response)
         access_token = data.get("access_token", False)
@@ -70,7 +71,10 @@ class PontoInterface(models.AbstractModel):
         url = PONTO_ENDPOINT + "/accounts"
         _logger.debug(_("GET request on %s"), url)
         response = requests.get(
-            url, params={"limit": 100}, headers=self._get_request_headers(access_data)
+            url,
+            params={"limit": 100},
+            headers=self._get_request_headers(access_data),
+            timeout=60,
         )
         data = self._get_response_data(response)
         for ponto_account in data.get("data", []):
@@ -128,7 +132,12 @@ class PontoInterface(models.AbstractModel):
         _logger.debug(
             _("GET request to %s with headers %s and params %s"), url, params, headers
         )
-        response = requests.get(url, params=params, headers=headers)
+        response = requests.get(
+            url,
+            params=params,
+            headers=headers,
+            timeout=(60, 300),
+        )
         return self._get_response_data(response)
 
     def _get_response_data(self, response):