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

handle error message when unable to login

Browse source
This commit is contained in:
Romain Bignon 2012-09-15 11:04:49 +02:00
commit 4101ea9c3d
2 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
modules/societegenerale/browser.py
View file

@ -18,7 +18,7 @@
# along with weboob. If not, see <http://www.gnu.org/licenses/>. # along with weboob. If not, see <http://www.gnu.org/licenses/>.
from weboob.tools.browser import BaseBrowser, BrowserIncorrectPassword from weboob.tools.browser import BaseBrowser, BrowserIncorrectPassword, BrowserUnavailable
from .pages.accounts_list import AccountsList, AccountHistory from .pages.accounts_list import AccountsList, AccountHistory
from .pages.login import LoginPage, BadLoginPage from .pages.login import LoginPage, BadLoginPage
@ -61,10 +61,16 @@ class SocieteGenerale(BaseBrowser):
self.page.login(self.username, self.password) self.page.login(self.username, self.password)
if self.is_on_page(LoginPage) or \ if self.is_on_page(LoginPage):
self.is_on_page(BadLoginPage):
raise BrowserIncorrectPassword() raise BrowserIncorrectPassword()
if self.is_on_page(BadLoginPage):
error = self.page.get_error()
if error.startswith('Votre session a'):
raise BrowserUnavailable('Session has expired')
else:
raise BrowserIncorrectPassword(error)
def get_accounts_list(self): def get_accounts_list(self):
if not self.is_on_page(AccountsList): if not self.is_on_page(AccountsList):
self.location('/restitution/cns_listeprestation.html') self.location('/restitution/cns_listeprestation.html')

3
modules/societegenerale/pages/login.py
View file

@ -85,4 +85,5 @@ class LoginPage(BasePage):
class BadLoginPage(BasePage): class BadLoginPage(BasePage):
pass def get_error(self):
return self.document.xpath('//span[@class="error_msg"]')[0].text.strip()