Move condition code in BaseApplication
The count parameter is now consistent with condition parameter close #1289
This commit is contained in:
parent
f7c4e4c89b
commit
a3e6e98b4e
2 changed files with 10 additions and 11 deletions
|
|
@ -34,13 +34,13 @@ from weboob.tools.config.iconfig import ConfigError
|
|||
from weboob.tools.log import createColoredFormatter, getLogger
|
||||
from weboob.tools.misc import to_unicode
|
||||
|
||||
|
||||
__all__ = ['BaseApplication']
|
||||
|
||||
|
||||
class MoreResultsAvailable(Exception):
|
||||
pass
|
||||
|
||||
|
||||
class ApplicationStorage(object):
|
||||
def __init__(self, name, storage):
|
||||
self.name = name
|
||||
|
|
@ -249,8 +249,12 @@ class BaseApplication(object):
|
|||
return obj
|
||||
|
||||
def _do_complete_iter(self, backend, count, fields, res):
|
||||
modif = 0
|
||||
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()
|
||||
sub = self._do_complete_obj(backend, fields, sub)
|
||||
yield sub
|
||||
|
|
|
|||
|
|
@ -480,11 +480,6 @@ class ReplApplication(Cmd, ConsoleApplication):
|
|||
|
||||
if 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:
|
||||
self.condition = None
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue