correctly use iterators

This commit is contained in:
Romain Bignon 2012-11-10 13:15:01 +01:00
commit 3be3689439
2 changed files with 4 additions and 8 deletions

View file

@ -84,8 +84,7 @@ class CragrBackend(BaseBackend, ICapBank):
self.config['password'].get())
def iter_accounts(self):
for account in self.browser.get_accounts_list():
yield account
return self.browser.get_accounts_list()
def get_account(self, _id):
if not _id.isdigit():
@ -97,9 +96,7 @@ class CragrBackend(BaseBackend, ICapBank):
raise AccountNotFound()
def iter_history(self, account):
for history in self.browser.get_history(account):
yield history
return self.browser.get_history(account)
def transfer(self, account, to, amount, reason=None):
return self.browser.do_transfer(account, to, amount, reason)

View file

@ -87,7 +87,6 @@ class AccountsList(CragrBasePage):
"""
Returns the list of available bank accounts
"""
l = []
for div in self.document.getiterator('div'):
if div.attrib.get('class', '') in ('dv', 'headline') and div.getchildren()[0].tag in ('a', 'br'):
self.logger.debug("Analyzing div %s" % div)
@ -128,8 +127,8 @@ class AccountsList(CragrBasePage):
self.logger.debug(' the history link appears to be %s' % account._link_id)
else:
account._link_id = None
l.append(account)
return l
yield account
def get_history(self, start_index=0, start_offset=0):
"""