summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorErik Larsson <who+github@cnackers.org>2012-07-08 22:06:08 +0200
committerErik Larsson <who+github@cnackers.org>2012-07-08 23:27:13 +0200
commit5af2f62c049825a5f606b7041fa7e6f918d8486c (patch)
tree1cbbb30251db7889fe7db9d469db26e2a8e2e33a
parent4a4a03858e87df15b0722e1c8951d5b8592a346a (diff)
downloadslixmpp-5af2f62c049825a5f606b7041fa7e6f918d8486c.tar.gz
slixmpp-5af2f62c049825a5f606b7041fa7e6f918d8486c.tar.bz2
slixmpp-5af2f62c049825a5f606b7041fa7e6f918d8486c.tar.xz
slixmpp-5af2f62c049825a5f606b7041fa7e6f918d8486c.zip
Make sure that the last RSM stanza is returned from the iterator
-rw-r--r--sleekxmpp/plugins/xep_0059/rsm.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/sleekxmpp/plugins/xep_0059/rsm.py b/sleekxmpp/plugins/xep_0059/rsm.py
index 9335ed22..d4218202 100644
--- a/sleekxmpp/plugins/xep_0059/rsm.py
+++ b/sleekxmpp/plugins/xep_0059/rsm.py
@@ -47,6 +47,7 @@ class ResultIterator():
self.start = start
self.interface = interface
self.reverse = reverse
+ self._stop = False
def __iter__(self):
return self
@@ -62,6 +63,8 @@ class ResultIterator():
results will be the items before the current page
of items.
"""
+ if self._stop:
+ raise StopIteration
self.query[self.interface]['rsm']['before'] = self.reverse
self.query['id'] = self.query.stream.new_id()
self.query[self.interface]['rsm']['max'] = str(self.amount)
@@ -84,7 +87,7 @@ class ResultIterator():
first = int(r[self.interface]['rsm']['first_index'])
num_items = len(r[self.interface]['substanzas'])
if first + num_items == count:
- raise StopIteration
+ self._stop = True
if self.reverse:
self.start = r[self.interface]['rsm']['first']