From 226c4853a195940bcd69b24ea94ff1a2868ceedc Mon Sep 17 00:00:00 2001 From: Vincent Paredes Date: Mon, 1 Dec 2014 14:54:50 +0100 Subject: [PATCH] fixing statefull website with account's link changing --- modules/ganassurances/browser.py | 29 ++++++++++++++++------------- 1 file changed, 16 insertions(+), 13 deletions(-) diff --git a/modules/ganassurances/browser.py b/modules/ganassurances/browser.py index 45bd8f6a..614861f6 100644 --- a/modules/ganassurances/browser.py +++ b/modules/ganassurances/browser.py @@ -57,23 +57,26 @@ class GanAssurances(LoginBrowser): return self.page.get_list() def get_history(self, account): - if account._link is None: - return iter([]) + accounts = self.get_accounts_list() + for a in accounts: + if a.id == account.id: + self.location(a._link) + assert self.transactions.is_here() + return self.page.get_history() - self.location(account._link) + return iter([]) - assert self.transactions.is_here() - - return self.page.get_history() def get_coming(self, account): - if account._link is None: - return iter([]) + accounts = self.get_accounts_list() + for a in accounts: + if a.id == account.id: + self.location(a._link) + assert self.transactions.is_here() - self.location(account._link) - assert self.transactions.is_here() + self.location(self.page.get_coming_link()) + assert self.transactions.is_here() - self.location(self.page.get_coming_link()) - assert self.transactions.is_here() + return self.page.get_history() - return self.page.get_history() + return iter([])