new module ganassurances (ICapBank)

2012-12-21 12:29:31 +01:00 · 2012-12-21 12:29:31 +01:00 · d5fabd9c75
commit d5fabd9c75
parent 14b6ac9f46
6 changed files with 330 additions and 0 deletions
--- a/modules/ganassurances/pages.py
+++ b/modules/ganassurances/pages.py
@ -0,0 +1,116 @@
+# -*- coding: utf-8 -*-
+
+# Copyright(C) 2012 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 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
+
+
+__all__ = ['LoginPage', 'AccountsPage', 'TransactionsPage']
+
+
+class LoginPage(BasePage):
+    def login(self, login, passwd):
+        self.browser.select_form(name='loginForm')
+        self.browser.set_all_readonly(False)
+        self.browser['LoginPortletFormID'] = login.encode(self.browser.ENCODING)
+        self.browser['LoginPortletFormPassword1'] = passwd.encode(self.browser.ENCODING)
+        self.browser.submit(nologin=True)
+
+class AccountsPage(BasePage):
+    ACCOUNT_TYPES = {u'Solde des comptes bancaires - Groupama Banque':  Account.TYPE_CHECKING,
+                     u'Epargne bancaire constituée - Groupama Banque':  Account.TYPE_SAVINGS,
+                    }
+
+    def get_list(self):
+        account_type = Account.TYPE_UNKNOWN
+        accounts = []
+
+        for tr in self.document.xpath('//table[@class="ecli"]/tr'):
+            if tr.attrib.get('class', '') == 'entete':
+                account_type = self.ACCOUNT_TYPES.get(tr.find('th').text.strip(), Account.TYPE_UNKNOWN)
+                continue
+
+            tds = tr.findall('td')
+
+            balance = tds[-1].text.strip()
+            if balance == '':
+                continue
+
+            account = Account()
+            account.label = u' '.join([txt.strip() for txt in tds[0].itertext()])
+            account.label = re.sub(u'[ \xa0\u2022\r\n\t]+', u' ', account.label).strip()
+            account.id = re.findall('(\d+)', account.label)[0]
+            account.balance = Decimal(FrenchTransaction.clean_amount(balance))
+            account.currency = account.get_currency(balance)
+            account.type = account_type
+            m = re.search(r"javascript:submitForm\(([\w_]+),'([^']+)'\);", tds[0].find('a').attrib['onclick'])
+            if not m:
+                self.logger.warning('Unable to find link for %r' % account.label)
+                account._link = None
+            else:
+                account._link = m.group(2)
+
+            accounts.append(account)
+
+        return accounts
+
+
+class Transaction(FrenchTransaction):
+    PATTERNS = [(re.compile('^Facture (?P<dd>\d{2})/(?P<mm>\d{2})-(?P<text>.*) carte .*'),
+                                                                FrenchTransaction.TYPE_CARD),
+                (re.compile(u'^(Prlv( de)?|Ech(éance|\.)) (?P<text>.*)'),
+                                                                FrenchTransaction.TYPE_ORDER),
+                (re.compile('^(Vir|VIR)( de)? (?P<text>.*)'),
+                                                                FrenchTransaction.TYPE_TRANSFER),
+                (re.compile(u'^CHEQUE.*? (N° \w+)?$'),          FrenchTransaction.TYPE_CHECK),
+                (re.compile('^Cotis(ation)? (?P<text>.*)'),
+                                                                FrenchTransaction.TYPE_BANK),
+                (re.compile('(?P<text>Int .*)'),                FrenchTransaction.TYPE_BANK),
+               ]
+
+class TransactionsPage(BasePage):
+    def get_history(self):
+        count = 0
+        for tr in self.document.xpath('//table[@id="releve_operation"]/tr'):
+            tds = tr.findall('td')
+
+            if len(tds) < 4:
+                continue
+
+            t = Transaction(count)
+
+            date = u''.join([txt.strip() for txt in tds[0].itertext()])
+            raw = u' '.join([txt.strip() for txt in tds[1].itertext()])
+            debit = u''.join([txt.strip() for txt in tds[-2].itertext()])
+            credit = u''.join([txt.strip() for txt in tds[-1].itertext()])
+            t.parse(date, re.sub(r'[ ]+', ' ', raw))
+            t.set_amount(credit, debit)
+
+            yield t
+
+            count += 1
+
+    def get_coming_link(self):
+        a = self.document.getroot().cssselect('div#sous_nav ul li a.bt_sans_off')[0]
+        return re.sub('[ \t\r\n]+', '', a.attrib['href'])