76cb004eb4
One of the goal is to not import all modules needed by filters by loading the page file. In the same goal, move the import of parsers in the class definition.
174 lines
6.5 KiB
Python
174 lines
6.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright(C) 2009-2014 Florent Fourcot, Romain Bignon
|
|
#
|
|
# This file is part of weboob.
|
|
#
|
|
# weboob is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# weboob is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
from datetime import date, timedelta
|
|
import datetime
|
|
import re
|
|
|
|
from weboob.capabilities.bank import Account
|
|
from weboob.capabilities.base import NotAvailable
|
|
from weboob.tools.browser2.page import HTMLPage, LoggedPage, method
|
|
from weboob.tools.browser2.elements import ListElement, ItemElement
|
|
from weboob.tools.browser2.filters import Attr, CleanText, CleanDecimal, Filter, Field, MultiFilter, Date, Lower
|
|
from weboob.tools.capabilities.bank.transactions import FrenchTransaction
|
|
|
|
|
|
__all__ = ['AccountsList']
|
|
|
|
|
|
class Transaction(FrenchTransaction):
|
|
PATTERNS = [(re.compile(u'^retrait dab (?P<dd>\d{2})/(?P<mm>\d{2})/(?P<yy>\d{4}) (?P<text>.*)'), FrenchTransaction.TYPE_WITHDRAWAL),
|
|
(re.compile(u'^carte (?P<dd>\d{2})/(?P<mm>\d{2})/(?P<yy>\d{4}) (?P<text>.*)'), FrenchTransaction.TYPE_CARD),
|
|
(re.compile(u'^virement (sepa )?(emis vers|recu|emis)? (?P<text>.*)'), FrenchTransaction.TYPE_TRANSFER),
|
|
(re.compile(u'^cheque (?P<text>.*)'), FrenchTransaction.TYPE_CHECK),
|
|
(re.compile(u'^prelevement (?P<text>.*)'), FrenchTransaction.TYPE_ORDER),
|
|
(re.compile(u'^prélèvement sepa en faveur de (?P<text>.*)'), FrenchTransaction.TYPE_ORDER),
|
|
]
|
|
|
|
|
|
class AddPref(MultiFilter):
|
|
prefixes = {u'Courant': u'CC-', u'Livret A': 'LA-', u'Orange': 'LEO-',
|
|
u'Durable': u'LDD-', u"Titres": 'TITRE-', u'PEA': u'PEA-'}
|
|
|
|
def filter(self, values):
|
|
el, label = values
|
|
for key, pref in self.prefixes.items():
|
|
if key in label:
|
|
return pref + el
|
|
return el
|
|
|
|
|
|
class AddType(Filter):
|
|
types = {u'Courant': Account.TYPE_CHECKING, u'Livret A': Account.TYPE_SAVINGS,
|
|
u'Orange': Account.TYPE_SAVINGS, u'Durable': Account.TYPE_SAVINGS,
|
|
u'Titres': Account.TYPE_MARKET, u'PEA': Account.TYPE_MARKET}
|
|
|
|
def filter(self, label):
|
|
for key, acc_type in self.types.items():
|
|
if key in label:
|
|
return acc_type
|
|
return Account.TYPE_UNKNOWN
|
|
|
|
|
|
class PreHashmd5(MultiFilter):
|
|
def filter(self, values):
|
|
concat = ''
|
|
for value in values:
|
|
if type(value) is datetime.date:
|
|
concat += value.strftime('%d/%m/%Y')
|
|
else:
|
|
concat += u'%s' % value
|
|
return concat.encode('utf-8')
|
|
|
|
|
|
class INGDate(Date):
|
|
monthvalue = {u'janv.': '01', u'févr.': '02', u'mars': '03', u'avr.': '04',
|
|
u'mai': '05', u'juin': '06', u'juil.': '07', u'août': '08',
|
|
u'sept.': '09', u'oct.': '10', u'nov.': '11', u'déc.': '12'}
|
|
|
|
def filter(self, txt):
|
|
if txt == 'hier':
|
|
return (date.today() - timedelta(days=1))
|
|
elif txt == "aujourd'hui":
|
|
return date.today()
|
|
else:
|
|
frenchmonth = txt.split(' ')[1]
|
|
month = self.monthvalue[frenchmonth]
|
|
txt = txt.replace(' ', '')
|
|
txt = txt.replace(frenchmonth, '/%s/' % month)
|
|
return super(INGDate, self).filter(txt)
|
|
|
|
|
|
class INGCategory(Filter):
|
|
catvalue = {u'virt': u"Virement", u'autre': u"Autre",
|
|
u'plvt': u'Prélèvement', u'cb_ret': u"Carte retrait",
|
|
u'cb_ach': u'Carte achat', u'chq': u'Chèque',
|
|
u'frais': u'Frais bancaire', u'sepaplvt': u'Prélèvement'}
|
|
|
|
def filter(self, txt):
|
|
txt = txt.split('-')[0].lower()
|
|
try:
|
|
return self.catvalue[txt]
|
|
except:
|
|
return txt
|
|
|
|
|
|
class AccountsList(LoggedPage, HTMLPage):
|
|
def __init__(self, browser, response, *args, **kwargs):
|
|
super(AccountsList, self).__init__(browser, response, *args, **kwargs)
|
|
self.i = 0
|
|
|
|
@method
|
|
class get_list(ListElement):
|
|
item_xpath = '//a[@class="mainclic"]'
|
|
|
|
class item(ItemElement):
|
|
klass = Account
|
|
|
|
obj_currency = u'EUR'
|
|
obj__id = CleanText('span[@class="account-number"]')
|
|
obj_label = CleanText('span[@class="title"]')
|
|
obj_id = AddPref(Field('_id'), Field('label'))
|
|
obj_type = AddType(Field('label'))
|
|
obj_balance = CleanDecimal('span[@class="solde"]/label')
|
|
obj_coming = NotAvailable
|
|
obj__jid = Attr('//input[@name="javax.faces.ViewState"]', 'value')
|
|
|
|
@method
|
|
class get_transactions(ListElement):
|
|
item_xpath = '//table'
|
|
|
|
class item(ItemElement):
|
|
klass = Transaction
|
|
|
|
obj_id = None # will be overwrited by the browser
|
|
# we use lower for compatibility with the old website
|
|
obj_raw = Transaction.Raw(Lower('.//td[@class="lbl"]'))
|
|
obj_amount = CleanDecimal('.//td[starts-with(@class, "amount")]')
|
|
obj_date = INGDate(CleanText('.//td[@class="date"]'), dayfirst=True)
|
|
obj_rdate = Field('date')
|
|
obj__hash = PreHashmd5(Field('date'), Field('raw'), Field('amount'))
|
|
obj_category = INGCategory(Attr('.//td[@class="picto"]/span', 'class'))
|
|
|
|
def condition(self):
|
|
if self.el.find('.//td[@class="date"]') is None:
|
|
return False
|
|
if self.page.i < self.env['index']:
|
|
self.page.i += 1
|
|
return False
|
|
return True
|
|
|
|
def get_history_jid(self):
|
|
span = self.doc.xpath('//span[@id="index:panelASV"]')
|
|
if len(span) > 1:
|
|
# Assurance Vie, we do not support this kind of account.
|
|
return None
|
|
|
|
span = Attr('//span[starts-with(@id, "index:j_id")]', 'id')(self.doc)
|
|
jid = span.split(':')[1]
|
|
return jid
|
|
|
|
def islast(self):
|
|
havemore = self.doc.getroot().cssselect('.show-more-transactions')
|
|
if len(havemore) == 0:
|
|
return True
|
|
|
|
nomore = self.doc.getroot().cssselect('.no-more-transactions')
|
|
return (len(nomore) > 0)
|