163 lines
5.1 KiB
Python
163 lines
5.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright(C) 2012 Florent Fourcot
|
|
#
|
|
# 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 sys
|
|
|
|
from weboob.capabilities.bill import ICapBill, Detail, Subscription
|
|
from weboob.tools.application.repl import ReplApplication
|
|
from weboob.tools.application.formatters.iformatter import PrettyFormatter
|
|
|
|
|
|
__all__ = ['Boobill']
|
|
|
|
|
|
class SubscriptionsFormatter(PrettyFormatter):
|
|
MANDATORY_FIELDS = ('id', 'label')
|
|
|
|
def get_title(self, obj):
|
|
return obj.label
|
|
|
|
class Boobill(ReplApplication):
|
|
APPNAME = 'boobill'
|
|
VERSION = '0.c'
|
|
COPYRIGHT = 'Copyright(C) 2012 Florent Fourcot'
|
|
DESCRIPTION = 'Console application allowing to get and download bills.'
|
|
CAPS = ICapBill
|
|
COLLECTION_OBJECTS = (Subscription, )
|
|
EXTRA_FORMATTERS = {'subscriptions': SubscriptionsFormatter,
|
|
}
|
|
DEFAULT_FORMATTER = 'table'
|
|
COMMANDS_FORMATTERS = {'subscriptions': 'subscriptions',
|
|
'ls': 'subscriptions',
|
|
}
|
|
|
|
|
|
def main(self, argv):
|
|
self.load_config()
|
|
return ReplApplication.main(self, argv)
|
|
|
|
def do_subscriptions(self, line):
|
|
"""
|
|
subscriptions
|
|
|
|
List subscriptions
|
|
"""
|
|
self.start_format()
|
|
for backend, subscription in self.do('iter_subscription'):
|
|
self.cached_format(subscription)
|
|
self.flush()
|
|
|
|
def do_details(self, id):
|
|
"""
|
|
details Id
|
|
|
|
Get details of a subscription.
|
|
"""
|
|
|
|
id, backend_name = self.parse_id(id)
|
|
if not id:
|
|
print >>sys.stderr, 'Error: please give a subscription ID (hint: use subscriptions command)'
|
|
return 2
|
|
names = (backend_name,) if backend_name is not None else None
|
|
|
|
# XXX: should be generated by backend? -Flo
|
|
# XXX: no, but you should do it in a specific formatter -romain
|
|
mysum = Detail()
|
|
mysum.label = "Sum"
|
|
mysum.infos = "Generated by boobill"
|
|
mysum.price = 0.
|
|
|
|
self.start_format()
|
|
for backend, detail in self.do('get_details', id, backends=names):
|
|
self.format(detail)
|
|
mysum.price = detail.price + mysum.price
|
|
|
|
self.format(mysum)
|
|
self.flush()
|
|
|
|
def do_history(self, id):
|
|
"""
|
|
history Id
|
|
|
|
Get the history of a subscription.
|
|
"""
|
|
|
|
id, backend_name = self.parse_id(id)
|
|
if not id:
|
|
print >>sys.stderr, 'Error: please give a subscription ID (hint: use subscriptions command)'
|
|
return 2
|
|
names = (backend_name,) if backend_name is not None else None
|
|
|
|
self.start_format()
|
|
for backend, history in self.do('iter_history', id, backends=names):
|
|
self.format(history)
|
|
self.flush()
|
|
|
|
def do_bills(self, id):
|
|
"""
|
|
bills Id
|
|
|
|
Get the list of bills documents for subscription
|
|
id is the identifier of the backend
|
|
"""
|
|
|
|
id, backend_name = self.parse_id(id)
|
|
if not id:
|
|
print >>sys.stderr, 'Error: please give a subscription ID (hint: use subscriptions command)'
|
|
return 2
|
|
names = (backend_name,) if backend_name is not None else None
|
|
|
|
self.start_format()
|
|
for backend, date in self.do('iter_bills', id, backends=names):
|
|
self.format(date)
|
|
self.flush()
|
|
|
|
def do_download(self, line):
|
|
"""
|
|
download Id [FILENAME]
|
|
|
|
download the bill
|
|
id is the identifier of the bill (hint: try bills command)
|
|
FILENAME is where to write the file. If FILENAME is '-',
|
|
the file is written to stdout.
|
|
"""
|
|
id, dest = self.parse_command_args(line, 2, 1)
|
|
id, backend_name = self.parse_id(id)
|
|
if not id:
|
|
print >>sys.stderr, 'Error: please give a bill ID (hint: use bills command)'
|
|
return 2
|
|
names = (backend_name,) if backend_name is not None else None
|
|
|
|
if dest is None:
|
|
for backend, bill in self.do('get_bill', id, backends=names):
|
|
dest = id + "." + bill.format
|
|
|
|
for backend, buf in self.do('download_bill', id, backends=names):
|
|
if buf:
|
|
if dest == "-":
|
|
print buf
|
|
else:
|
|
try:
|
|
with open(dest, 'w') as f:
|
|
f.write(buf)
|
|
except IOError, e:
|
|
print >>sys.stderr, 'Unable to write bill in "%s": %s' % (dest, e)
|
|
return 1
|
|
return
|