From ca86b06f5a5ee306b9e87d33cb0a934f68652900 Mon Sep 17 00:00:00 2001 From: Quentin Defenouillere Date: Wed, 16 Sep 2020 17:14:08 +0200 Subject: [PATCH] [netfinca] New method check_action_needed This is mainly used during iter_accounts, but also in the cragr module before calling is_account_present. --- modules/netfinca/browser.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/modules/netfinca/browser.py b/modules/netfinca/browser.py index fea5952215..6c6d881b6a 100644 --- a/modules/netfinca/browser.py +++ b/modules/netfinca/browser.py @@ -38,12 +38,16 @@ def do_login(self): raise BrowserUnavailable() def iter_accounts(self): + self.accounts.stay_or_go() + self.check_action_needed() + return self.page.iter_accounts() + + def check_action_needed(self): self.accounts.stay_or_go() message = self.page.get_action_needed_message() if 'merci de renseigner les informations' in message: # Customers have to fill their e-mail address and phone number raise ActionNeeded(message) - return self.page.iter_accounts() def iter_investments(self, account): self.accounts.stay_or_go() -- GitLab