support deferred debit

This commit is contained in:
Romain Bignon 2013-01-15 15:39:56 +01:00
commit 07038a9989
5 changed files with 114 additions and 25 deletions

View file

@ -20,9 +20,10 @@
from .login import LoginPage, Initident, CheckPassword,repositionnerCheminCourant, BadLoginPage, AccountDesactivate, UnavailablePage
from .accountlist import AccountList
from .accounthistory import AccountHistory
from .accounthistory import AccountHistory, CardsList
from .transfer import TransferChooseAccounts, CompleteTransfer, TransferConfirm, TransferSummary
__all__ = ['LoginPage','Initident', 'CheckPassword', 'repositionnerCheminCourant', "AccountList", 'AccountHistory', 'BadLoginPage',
'AccountDesactivate', 'TransferChooseAccounts', 'CompleteTransfer', 'TransferConfirm', 'TransferSummary', 'UnavailablePage']
'AccountDesactivate', 'TransferChooseAccounts', 'CompleteTransfer', 'TransferConfirm', 'TransferSummary', 'UnavailablePage',
'CardsList']

View file

@ -18,13 +18,14 @@
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
import datetime
import re
from weboob.tools.capabilities.bank.transactions import FrenchTransaction
from weboob.tools.browser import BasePage
__all__ = ['AccountHistory']
__all__ = ['AccountHistory', 'CardsList']
class Transaction(FrenchTransaction):
@ -47,17 +48,39 @@ class Transaction(FrenchTransaction):
]
class AccountHistory(BasePage):
def get_history(self):
def get_next_link(self):
for a in self.document.xpath('//a[@class="btn_crt"]'):
txt = u''.join([txt.strip() for txt in a.itertext()])
if u'mois précédent' in txt:
return a.attrib['href']
def get_history(self, deferred=False):
"""
deffered is True when we are on a card page.
"""
mvt_table = self.document.xpath("//table[@id='mouvements']", smart_strings=False)[0]
mvt_ligne = mvt_table.xpath("./tbody/tr")
operations = []
if deferred:
# look for the debit date, and if it is already debited
txt = u''.join([txt.strip() for txt in self.document.xpath('//div[@class="infosynthese"]')[0].itertext()])
m = re.search('(\d+)/(\d+)/(\d+)', txt)
if m:
debit_date = datetime.date(*map(int, reversed(m.groups())))
coming = 'En cours' in txt
else:
coming = False
for mvt in mvt_ligne:
op = Transaction(len(operations))
op.parse(date=mvt.xpath("./td/span")[0].text.strip(),
raw=unicode(self.parser.tocleanstring(mvt.xpath('./td/span')[1]).strip()))
if op.label.startswith('DEBIT CARTE BANCAIRE DIFFERE'):
continue
r = re.compile(r'\d+')
tmp = mvt.xpath("./td/span/strong")
@ -70,5 +93,22 @@ class AccountHistory(BasePage):
op.set_amount(amount)
if deferred:
op.rdate = op.date
op.date = debit_date
# on card page, amounts are without sign
if op.amount > 0:
op.amount = - op.amount
op._coming = coming
operations.append(op)
return operations
class CardsList(BasePage):
def get_cards(self):
cards = []
for tr in self.document.xpath('//table[@class="dataNum"]/tbody/tr'):
cards.append(tr.xpath('.//a')[0].attrib['href'])
return cards

View file

@ -24,6 +24,7 @@ from weboob.capabilities.bank import Account, AccountNotFound
from weboob.tools.browser import BasePage
from weboob.tools.misc import to_unicode
from weboob.tools.capabilities.bank.transactions import FrenchTransaction
from weboob.tools.ordereddict import OrderedDict
__all__ = ['AccountList']
@ -31,7 +32,7 @@ __all__ = ['AccountList']
class AccountList(BasePage):
def on_loaded(self):
self.account_list = []
self.accounts = OrderedDict()
self.parse_table('comptes')
self.parse_table('comptesEpargne')
self.parse_table('comptesTitres')
@ -39,7 +40,7 @@ class AccountList(BasePage):
self.parse_table('comptesRetraireEuros')
def get_accounts_list(self):
return self.account_list
return self.accounts.itervalues()
def parse_table(self, what):
tables = self.document.xpath("//table[@id='%s']" % what, smart_strings=False)
@ -66,10 +67,19 @@ class AccountList(BasePage):
account.id = tmp_id
account.currency = account.get_currency(tmp_balance)
account.balance = Decimal(FrenchTransaction.clean_amount(tmp_balance))
self.account_list.append(account)
if account.id in self.accounts:
a = self.accounts[account.id]
a._card_links.append(account._link_id)
if not a.coming:
a.coming = Decimal('0.0')
a.coming += account.balance
else:
account._card_links = []
self.accounts[account.id] = account
def get_account(self, id):
for account in self.account_list:
if account.id == id:
return account
raise AccountNotFound('Unable to find account: %s' % id)
try:
return self.accounts[id]
except KeyError:
raise AccountNotFound('Unable to find account: %s' % id)