From 6544bbfcdcf682dcc9e77c10d739985eb512f464 Mon Sep 17 00:00:00 2001 From: Laurent Bachelier Date: Sat, 3 Mar 2012 18:26:59 +0100 Subject: [PATCH] Fix account list: get all accounts There is also more logic in the xpath query, which I find more readable. --- modules/bnporc/pages/accounts_list.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/bnporc/pages/accounts_list.py b/modules/bnporc/pages/accounts_list.py index 35349e8c..d714bae6 100644 --- a/modules/bnporc/pages/accounts_list.py +++ b/modules/bnporc/pages/accounts_list.py @@ -34,8 +34,8 @@ class AccountsList(BasePage): def get_list(self): l = [] - for tr in self.document.getiterator('tr'): - if not 'class' in tr.attrib and tr.find('td') is not None and tr.find('td').attrib.get('class', '') == 'typeTitulaire': + for tr in self.document.xpath('//table[@class="tableCompte"]//tr[not(@class)]'): + if tr.find('td') is not None and tr.find('td').attrib.get('class', '') == 'typeTitulaire': account = Account() account.id = tr.xpath('.//td[@class="libelleCompte"]/input')[0].attrib['id'][len('libelleCompte'):] account.link_id = account.id