fix support of PEA accounts

This commit is contained in:
Baptiste Delpey 2015-02-23 18:49:51 +01:00 committed by Romain Bignon
commit 504240c29b
2 changed files with 42 additions and 2 deletions

View file

@ -25,7 +25,7 @@ from dateutil.relativedelta import relativedelta
from weboob.deprecated.browser import Browser, BrowserIncorrectPassword
from .pages.login import LoginPage
from .pages.accounts_list import GlobalAccountsList, AccountsList, AccountHistoryPage, InvestmentHistoryPage
from .pages.accounts_list import GlobalAccountsList, AccountsList, AccountHistoryPage, InvestmentHistoryPage, PeaHistoryPage
__all__ = ['Fortuneo']
@ -47,7 +47,7 @@ class Fortuneo(Browser):
'.*/prive/mes-comptes/compte-courant/consulter-situation/consulter-solde\.jsp.*' : AccountHistoryPage,
'.*/prive/mes-comptes/compte-titres-.*': InvestmentHistoryPage,
'.*/prive/mes-comptes/assurance-vie.*': InvestmentHistoryPage,
'.*/prive/mes-comptes/pea.*': InvestmentHistoryPage,
'.*/prive/mes-comptes/pea.*': PeaHistoryPage,
'.*/prive/mes-comptes/compte-especes.*': InvestmentHistoryPage,
}

View file

@ -49,6 +49,45 @@ class Transaction(FrenchTransaction):
(re.compile('^(?P<category>REMISE CHEQUES)(?P<text>.*)'), FrenchTransaction.TYPE_DEPOSIT),
]
class PeaHistoryPage(Page):
COL_LABEL = 0
COL_UNITVALUE = 1
COL_QUANTITY = 3
COL_UNITPRICE = 4
COL_VALUATION = 5
COL_PERF = 6
COL_WEIGHT = 7
def get_investments(self):
for line in self.document.xpath('//table[@id="t_intraday"]/tbody/tr'):
if line.find_class('categorie') or line.find_class('detail'):
continue
cols = line.findall('td')
inv = Investment()
inv.label = self.parser.tocleanstring(cols[self.COL_LABEL])
link = cols[self.COL_LABEL].xpath('./a[contains(@href, "cdReferentiel")]')[0]
inv.id = unicode(re.search('cdReferentiel=(.*)', link.attrib['href']).group(1))
inv.code = re.match('^[A-Z]+[0-9]+(.*)$', inv.id).group(1)
inv.quantity = self.parse_decimal(cols[self.COL_QUANTITY])
inv.unitprice = self.parse_decimal(cols[self.COL_UNITPRICE])
inv.unitvalue = self.parse_decimal(cols[self.COL_UNITVALUE])
inv.valuation = self.parse_decimal(cols[self.COL_VALUATION])
inv.diff = Decimal(Transaction.clean_amount(cols[self.COL_PERF].text.strip()))
yield inv
def parse_decimal(self, string):
value = self.parser.tocleanstring(string)
if value == '-':
return NotAvailable
return Decimal(Transaction.clean_amount(value))
def get_operations(self, _id):
return iter([])
class InvestmentHistoryPage(Page):
COL_LABEL = 0
COL_QUANTITY = 1
@ -84,6 +123,7 @@ class InvestmentHistoryPage(Page):
def get_operations(self, _id):
return iter([])
class AccountHistoryPage(Page):
def get_investments(self):
return iter([])