From 19eee58709dbdc47b496935fb84cd34f57fafb51 Mon Sep 17 00:00:00 2001
From: Vincent Ardisson <va@budget-insight.com>
Date: Fri, 9 Dec 2016 17:09:11 +0100
Subject: [PATCH] [bforbank] implement investments/history for MARKET accounts
 with lcl

lcl module already implements the same website
---
 modules/bforbank/browser.py | 35 +++++++++++++++++++++++++++++++++--
 modules/bforbank/pages.py   | 12 +++++++++++-
 2 files changed, 44 insertions(+), 3 deletions(-)

diff --git a/modules/bforbank/browser.py b/modules/bforbank/browser.py
index e2ac4f2863..3d2f313f8a 100644
--- a/modules/bforbank/browser.py
+++ b/modules/bforbank/browser.py
@@ -24,7 +24,7 @@
 
 from .pages import (
     LoginPage, ErrorPage, AccountsPage, HistoryPage, LoanHistoryPage, RibPage,
-    LifeInsuranceIframe, LifeInsuranceRedir
+    LifeInsuranceIframe, LifeInsuranceRedir, TitrePage, BoursePage
 )
 from .spirica_browser import SpiricaBrowser
 
@@ -42,10 +42,16 @@ class BforbankBrowser(LoginBrowser):
     lifeinsurance_iframe = URL(r'/client/accounts/lifeInsurance/consultationDetailSpirica.action', LifeInsuranceIframe)
     lifeinsurance_redir = URL(r'https://assurance-vie.bforbank.com:443/sylvea/welcomeSSO.xhtml', LifeInsuranceRedir)
 
+    titre = URL(r'/client/accounts/stocks/consultation/noFramePartenaireCATitres.action\?id=0', TitrePage)
+    bourse = URL('https://bourse.bforbank.com/netfinca-titres/servlet/com.netfinca.frontcr.synthesis.HomeSynthesis',
+                 'https://bourse.bforbank.com/netfinca-titres/servlet/com.netfinca.frontcr.account.*',
+                 BoursePage)
+
     def __init__(self, weboob, birthdate, *args, **kwargs):
         super(BforbankBrowser, self).__init__(*args, **kwargs)
         self.birthdate = birthdate
         self.accounts = None
+        self.weboob = weboob
 
         self.spirica = SpiricaBrowser(weboob, 'https://assurance-vie.bforbank.com:443/', None, None)
 
@@ -71,7 +77,11 @@ def iter_accounts(self):
     @need_login
     def get_history(self, account):
         if account.type == Account.TYPE_MARKET:
-            raise NotImplementedError()
+            bourse_account = self.get_bourse_account(account)
+
+            self.location(bourse_account._link_id)
+            assert self.bourse.is_here()
+            return self.page.iter_history()
         elif account.type == Account.TYPE_LIFE_INSURANCE:
             self.goto_spirica(account)
             return self.spirica.iter_history(account)
@@ -94,9 +104,30 @@ def goto_spirica(self, account):
         self.spirica.session.cookies.update(self.session.cookies)
         self.spirica.logged = True
 
+    def get_bourse_account(self, account):
+        self.titre.go()
+        assert self.titre.is_here()
+        self.location(self.page.get_redir()) # "login" to bourse page
+
+        self.bourse.go()
+        assert self.bourse.is_here()
+
+        self.logger.debug('searching account matching %r', account)
+        for bourse_account in self.page.get_list():
+            self.logger.debug('iterating account %r', bourse_account)
+            if bourse_account.id.startswith(account.id[3:]):
+                return bourse_account
+
     @need_login
     def iter_investment(self, account):
         if account.type == Account.TYPE_LIFE_INSURANCE:
             self.goto_spirica(account)
             return self.spirica.iter_investment(account)
+        elif account.type == Account.TYPE_MARKET:
+            bourse_account = self.get_bourse_account(account)
+
+            self.location(bourse_account._market_link)
+            assert self.bourse.is_here()
+            return self.page.iter_investment()
+
         raise NotImplementedError()
diff --git a/modules/bforbank/pages.py b/modules/bforbank/pages.py
index c6b5bfa54f..9fef7739bd 100644
--- a/modules/bforbank/pages.py
+++ b/modules/bforbank/pages.py
@@ -21,7 +21,7 @@
 from StringIO import StringIO
 from PIL import Image
 
-from weboob.browser.pages import LoggedPage, HTMLPage, pagination
+from weboob.browser.pages import LoggedPage, HTMLPage, pagination, AbstractPage
 from weboob.browser.elements import method, ListElement, ItemElement
 from weboob.capabilities.bank import Account
 from weboob.browser.filters.html import Link, Attr
@@ -206,3 +206,13 @@ def get_redir(self):
             match = re.search(r'URL=([^\s"\']+)', meta)
             if match:
                 return match.group(1)
+
+
+class TitrePage(LoggedPage, HTMLPage):
+    def get_redir(self):
+        return Regexp(CleanText('//body/@onload'), r"document.location='(.*)'", r'\1')(self.doc)
+
+
+class BoursePage(AbstractPage):
+    PARENT = 'lcl'
+    PARENT_URL = 'bourse'
-- 
GitLab