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

fixes on console applications

Browse source
This commit is contained in:
Romain Bignon 2010-09-23 17:05:32 +02:00
commit 65bcd201e0
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
weboob/tools/application/console.py
View file

@ -48,6 +48,7 @@ class ConsoleApplication(BaseApplication):
SYNOPSIS += ' %prog [--help] [--version]' SYNOPSIS += ' %prog [--help] [--version]'
def __init__(self): def __init__(self):
self.enabled_backends = set()
option_parser = OptionParser(self.SYNOPSIS, version=self._get_optparse_version()) option_parser = OptionParser(self.SYNOPSIS, version=self._get_optparse_version())
app_options = OptionGroup(option_parser, '%s Options' % self.APPNAME.capitalize()) app_options = OptionGroup(option_parser, '%s Options' % self.APPNAME.capitalize())
self.add_application_options(app_options) self.add_application_options(app_options)
@ -277,6 +278,8 @@ class ConsoleApplication(BaseApplication):
command = staticmethod(command) command = staticmethod(command)
def load_backends(self, caps=None, names=None, *args, **kwargs): def load_backends(self, caps=None, names=None, *args, **kwargs):
if names is None:
names = self.enabled_backends
loaded_backends = BaseApplication.load_backends(self, caps, names, *args, **kwargs) loaded_backends = BaseApplication.load_backends(self, caps, names, *args, **kwargs)
if not loaded_backends: if not loaded_backends:
print 'Cannot start application: no configured backend was found.' print 'Cannot start application: no configured backend was found.'
@ -318,9 +321,12 @@ class ConsoleApplication(BaseApplication):
except BackendNotFound, e: except BackendNotFound, e:
logging.error(e) logging.error(e)
def set_requested_backends(self, requested_backends):
self.enabled_backends = requested_backends
def do(self, function, *args, **kwargs): def do(self, function, *args, **kwargs):
""" """
Call Weboob.do(), after having filled the yielded object, if selected fields are given by user. Call Weboob.do(), after having filled the yielded object, if selected fields are given by user.
""" """
return self.weboob.do(self.complete, self.options.count, self.selected_fields, function, *args, **kwargs) return self.weboob.do(self._complete, self.options.count, self.selected_fields, function, *args, **kwargs)