raise BrowserUnavailable on unavailable page

This commit is contained in:
Romain Bignon 2013-01-07 22:12:06 +01:00
commit ec05550073

View file

@ -22,7 +22,7 @@ from urlparse import urlsplit, parse_qsl
from decimal import Decimal from decimal import Decimal
import re import re
from weboob.tools.browser import BasePage from weboob.tools.browser import BasePage, BrowserUnavailable
from weboob.capabilities.bank import Account from weboob.capabilities.bank import Account
from weboob.tools.capabilities.bank.transactions import FrenchTransaction from weboob.tools.capabilities.bank.transactions import FrenchTransaction
@ -32,8 +32,12 @@ __all__ = ['LoginPage', 'IndexPage', 'AccountsPage', 'TransactionsPage', 'Unavai
class UnavailablePage(BasePage): class UnavailablePage(BasePage):
def on_loaded(self): def on_loaded(self):
a = self.document.xpath('//a[@class="btn"]')[0] try:
self.browser.location(a.attrib['href']) a = self.document.xpath('//a[@class="btn"]')[0]
except KeyError:
raise BrowserUnavailable()
else:
self.browser.location(a.attrib['href'])
class LoginPage(BasePage): class LoginPage(BasePage):
def login(self, login, passwd): def login(self, login, passwd):