From 40877f358913ce7a666ae7b07d417672d5c1a911 Mon Sep 17 00:00:00 2001
From: Vincent A <dev@indigo.re>
Date: Mon, 8 Mar 2021 18:22:14 +0100
Subject: [PATCH] [caissedepargne] type some NOTICE documents

---
 modules/caissedepargne/module.py |  4 +++-
 modules/caissedepargne/pages.py  | 12 ++++++++----
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/modules/caissedepargne/module.py b/modules/caissedepargne/module.py
index ae28d0f4ae..0afc40569d 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 af17911b91..d453111e89 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')
-- 
GitLab