Add retry_after property to RateLimitException

glensc · Dec 12, 2022 · 16dbc90 · 16dbc90
1 parent 6f68e9b
commit 16dbc90
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/trakt/errors.py b/trakt/errors.py
@@ -102,6 +102,10 @@ class RateLimitException(TraktException):
     http_code = 429
     message = 'Rate Limit Exceeded'
 
+    @property
+    def retry_after(self):
+        return int(self.response.headers.get("Retry-After", 1))
+
 
 class TraktInternalException(TraktException):
     """TraktException type to be raised when a 500 error is raised"""