diff --git a/modules/caissedepargne/module.py b/modules/caissedepargne/module.py index ae28d0f4ae17f79259418672ee8c046c48f7b53e..0afc40569de217d17f55b659f9875f44c904865d 100644 --- a/modules/caissedepargne/module.py +++ b/modules/caissedepargne/module.py @@ -57,7 +57,9 @@ class CaisseEpargneModule(Module, CapBankWealth, CapBankTransferAddRecipient, Ca ValueTransient('request_information'), ) - accepted_document_types = (DocumentTypes.STATEMENT, DocumentTypes.OTHER,) + accepted_document_types = ( + DocumentTypes.STATEMENT, DocumentTypes.OTHER, DocumentTypes.NOTICE, + ) def create_default_browser(self): return self.create_browser( diff --git a/modules/caissedepargne/pages.py b/modules/caissedepargne/pages.py index af17911b912a40500c4c1b53a29855b86f7fb7ff..d453111e89105e97aa2a093efb16732d53172f13 100644 --- a/modules/caissedepargne/pages.py +++ b/modules/caissedepargne/pages.py @@ -2525,10 +2525,14 @@ class item(ItemElement): obj_label = Format('%s %s', CleanText('./td[3]'), CleanText('./td[2]')) obj_date = Date(CleanText('./td[2]'), dayfirst=True) - def obj_type(self): - if 'Relevé' in CleanText('./td[3]')(self): - return DocumentTypes.STATEMENT - return DocumentTypes.OTHER + _type_text = CleanText('./td[3]') + + TYPES_PATTERNS = { + "Relevé": DocumentTypes.STATEMENT, + "Info préalable à tarification": DocumentTypes.NOTICE, + "Information annuelle": DocumentTypes.NOTICE, + } + obj_type = MapIn(_type_text, TYPES_PATTERNS, default=DocumentTypes.OTHER) def download_document(self, document): form = self.get_form(id='main')