weboob-devel/modules/alloresto/pages.py
2014-03-22 16:06:51 +01:00

76 lines
2.7 KiB
Python

# -*- coding: utf-8 -*-
# Copyright(C) 2014 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/>.
import datetime
from decimal import Decimal
from weboob.tools.browser2.page import HTMLPage, LoggedPage, method, ItemElement
from weboob.tools.browser2.filters import CleanDecimal, CleanText, Filter, TableCell
from weboob.capabilities.bank import Account
from weboob.tools.capabilities.bank.transactions import FrenchTransaction as Transaction
__all__ = ['LoginPage', 'AccountsPage']
class LoginPage(HTMLPage):
def login(self, username, password):
form = self.get_form(nr=1)
form['uname'] = username
form['pass'] = password
form.submit()
class AccountsPage(LoggedPage, HTMLPage):
@method
class iter_accounts(ItemElement):
def __call__(self):
return self
klass = Account
obj_id = '0'
obj_label = u'Compte miams'
obj_balance = CleanDecimal('//div[@class="compteur"]//strong')
obj_currency = u'MIAM'
obj_coming = CleanDecimal('//table[@id="solde_acquisition_lignes"]//th[@class="col_montant"]', default=Decimal('0'))
class MyDate(Filter):
MONTHS = ['janv', u'févr', u'mars', u'avr', u'mai', u'juin', u'juil', u'août', u'sept', u'oct', u'nov', u'déc']
def filter(self, txt):
day, month, year = txt.split(' ')
day = int(day)
year = int(year)
month = self.MONTHS.index(month.rstrip('.')) + 1
return datetime.date(year, month, day)
def get_transactions(self, type='consommable'):
class get_history(Transaction.TransactionsElement):
head_xpath = '//table[@id="solde_%s_lignes"]//thead//tr/th/text()' % type
item_xpath = '//table[@id="solde_%s_lignes"]//tbody/tr' % type
col_date = u"Date de valeur"
col_raw = u"Motif"
class item(Transaction.TransactionElement):
obj_amount = Transaction.Amount('./td[last()]')
obj_date = AccountsPage.MyDate(CleanText(TableCell('date')))
return get_history(self)()