use iterator instead of get_backends
This commit is contained in:
parent
bfdc3e014f
commit
a4dd17631f
2 changed files with 7 additions and 11 deletions
|
|
@ -55,15 +55,12 @@ class Weboob:
|
|||
module = self.modules_loader[modname]
|
||||
self.backends[instname] = module.create_backend(self)
|
||||
|
||||
def get_backends(self, caps=None):
|
||||
def iter_backends(self, caps=None):
|
||||
if caps is None:
|
||||
return self.backends
|
||||
|
||||
d = {}
|
||||
for name, backend in self.backends.iteritems():
|
||||
if backend.has_caps(caps):
|
||||
d[name] = backend
|
||||
return d
|
||||
return self.backends.iteritems()
|
||||
else:
|
||||
return dict((name, backend) for name, backend in self.backends.iteritems()
|
||||
if backend.has_caps(caps)).iteritems()
|
||||
|
||||
def schedule(self, interval, function, *args):
|
||||
self.scheduler.enter(interval, 1, function, args)
|
||||
|
|
|
|||
|
|
@ -57,9 +57,8 @@ class Application(BaseApplication):
|
|||
self.weboob.loop()
|
||||
|
||||
def process(self):
|
||||
backends = self.weboob.get_backends()
|
||||
for name, b in backends.iteritems():
|
||||
for message in b.iter_new_messages():
|
||||
for name, backend in self.weboob.iter_backends():
|
||||
for message in backend.iter_new_messages():
|
||||
self.send_email(name, message)
|
||||
|
||||
def send_email(self, backend_name, mail):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue