From a069ae387ea09e9358400fa2f25df82477040fe2 Mon Sep 17 00:00:00 2001 From: Romain Bignon Date: Wed, 26 Nov 2014 14:27:01 +0100 Subject: [PATCH] detection of accounts type --- modules/fortuneo/pages/accounts_list.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/modules/fortuneo/pages/accounts_list.py b/modules/fortuneo/pages/accounts_list.py index f8db92cb..a5bc8520 100644 --- a/modules/fortuneo/pages/accounts_list.py +++ b/modules/fortuneo/pages/accounts_list.py @@ -122,6 +122,10 @@ class AccountsList(Page): form = self.document.xpath('//form[@name="InformationsPersonnellesForm"]') return len(form) > 0 + ACCOUNT_TYPES = {'mes-comptes/compte-courant': Account.TYPE_CHECKING, + 'mes-comptes/assurance-vie': Account.TYPE_MARKET, + 'mes-comptes/livret': Account.TYPE_LOAN, + } def get_list(self): for cpt in self.document.xpath(".//*[@class='synthese_id_compte']"): account = Account() @@ -154,8 +158,11 @@ class AccountsList(Page): account._link_id = url_to_parse # account.label - temp_label = cpt.xpath('./text()')[1].replace(u'-\xa0', '').replace("\n", "").replace("\t", "") - account.label = " ".join(temp_label.split(" ")[:2]) + account.label = cpt.xpath('./text()')[1].replace(u'-\xa0', '').replace("\n", "").replace("\t", "") + + for pattern, type in self.ACCOUNT_TYPES.iteritems(): + if pattern in account._link_id: + account.type = type yield account