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

ability to override the flush() method

Browse source
This commit is contained in:
Romain Bignon 2014-07-01 20:37:58 +02:00
commit 18c1f46922
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
weboob/tools/browser2/page.py
View file

@ -654,7 +654,7 @@ class ListElement(AbstractElement):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super(ListElement, self).__init__(*args, **kwargs) super(ListElement, self).__init__(*args, **kwargs)
self.logger = getLogger(self.__class__.__name__.lower()) self.logger = getLogger(self.__class__.__name__.lower())
self.objects = {} self.objects = OrderedDict()
def __call__(self, *args, **kwargs): def __call__(self, *args, **kwargs):
for key, value in kwargs.iteritems(): for key, value in kwargs.iteritems():
@ -676,11 +676,15 @@ class ListElement(AbstractElement):
yield obj yield obj
if self.flush_at_end: if self.flush_at_end:
for obj in self.objects.itervalues(): for obj in self.flush():
yield obj yield obj
self.check_next_page() self.check_next_page()
def flush(self):
for obj in self.objects.itervalues():
yield obj
def check_next_page(self): def check_next_page(self):
if not hasattr(self, 'next_page'): if not hasattr(self, 'next_page'):
return return