From 1c71ab7a7684d6ab7a80cfd017156c2cbfec5fa5 Mon Sep 17 00:00:00 2001 From: Romain Bignon Date: Sat, 12 Oct 2013 12:25:34 +0200 Subject: [PATCH] do not crash when there is no full list page --- modules/banquepopulaire/browser.py | 9 +++++---- modules/banquepopulaire/pages.py | 3 +++ 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/modules/banquepopulaire/browser.py b/modules/banquepopulaire/browser.py index 1c212245..ae41acf9 100644 --- a/modules/banquepopulaire/browser.py +++ b/modules/banquepopulaire/browser.py @@ -88,10 +88,11 @@ class BanquePopulaire(BaseBrowser): if self.page.is_error(): raise BrokenPageError('Unable to go on the accounts list page') - self.select_form(nr=0) - self.set_all_readonly(False) - self['dialogActionPerformed'] = 'EQUIPEMENT_COMPLET' - self.submit() + if self.page.is_short_list(): + self.select_form(nr=0) + self.set_all_readonly(False) + self['dialogActionPerformed'] = 'EQUIPEMENT_COMPLET' + self.submit() self.token = self.page.get_token() diff --git a/modules/banquepopulaire/pages.py b/modules/banquepopulaire/pages.py index 35b680ba..f7fd31ee 100644 --- a/modules/banquepopulaire/pages.py +++ b/modules/banquepopulaire/pages.py @@ -210,6 +210,9 @@ class AccountsPage(BasePage): return False + def is_short_list(self): + return len(self.document.xpath('//script[contains(text(), "EQUIPEMENT_COMPLET")]')) > 0 + def get_list(self): account_type = Account.TYPE_UNKNOWN