161 lines
5.9 KiB
Python
161 lines
5.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright(C) 2010-2012 Julien Veyssier
|
|
#
|
|
# 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 urlparse import urlparse, parse_qs
|
|
from decimal import Decimal
|
|
import re
|
|
|
|
from weboob.tools.browser import BasePage
|
|
from weboob.capabilities.bank import Account
|
|
from weboob.tools.capabilities.bank.transactions import FrenchTransaction
|
|
|
|
class LoginPage(BasePage):
|
|
def login(self, login, passwd):
|
|
self.browser.select_form(nr=0)
|
|
self.browser['_cm_user'] = login
|
|
self.browser['_cm_pwd'] = passwd
|
|
self.browser.submit(nologin=True)
|
|
|
|
class LoginErrorPage(BasePage):
|
|
pass
|
|
|
|
class InfoPage(BasePage):
|
|
pass
|
|
|
|
class TransfertPage(BasePage):
|
|
pass
|
|
|
|
class UserSpacePage(BasePage):
|
|
pass
|
|
|
|
class AccountsPage(BasePage):
|
|
def get_list(self):
|
|
ids = set()
|
|
|
|
for tr in self.document.getiterator('tr'):
|
|
first_td = tr.getchildren()[0]
|
|
if (first_td.attrib.get('class', '') == 'i g' or first_td.attrib.get('class', '') == 'p g') \
|
|
and first_td.find('a') is not None:
|
|
account = Account()
|
|
account.label = u"%s"%first_td.find('a').text.strip().lstrip(' 0123456789').title()
|
|
account._link_id = first_td.find('a').get('href', '')
|
|
if account._link_id.startswith('POR_SyntheseLst'):
|
|
continue
|
|
|
|
url = urlparse(account._link_id)
|
|
p = parse_qs(url.query)
|
|
if not 'rib' in p:
|
|
continue
|
|
|
|
account.id = p['rib'][0]
|
|
|
|
if account.id in ids:
|
|
continue
|
|
|
|
ids.add(account.id)
|
|
|
|
s = tr.getchildren()[2].text
|
|
if s.strip() == "":
|
|
s = tr.getchildren()[1].text
|
|
balance = u''
|
|
for c in s:
|
|
if c.isdigit() or c == '-':
|
|
balance += c
|
|
if c == ',':
|
|
balance += '.'
|
|
account.balance = Decimal(balance)
|
|
yield account
|
|
|
|
def next_page_url(self):
|
|
""" TODO pouvoir passer à la page des comptes suivante """
|
|
return 0
|
|
|
|
class Transaction(FrenchTransaction):
|
|
PATTERNS = [(re.compile('^VIR(EMENT)? (?P<text>.*)'), FrenchTransaction.TYPE_TRANSFER),
|
|
(re.compile('^PRLV (?P<text>.*)'), FrenchTransaction.TYPE_ORDER),
|
|
(re.compile('^(?P<text>.*) CARTE \d+ PAIEMENT CB (?P<dd>\d{2})(?P<mm>\d{2}) ?(.*)$'),
|
|
FrenchTransaction.TYPE_CARD),
|
|
(re.compile('^RETRAIT DAB (?P<dd>\d{2})(?P<mm>\d{2}) (?P<text>.*) CARTE \d+'),
|
|
FrenchTransaction.TYPE_WITHDRAWAL),
|
|
(re.compile('^CHEQUE$'), FrenchTransaction.TYPE_CHECK),
|
|
(re.compile('^COTIS\.? (?P<text>.*)'), FrenchTransaction.TYPE_BANK),
|
|
(re.compile('^REMISE (?P<text>.*)'), FrenchTransaction.TYPE_DEPOSIT),
|
|
]
|
|
|
|
|
|
class OperationsPage(BasePage):
|
|
def get_history(self):
|
|
index = 0
|
|
for tr in self.document.getiterator('tr'):
|
|
# columns can be:
|
|
# - date | value | operation | debit | credit | contre-valeur
|
|
# - date | value | operation | debit | credit
|
|
# - date | operation | debit | credit
|
|
# That's why we skip any extra columns, and take operation, debit
|
|
# and credit from last instead of first indexes.
|
|
tds = tr.getchildren()[:5]
|
|
if len(tds) < 4:
|
|
continue
|
|
|
|
if tds[0].attrib.get('class', '') == 'i g' or \
|
|
tds[0].attrib.get('class', '') == 'p g' or \
|
|
tds[0].attrib.get('class', '').endswith('_c1 c _c1'):
|
|
operation = Transaction(index)
|
|
index += 1
|
|
|
|
# Find different parts of label
|
|
parts = []
|
|
if len(tds[-3].findall('a')) > 0:
|
|
parts = [a.text.strip() for a in tds[-3].findall('a')]
|
|
else:
|
|
parts.append(tds[-3].text.strip())
|
|
if tds[-3].find('br') is not None:
|
|
parts.append(tds[-3].find('br').tail.strip())
|
|
|
|
# To simplify categorization of CB, reverse order of parts to separate
|
|
# location and institution.
|
|
if parts[0].startswith('PAIEMENT CB'):
|
|
parts.reverse()
|
|
|
|
operation.parse(date=tds[0].text,
|
|
raw=u' '.join(parts))
|
|
|
|
if tds[-1].text is not None and len(tds[-1].text) > 2:
|
|
s = tds[-1].text.strip()
|
|
elif tds[-1].text is not None and len(tds[-2].text) > 2:
|
|
s = tds[-2].text.strip()
|
|
else:
|
|
s = "0"
|
|
balance = u''
|
|
for c in s:
|
|
if c.isdigit() or c == "-":
|
|
balance += c
|
|
if c == ',':
|
|
balance += '.'
|
|
operation.amount = Decimal(balance)
|
|
yield operation
|
|
|
|
def next_page_url(self):
|
|
""" TODO pouvoir passer à la page des opérations suivantes """
|
|
return 0
|
|
|
|
class NoOperationsPage(OperationsPage):
|
|
def get_history(self):
|
|
return iter([])
|