nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

add a 'type' field on Transaction object

Browse source
This commit is contained in:
Romain Bignon 2012-02-26 15:05:04 +01:00
commit b8d05430ec
1 changed files with 31 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

38
weboob/capabilities/bank.py
View file

@ -1,6 +1,6 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# Copyright(C) 2010-2011 Romain Bignon # Copyright(C) 2010-2012 Romain Bignon
# #
# This file is part of weboob. # This file is part of weboob.
# #
@ -24,8 +24,7 @@ from .base import CapBaseObject
from .collection import ICapCollection, CollectionNotFound from .collection import ICapCollection, CollectionNotFound
__all__ = ['Account', 'AccountNotFound', 'TransferError', 'ICapBank', __all__ = ['Account', 'AccountNotFound', 'TransferError', 'ICapBank', 'Transaction']
'Operation']
class AccountNotFound(Exception): class AccountNotFound(Exception):
@ -53,16 +52,29 @@ class Account(Recipient):
return u"<Account id=%r label=%r>" % (self.id, self.label) return u"<Account id=%r label=%r>" % (self.id, self.label)
class Operation(CapBaseObject): class Transaction(CapBaseObject):
TYPE_UNKNOWN = 0
TYPE_TRANSFER = 1
TYPE_ORDER = 2
TYPE_CHECK = 3
TYPE_DEPOSIT = 4
TYPE_PAYBACK = 5
TYPE_WITHDRAWAL = 6
TYPE_CARD = 7
TYPE_LOAN_PAYMENT = 8
TYPE_BANK = 9
def __init__(self, id): def __init__(self, id):
CapBaseObject.__init__(self, id) CapBaseObject.__init__(self, id)
self.add_field('date', (basestring, datetime, date)) self.add_field('date', (basestring, datetime, date))
self.add_field('type', int, self.TYPE_UNKNOWN)
self.add_field('text', unicode)
self.add_field('category', unicode) self.add_field('category', unicode)
self.add_field('label', unicode) self.add_field('label', unicode)
self.add_field('amount', float) self.add_field('amount', float)
def __repr__(self): def __repr__(self):
return "<Operation date='%s' label='%s' amount=%s>" % (self.date, return "<Transaction date='%s' label='%s' amount=%s>" % (self.date,
self.label, self.amount) self.label, self.amount)
class Transfer(CapBaseObject): class Transfer(CapBaseObject):
@ -87,10 +99,22 @@ class ICapBank(ICapCollection):
def get_account(self, _id): def get_account(self, _id):
raise NotImplementedError() raise NotImplementedError()
def iter_operations(self, account): def iter_history(self, account):
"""
Iter history of transactions on a specific account.
@param account [Account]
@return [iter(Transaction)]
"""
raise NotImplementedError() raise NotImplementedError()
def iter_history(self, account): def iter_coming(self, account):
"""
Iter coming transactions on a specific account.
@param account [Account]
@return [iter(Transaction)]
"""
raise NotImplementedError() raise NotImplementedError()
def iter_transfer_recipients(self, account): def iter_transfer_recipients(self, account):