From 06797c956980fbc2650e24ad79618b67ecd94edb Mon Sep 17 00:00:00 2001 From: Olivier Da Rocha Date: Wed, 13 Nov 2019 17:17:55 +0100 Subject: [PATCH] [browser/exceptions] BrowserTooManyRequests: Add message parameter This will prevent giving a message to BrowserTooManyRequests instead of a next_try by mistake. --- weboob/browser/exceptions.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/weboob/browser/exceptions.py b/weboob/browser/exceptions.py index ea9bcb2230..a97fbaf091 100644 --- a/weboob/browser/exceptions.py +++ b/weboob/browser/exceptions.py @@ -54,7 +54,9 @@ class BrowserTooManyRequests(BrowserUnavailable): # in exactly 24h NEXT_24H = datetime.datetime.now() + relativedelta(days=1) - def __init__(self, next_try=None): + def __init__(self, message='', next_try=None): + super(BrowserTooManyRequests, self).__init__(message) + if next_try is None: next_try = BrowserTooManyRequests.NEXT_24H @@ -64,4 +66,4 @@ def __init__(self, next_try=None): self.next_try = next_try def __str__(self): - return 'Too many requests, next_try set %s' % self.next_try + return super(BrowserTooManyRequests, self).__str__() or 'Too many requests, next_try set %s' % self.next_try -- GitLab