...

1 files changed, 13 insertions, 2 deletions

diff --git a/overlays/worktime/worktime.py b/overlays/worktime/worktime.py
index 166bea1c..46197b6e 100755
--- a/overlays/worktime/worktime.py
+++ b/overlays/worktime/worktime.py
@@ -30,6 +30,9 @@ from tabulate import tabulate
 from itertools import groupby
 from functools import cache
 
+import backoff
+
+
 class TogglAPISection(Enum):
     TOGGL = '/api/v8'
     REPORTS = '/reports/api/v2'
@@ -62,7 +65,17 @@ class TogglAPI(object):
         return uri
 
     def _query(self, url, method):
+        response = self._raw_query(url, method)
+        response.raise_for_status()
+        return response
 
+    @backoff.on_predicate(
+        backoff.expo,
+        factor=0.1, max_value=2,
+        predicate=lambda r: r.status_code == 429,
+        max_time=10,
+    )
+    def _raw_query(self, url, method):
         headers = {'content-type': 'application/json'}
         response = None
 
@@ -73,8 +86,6 @@ class TogglAPI(object):
         else:
             raise ValueError(f"Undefined HTTP method “{method}”")
 
-        response.raise_for_status()
-
         return response
 
     def get_billable_hours(self, start_date, end_date=datetime.now(timezone.utc), rounding=False):