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

WebNip.iter_backends takes a new optional parameter 'module'

Browse source
This commit is contained in:
Romain Bignon 2014-10-08 16:24:49 +02:00
commit 1a8a0db712
2 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
contrib/boobot.py
View file

@ -173,7 +173,7 @@ class MyThread(Thread):
def check_twitter(self): def check_twitter(self):
nb_tweets = 10 nb_tweets = 10
for _, backend in self.weboob.load_backends(modules=['twitter']).iteritems(): for backend in self.weboob.iter_backends(module='twitter'):
for thread in list(itertools.islice(backend.iter_resources(None, ['search', 'weboob']), for thread in list(itertools.islice(backend.iter_resources(None, ['search', 'weboob']),
0, 0,
nb_tweets)): nb_tweets)):

7
weboob/core/ouiboube.py
View file

@ -184,7 +184,7 @@ class WebNip(object):
""" """
return len(self.backend_instances) return len(self.backend_instances)
def iter_backends(self, caps=None): def iter_backends(self, caps=None, module=None):
""" """
Iter on each backends. Iter on each backends.
@ -192,10 +192,13 @@ class WebNip(object):
:param caps: optional list of capabilities to select backends :param caps: optional list of capabilities to select backends
:type caps: tuple[:class:`weboob.capabilities.base.Capability`] :type caps: tuple[:class:`weboob.capabilities.base.Capability`]
:param module: optional name of module
:type module: :class:`basestring`
:rtype: iter[:class:`weboob.tools.backend.Module`] :rtype: iter[:class:`weboob.tools.backend.Module`]
""" """
for _, backend in sorted(self.backend_instances.iteritems()): for _, backend in sorted(self.backend_instances.iteritems()):
if caps is None or backend.has_caps(caps): if caps is None or backend.has_caps(caps) and \
module is None or backend.NAME == module:
with backend: with backend:
yield backend yield backend