diff --git a/modules/banquepopulaire/pages.py b/modules/banquepopulaire/pages.py index d5888069f864991316f9dcf31bb1c18fe1813366..b0846109ed63fc67f10ad7fa4bd4dbab05a170cf 100644 --- a/modules/banquepopulaire/pages.py +++ b/modules/banquepopulaire/pages.py @@ -28,7 +28,10 @@ from PIL import Image, ImageFilter from weboob.browser.elements import method, DictElement, ItemElement -from weboob.browser.filters.standard import CleanText, CleanDecimal, Regexp, Eval, Date, Field +from weboob.browser.filters.standard import ( + CleanText, CleanDecimal, Regexp, Eval, + Date, Field, MapIn, +) from weboob.browser.filters.html import Attr, Link, AttributeNotFound from weboob.browser.filters.json import Dict from weboob.exceptions import BrowserUnavailable, BrowserIncorrectPassword, ActionNeeded @@ -347,13 +350,12 @@ def get_access_token(self): def get_user_type(self): user_subscription = Dict('characteristics/subscribeTypeItems/0/label')(self.doc) user_types = { - 'abonnement Particulier': 'part', - 'abonnement Personne Morale': 'ent', - 'abonnement EI (pro)': 'pro', + 'Particulier': 'part', + 'Personne Protégé': 'part', + 'Personne Morale': 'ent', + 'EI': 'pro', } - user_type = user_types.get(user_subscription) - assert user_type, "%s user type is not yet handle" % user_subscription - return user_type + return MapIn(self.doc, user_types).filter(user_subscription) class VkImagePage(AbstractPage):