4b56f56812
This is an OpenPGP/MIME signed message (RFC 4880 and 3156) Signed-off-by: Cédric Félizard <cedric@felizard.fr> Signed-off-by: Romain Bignon <romain@symlink.me>
91 lines
3.2 KiB
Python
91 lines
3.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright(C) 2015 Cédric Félizard
|
|
#
|
|
# 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/>.
|
|
|
|
|
|
import datetime
|
|
import re
|
|
from decimal import Decimal
|
|
from weboob.browser.pages import HTMLPage, LoggedPage
|
|
from weboob.capabilities.bank import Account
|
|
from weboob.tools.capabilities.bank.transactions import AmericanTransaction as EnglishTransaction
|
|
|
|
|
|
__all__ = ['LoginPage', 'AccountPage', 'HistoryPage']
|
|
|
|
|
|
class LoginPage(HTMLPage):
|
|
def login(self, username, password):
|
|
form = self.get_form(name='aspnetForm')
|
|
form['ctl00$chi$txtUserName'] = username
|
|
form['ctl00$chi$txtPassword'] = password
|
|
form.submit()
|
|
|
|
|
|
class AccountPage(LoggedPage, HTMLPage):
|
|
def get_accounts(self):
|
|
for el in self.doc.getroot().cssselect('div#content tr.row'):
|
|
account = Account()
|
|
|
|
balance = el.cssselect('td.Balance')[0].text
|
|
account.balance = Decimal(Transaction.clean_amount(balance))
|
|
account.id = el.cssselect('span')[0].text.strip()
|
|
account.currency = u'NZD' # TODO: handle other currencies
|
|
|
|
if el.cssselect('td.AccountName > a'):
|
|
label_el = el.cssselect('td.AccountName > a')[0]
|
|
account._link = label_el.get('href')
|
|
else:
|
|
label_el = el.cssselect('td.AccountName')[0]
|
|
account._link = None
|
|
|
|
account.label = unicode(label_el.text.strip())
|
|
|
|
yield account
|
|
|
|
|
|
class HistoryPage(LoggedPage, HTMLPage):
|
|
def get_history(self):
|
|
# TODO: get more results from "next" page, only 15 transactions per page
|
|
for el in self.doc.getroot().cssselect('div#content tr.row'):
|
|
transaction = Transaction()
|
|
|
|
label = unicode(el.cssselect('td.tranDesc')[0].text)
|
|
transaction.label = label
|
|
|
|
for pattern, _type in Transaction.PATTERNS:
|
|
match = pattern.match(label)
|
|
if match:
|
|
transaction.type = _type
|
|
break
|
|
|
|
date = el.cssselect('td.tranDate')[0].text
|
|
transaction.date = datetime.datetime.strptime(date, '%d %b \'%y')
|
|
|
|
amount = el.cssselect('td.tranAmnt')[0].text
|
|
transaction.amount = Decimal(Transaction.clean_amount(amount))
|
|
|
|
yield transaction
|
|
|
|
|
|
class Transaction(EnglishTransaction):
|
|
PATTERNS = [
|
|
(re.compile(r'^POS W/D (?P<text>.*)'), EnglishTransaction.TYPE_CARD),
|
|
(re.compile(r'^ATM W/D (?P<text>.*)'), EnglishTransaction.TYPE_WITHDRAWAL),
|
|
(re.compile(r'^(PAY|FROM) (?P<text>.*)'), EnglishTransaction.TYPE_TRANSFER),
|
|
]
|