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

Move condition code in BaseApplication

Browse source 
The count parameter is now consistent with condition parameter
close #1289
This commit is contained in:
Florent 2013-12-15 16:47:42 +01:00
commit a3e6e98b4e
2 changed files with 10 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -34,13 +34,13 @@ from weboob.tools.config.iconfig import ConfigError
from weboob.tools.log import createColoredFormatter, getLogger from weboob.tools.log import createColoredFormatter, getLogger
from weboob.tools.misc import to_unicode from weboob.tools.misc import to_unicode
__all__ = ['BaseApplication'] __all__ = ['BaseApplication']
class MoreResultsAvailable(Exception): class MoreResultsAvailable(Exception):
pass pass
class ApplicationStorage(object): class ApplicationStorage(object):
def __init__(self, name, storage): def __init__(self, name, storage):
self.name = name self.name = name
@ -249,8 +249,12 @@ class BaseApplication(object):
return obj return obj
def _do_complete_iter(self, backend, count, fields, res): def _do_complete_iter(self, backend, count, fields, res):
modif = 0
for i, sub in enumerate(res): for i, sub in enumerate(res):
if count and i == count: if self.condition and not self.condition.is_valid(sub):
modif += 1
else:
if count and i - modif == count:
raise MoreResultsAvailable() raise MoreResultsAvailable()
sub = self._do_complete_obj(backend, fields, sub) sub = self._do_complete_obj(backend, fields, sub)
yield sub yield sub

5
weboob/tools/application/repl.py
View file

@ -480,11 +480,6 @@ class ReplApplication(Cmd, ConsoleApplication):
if self.options.condition: if self.options.condition:
self.condition = ResultsCondition(self.options.condition) self.condition = ResultsCondition(self.options.condition)
# Enable infinite search by default is condition is set
# (count applies on the non-filtered result, and can be confusing for users)
if self._is_default_count:
self.options.count = None
self._is_default_count = False
else: else:
self.condition = None self.condition = None