From d978b1193b7e0704303bd7bf1fa7fa2bdc85d94d Mon Sep 17 00:00:00 2001
From: Nicolas Vergnac <nicolas.vergnac@budget-insight.com>
Date: Fri, 28 May 2021 12:26:30 +0200
Subject: [PATCH] [wiseed] handle blocked account at login

user keep getting blocked account message on login
---
 modules/wiseed/pages.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/modules/wiseed/pages.py b/modules/wiseed/pages.py
index 7ab359f1da..5e13608b4a 100644
--- a/modules/wiseed/pages.py
+++ b/modules/wiseed/pages.py
@@ -25,7 +25,7 @@
     CleanText, CleanDecimal, Regexp, Coalesce,
 )
 from woob.browser.elements import method, ItemElement, TableElement
-from woob.exceptions import BrowserIncorrectPassword
+from woob.exceptions import BrowserIncorrectPassword, BrowserUserBanned
 from woob.capabilities.base import NotAvailable
 from woob.capabilities.wealth import Investment
 from woob.tools.capabilities.bank.investments import create_french_liquidity
@@ -42,7 +42,10 @@ def raise_error(self):
         msg = CleanText('//div[has-class("alert-danger")]')(self.doc)
         if 'Email ou mot de passe invalide' in msg:
             raise BrowserIncorrectPassword(msg)
-        assert False, 'unhandled message %r' % msg
+        elif 'la connexion à votre compte est bloquée' in msg:
+            raise BrowserUserBanned(msg)
+
+        raise AssertionError('unhandled message : %s' % msg)
 
 
 class LandPage(LoggedPage, HTMLPage):
-- 
GitLab