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

add parameter END_DATE to 'history' and 'coming' commands (closes #1150)

Browse source 
It iterates on all transactions until the specified date
This commit is contained in:
Romain Bignon 2013-07-27 18:43:23 +02:00
commit 97ade8e4ec
1 changed files with 39 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

60
weboob/applications/boobank/boobank.py
View file

@ -18,6 +18,9 @@
# along with weboob. If not, see <http://www.gnu.org/licenses/>. # along with weboob. If not, see <http://www.gnu.org/licenses/>.
from datetime import date
from dateutil.relativedelta import relativedelta
from dateutil.parser import parse as parse_date
from decimal import Decimal, InvalidOperation from decimal import Decimal, InvalidOperation
import sys import sys
@ -235,26 +238,47 @@ class Boobank(ReplApplication):
""" """
return self.do_ls(line) return self.do_ls(line)
def show_history(self, command, line):
id, end_date = self.parse_command_args(line, 2, 1)
account = self.get_object(id, 'get_account', [])
if not account:
print >>sys.stderr, 'Error: please give an account ID (hint: use list command)'
return 2
if end_date is not None:
try:
end_date = parse_date(end_date)
except ValueError:
print >>sys.stderr, '"%s" is an incorrect date format (for example %s)' % (end_date, (date.today() - relativedelta(months=1)).strftime('%Y-%m-%d'))
return 3
old_count = self.options.count
self.options.count = None
self.start_format()
for backend, transaction in self.do(command, account, backends=account.backend):
if end_date is not None and transaction.date < end_date:
break
self.format(transaction)
if end_date is not None:
self.options.count = old_count
def complete_history(self, text, line, *ignored): def complete_history(self, text, line, *ignored):
args = line.split(' ') args = line.split(' ')
if len(args) == 2: if len(args) == 2:
return self._complete_account() return self._complete_account()
@defaultcount(10) @defaultcount(10)
def do_history(self, id): def do_history(self, line):
""" """
history ID history ID [END_DATE]
Display history of transactions. Display history of transactions.
"""
account = self.get_object(id, 'get_account', [])
if not account:
print >>sys.stderr, 'Error: please give an account ID (hint: use list command)'
return 2
self.start_format() If END_DATE is supplied, list all transactions until this date.
for backend, transaction in self.do('iter_history', account, backends=account.backend): """
self.format(transaction) return self.show_history('iter_history', line)
def complete_coming(self, text, line, *ignored): def complete_coming(self, text, line, *ignored):
args = line.split(' ') args = line.split(' ')
@ -262,21 +286,15 @@ class Boobank(ReplApplication):
return self._complete_account() return self._complete_account()
@defaultcount(10) @defaultcount(10)
def do_coming(self, id): def do_coming(self, line):
""" """
coming ID coming ID [END_DATE]
Display future transactions. Display future transactions.
"""
account = self.get_object(id, 'get_account', [])
if not account:
print >>sys.stderr, 'Error: please give an account ID (hint: use list command)'
return 2
self.start_format(id=id) If END_DATE is supplied, show all transactions list this date.
for backend, transaction in self.do('iter_coming', account, backends=account.backend): """
self.format(transaction) return self.show_history('iter_coming', line)
self.flush()
def complete_transfer(self, text, line, *ignored): def complete_transfer(self, text, line, *ignored):
args = line.split(' ') args = line.split(' ')