nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[myhabit] Retry after request timeout. Closes #1741

Browse source
This commit is contained in:
Oleg Plakhotniuk 2015-02-26 03:54:18 -06:00 • committed by Romain Bignon
commit 0cffab661d
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
modules/myhabit/browser.py
View file

@ -18,6 +18,8 @@
# along with weboob. If not, see <http://www.gnu.org/licenses/>. # along with weboob. If not, see <http://www.gnu.org/licenses/>.
from requests.exceptions import Timeout
from weboob.tools.capabilities.bank.transactions import \ from weboob.tools.capabilities.bank.transactions import \
AmericanTransaction as AmTr AmericanTransaction as AmTr
from weboob.browser import LoginBrowser, URL, need_login from weboob.browser import LoginBrowser, URL, need_login
@ -124,6 +126,7 @@ class OrderPage(MyHabitPage):
class MyHabit(LoginBrowser): class MyHabit(LoginBrowser):
BASEURL = 'https://www.myhabit.com' BASEURL = 'https://www.myhabit.com'
MAX_RETRIES = 10
login = URL(r'/signin', r'https://www.amazon.com/ap/signin.*$', LoginPage) login = URL(r'/signin', r'https://www.amazon.com/ap/signin.*$', LoginPage)
order = URL(r'/vieworders\?.*appAction=ViewOrdersDetail.*', OrderPage) order = URL(r'/vieworders\?.*appAction=ViewOrdersDetail.*', OrderPage)
history = URL(r'/vieworders$', history = URL(r'/vieworders$',
@ -176,3 +179,11 @@ class MyHabit(LoginBrowser):
def do_login(self): def do_login(self):
if not self.login.go().login(self.username, self.password).logged: if not self.login.go().login(self.username, self.password).logged:
raise BrowserIncorrectPassword() raise BrowserIncorrectPassword()
def location(self, *args, **kwargs):
for i in xrange(self.MAX_RETRIES):
try:
return super(MyHabit, self).location(*args, **kwargs)
except Timeout as e:
pass
raise e