ability to override the flush() method
This commit is contained in:
parent
7f03596605
commit
18c1f46922
1 changed files with 6 additions and 2 deletions
|
|
@ -654,7 +654,7 @@ class ListElement(AbstractElement):
|
|||
def __init__(self, *args, **kwargs):
|
||||
super(ListElement, self).__init__(*args, **kwargs)
|
||||
self.logger = getLogger(self.__class__.__name__.lower())
|
||||
self.objects = {}
|
||||
self.objects = OrderedDict()
|
||||
|
||||
def __call__(self, *args, **kwargs):
|
||||
for key, value in kwargs.iteritems():
|
||||
|
|
@ -676,11 +676,15 @@ class ListElement(AbstractElement):
|
|||
yield obj
|
||||
|
||||
if self.flush_at_end:
|
||||
for obj in self.objects.itervalues():
|
||||
for obj in self.flush():
|
||||
yield obj
|
||||
|
||||
self.check_next_page()
|
||||
|
||||
def flush(self):
|
||||
for obj in self.objects.itervalues():
|
||||
yield obj
|
||||
|
||||
def check_next_page(self):
|
||||
if not hasattr(self, 'next_page'):
|
||||
return
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue