Skip to content
pages.py 13.3 KiB
Newer Older
Romain Bignon's avatar
Romain Bignon committed
# -*- coding: utf-8 -*-

# Copyright(C) 2015      Romain Bignon
#
# This file is part of a weboob module.
Romain Bignon's avatar
Romain Bignon committed
#
# This weboob module is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
Romain Bignon's avatar
Romain Bignon committed
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This weboob module is distributed in the hope that it will be useful,
Romain Bignon's avatar
Romain Bignon committed
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
Romain Bignon's avatar
Romain Bignon committed
#
# You should have received a copy of the GNU Lesser General Public License
# along with this weboob module. If not, see <http://www.gnu.org/licenses/>.
from __future__ import unicode_literals

Romain Bignon's avatar
Romain Bignon committed
import re

from weboob.exceptions import BrowserIncorrectPassword
from weboob.browser.pages import LoggedPage, HTMLPage, pagination, PDFPage
from weboob.browser.elements import method, ItemElement, TableElement
Romain Bignon's avatar
Romain Bignon committed
from weboob.capabilities.bank import Account
from weboob.capabilities.base import NotAvailable
from weboob.capabilities.profile import Profile
from weboob.browser.filters.standard import CleanText, CleanDecimal, Async, Regexp, Join, Field
from weboob.browser.filters.html import Attr, Link, TableCell, ColumnNotFound
Romain Bignon's avatar
Romain Bignon committed
from weboob.tools.capabilities.bank.transactions import FrenchTransaction
from weboob.tools.capabilities.bank.iban import is_iban_valid
from weboob.tools.compat import basestring
from weboob.tools.pdf import extract_text
class MyCleanText(CleanText):
    @classmethod
    def clean(cls, txt, children=True, newlines=True, normalize='NFC'):
        if not isinstance(txt, basestring):
            txt = u'\n'.join([t.strip() for t in txt.itertext()])

        return txt


Romain Bignon's avatar
Romain Bignon committed
class LoginPage(HTMLPage):
    def login(self, username, password):
        form = self.get_form()
        form['identifiant'] = username
        form['motpasse'] = password
        form.submit()


class LoginConfirmPage(HTMLPage):
    def on_load(self):
        error = CleanText('//td[has-class("ColonneLibelle")]')(self.doc)
        if len(error) > 0:
            raise BrowserIncorrectPassword(error)

Romain Bignon's avatar
Romain Bignon committed
class AccountsPage(LoggedPage, HTMLPage):
    def get_acc_link(self):
        msg = CleanText('//body[@class="message"]')(self.doc)
        if msg:
            acc_link = Link('//div[@class="Boutons"]/a', 'href')(self.doc)
            return acc_link

Romain Bignon's avatar
Romain Bignon committed
    @method
    class iter_accounts(TableElement):
Romain Bignon's avatar
Romain Bignon committed
        item_xpath = '//table[has-class("TableBicolore")]//tr[@id and count(td) > 4]'
        head_xpath = '//table[has-class("TableBicolore")]//tr/td[@id]/@id'

        col_id = 'idCompteLibelle'
        col_label = 'idCompteIntitule'
        col_balance = 'idCompteSolde'
        col_currency = 'idCompteSoldeUM'
        col_rib = 'idCompteRIB'
        col_type = 'idCompteNature'
Romain Bignon's avatar
Romain Bignon committed

        class item(ItemElement):
            klass = Account

            def condition(self):
                return CleanDecimal(TableCell('balance'), replace_dots=True, default=NotAvailable)(self) is not NotAvailable

            TYPE = {
                'COMPTE COURANT': Account.TYPE_CHECKING,
                'COMPTE TRANSACTION': Account.TYPE_CHECKING,
                'COMPTE ORDINAIRE': Account.TYPE_CHECKING,
            }
            TYPE_BY_LABELS = {
                'CAV': Account.TYPE_CHECKING,
            }

            obj_id = CleanText(TableCell('id'))
            obj_label = CleanText(TableCell('label'))
            obj_currency = FrenchTransaction.Currency(TableCell('currency'))
            obj_balance = CleanDecimal(TableCell('balance'), replace_dots=True)

            def obj__link(self):
                return Attr(TableCell('id')(self)[0].xpath('./a'), 'href')(self)

            def obj__url(self):
                return Link(TableCell('rib')(self)[0].xpath('./a[img[starts-with(@alt, "RIB")]]'), default=None)(self)

            def load_iban(self):
                link = Link(TableCell('rib')(self)[0].xpath('./a[img[starts-with(@alt, "RIB")]]'), default=None)(self)
                return self.page.browser.async_open(link)

            def obj_type(self):
                try:
                    el_to_check = CleanText(TableCell('type'))(self)
                    type_dict = self.TYPE
                except ColumnNotFound:
                    el_to_check = Field('label')(self)
                    type_dict = self.TYPE_BY_LABELS

                for k, v in type_dict.items():
                    if el_to_check.startswith(k):
                        return v
                return Account.TYPE_UNKNOWN

            def obj_iban(self):
                rib_page = Async('iban').loaded_page(self)
                if 'RibPdf' in rib_page.url:
                    return rib_page.get_iban()
                return Join('', Regexp(CleanText('//td[has-class("ColonneCode")][contains(text(), "IBAN")]'), r'\b((?!IBAN)[A-Z0-9]+)\b', nth='*'))(rib_page.doc) or NotAvailable


class RibPage(LoggedPage, HTMLPage):
    def get_profile(self):
        profile = Profile()

        # profile is inside a <td> separated with a simple <br> without <span> or <div>
        profile_txt = MyCleanText('//div[@class="TableauAffichage"]/table/tr[3]/td[1]')(self.doc).split('\n')
        i_name = 0
        profile.name = u''
        # name can be on one, two, (more ?) lines, so we stop when line start by a number, we suppose it's the address number
        while not re.search('^\d', profile_txt[i_name]):
            profile.name += ' ' + profile_txt[i_name]
            i_name += 1

        profile.name = profile.name.strip()
        profile.address = u''
        # address is not always on two lines, so we consider every lines from here to before last are address, (last one is country)
        for i in range(i_name, len(profile_txt)-1):
            profile.address += ' ' + profile_txt[i]

        profile.address = profile.address.strip()
        profile.country = profile_txt[-1]

        profile.name = profile.name.replace('MONSIEUR ', '').replace('MADAME ', '')

        return profile
class RibPDFPage(LoggedPage, PDFPage):
    def get_iban(self):
        text = extract_text(self.doc)
        iban = re.search(r'IBAN([A-Z]{2}\d+)', text).group(1)
        assert is_iban_valid(iban), 'did not parse IBAN properly'
        return iban


Romain Bignon's avatar
Romain Bignon committed
class Transaction(FrenchTransaction):
    PATTERNS = [(re.compile(r'^VIR(EMENT)?( SEPA)? (?P<text>.*)'),                      FrenchTransaction.TYPE_TRANSFER),
                (re.compile(r'^PRLV (?P<text>.*)'),                                     FrenchTransaction.TYPE_ORDER),
                (re.compile(r'^(?P<text>.*) CARTE \d+ PAIEMENT CB\s+(?P<dd>\d{2})(?P<mm>\d{2}) ?(.*)$'),
                                                                                       FrenchTransaction.TYPE_CARD),
                (re.compile(r'^RETRAIT DAB (?P<dd>\d{2})(?P<mm>\d{2}) (?P<text>.*) CARTE [\*\d]+'),
                                                                                       FrenchTransaction.TYPE_WITHDRAWAL),
                (re.compile(r'^CHEQUE( (?P<text>.*))?$'),                               FrenchTransaction.TYPE_CHECK),
                (re.compile(r'^(F )?COTIS\.? (?P<text>.*)'),                            FrenchTransaction.TYPE_BANK),
                (re.compile(r'^(REMISE|REM.CHQ) (?P<text>.*)'),                         FrenchTransaction.TYPE_DEPOSIT),
                (re.compile(r'^(?P<text>.*) CARTE BLEUE'),                              FrenchTransaction.TYPE_CARD),
                (re.compile(r'^PRVL SEPA (?P<text>.*)'),                                FrenchTransaction.TYPE_ORDER),
                (re.compile(r'^(?P<text>(INT. DEBITEURS).*)'),                          FrenchTransaction.TYPE_BANK),
                (re.compile(r'^(?P<text>.*(VIR EMIS).*)'),                              FrenchTransaction.TYPE_TRANSFER),
                (re.compile(r'^(?P<text>.*(\bMOUVEMENT\b).*)'),                         FrenchTransaction.TYPE_TRANSFER),
                (re.compile(r'^(?P<text>.*(ARRETE TRIM.).*)'),                          FrenchTransaction.TYPE_BANK),
                (re.compile(r'^(?P<text>.*(TENUE DE DOSSIE).*)'),                       FrenchTransaction.TYPE_BANK),
                (re.compile(r'^(?P<text>.*(RELEVE LCR ECH).*)'),                        FrenchTransaction.TYPE_ORDER),
                (re.compile(r'^(?P<text>.*(\+ FORT DECOUVERT).*)'),                     FrenchTransaction.TYPE_BANK),
                (re.compile(r'^(?P<text>.*(EXTRANET @THEMI).*)'),                       FrenchTransaction.TYPE_BANK),
                (re.compile(r'^(?P<text>.*(REL CPT DEBITEU).*)'),                       FrenchTransaction.TYPE_ORDER),
                (re.compile(r"^(?P<text>.*(\bAFFRANCHISSEMENT\b).*)"),                  FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(REMISE VIREMENTS MAGNE).*)"),                FrenchTransaction.TYPE_TRANSFER),
                (re.compile(r"^(?P<text>.*(\bEFFET\b).*)"),                             FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(\bMANIP\.\b).*)"),                           FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(INTERETS SUR REMISE PTF).*)"),               FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(REMISE ESCOMPTE PTF).*)"),                   FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(RETENUE DE GARANTIE).*)"),                   FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(RESTITUTION RETENUE GARANTIE).*)"),          FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(\bAMENDES\b).*)"),                           FrenchTransaction.TYPE_TRANSFER),
                (re.compile(r"^(?P<text>.*(\bOA\b).*)"),                                FrenchTransaction.TYPE_BANK),
                (re.compile(r"^.* COTIS ANN (?P<text>.*)"),                             FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(FORFAIT CENT\.RE).*)"),                      FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(ENVOI CB).*)"),                              FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(RET\.SDD).*)"),                              FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(RETOUR PVL ACD EXPERTISE).*)"),              FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(Annulation PAR REJ\/CHQ).*)"),               FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(REJET CHEQUE).*)"),                          FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(CHQ PAYE INFRAC).*)"),                       FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>^(CHQ IRREGULIER).*)"),                         FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(ERREUR REMISE C).*)"),                       FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>^(\bREMCHQ\b).*)"),                             FrenchTransaction.TYPE_DEPOSIT),
                (re.compile(r"^(?P<text>^(RETOUR PVL).*)"),                             FrenchTransaction.TYPE_TRANSFER),
                (re.compile(r"^(?P<text>.*(\bTRANSFERT\b).*)"),                         FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(\bCONFIRMATION\b).*)"),                      FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(CAUTION AVEC GAGE).*)"),                     FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(\bRAPATRIEMENT\b).*)"),                      FrenchTransaction.TYPE_BANK),
                (re.compile(r"^(?P<text>.*(CHANGE REF).*)"),                            FrenchTransaction.TYPE_BANK),
                (re.compile(r'^CARTE DU'),                                              FrenchTransaction.TYPE_CARD),
                (re.compile(r'^(VIR (SEPA)?|Vir|VIR.)(?P<text>.*)'),                    FrenchTransaction.TYPE_TRANSFER),
                (re.compile(r'^VIREMENT DE (?P<text>.*)'),                              FrenchTransaction.TYPE_TRANSFER),
                (re.compile(r'^(CHQ|CHEQUE) (?P<text>.*)'),                             FrenchTransaction.TYPE_CHECK),
                (re.compile(r'^(PRLV SEPA|PRELEVEMENT) (?P<text>.*)'),                  FrenchTransaction.TYPE_ORDER),
Romain Bignon's avatar
Romain Bignon committed
               ]


class HistoryPage(LoggedPage, HTMLPage):
    @pagination
    @method
    class get_operations(Transaction.TransactionsElement):
        def next_page(self):
            for script in self.page.doc.xpath('//script'):
                m = re.search(r"getCodePagination\('(\d+)','(\d+)','([^']+)'.*", script.text or '', re.MULTILINE)
                if m:
                    cur_page = int(m.group(1))
                    nb_pages = int(m.group(2))
                    baseurl = m.group(3)

                    if cur_page < nb_pages:
                        return baseurl + '&numeroPage=%s&nbrPage=%s' % (cur_page + 1, nb_pages)

        head_xpath = '//div[has-class("TableauBicolore")]/table/tr[not(@id)]/td'
        item_xpath = '//div[has-class("TableauBicolore")]/table/tr[@id and count(td) > 3]'
        col_date = ['Date comptable', "Date d'opération"]
Romain Bignon's avatar
Romain Bignon committed
        col_vdate = ['Date de valeur']
        col_raw = ["Libellé de l'opération"]
Romain Bignon's avatar
Romain Bignon committed

        class item(Transaction.TransactionElement):
            pass