diff options
| author | root <madhurgarg96@gmail.com> | 2019-08-29 11:25:26 +0530 |
|---|---|---|
| committer | Maxime “pep” Buquet <pep@bouah.net> | 2019-09-08 14:22:48 +0200 |
| commit | 11f707987dd3c1b80918ce4db81d31cd98a07ad1 (patch) | |
| tree | 5c6abd5247327e51f4e8b548f61ffb8ba05efd82 | |
| parent | db13794e0f0218af3ae76e852a704a6e86244b29 (diff) | |
| download | slixmpp-11f707987dd3c1b80918ce4db81d31cd98a07ad1.tar.gz slixmpp-11f707987dd3c1b80918ce4db81d31cd98a07ad1.tar.bz2 slixmpp-11f707987dd3c1b80918ce4db81d31cd98a07ad1.tar.xz slixmpp-11f707987dd3c1b80918ce4db81d31cd98a07ad1.zip | |
Added amount parameter, so that limit on the msgs received per query can be changed.
| -rw-r--r-- | slixmpp/plugins/xep_0059/rsm.py | 4 | ||||
| -rw-r--r-- | slixmpp/plugins/xep_0313/mam.py | 7 |
2 files changed, 7 insertions, 4 deletions
diff --git a/slixmpp/plugins/xep_0059/rsm.py b/slixmpp/plugins/xep_0059/rsm.py index 2bc68138..a7aaa7ce 100644 --- a/slixmpp/plugins/xep_0059/rsm.py +++ b/slixmpp/plugins/xep_0059/rsm.py @@ -142,7 +142,7 @@ class XEP_0059(BasePlugin): def session_bind(self, jid): self.xmpp['xep_0030'].add_feature(Set.namespace) - def iterate(self, stanza, interface, results='substanzas',reverse=False, + def iterate(self, stanza, interface, results='substanzas', amount=10, reverse=False, recv_interface=None, pre_cb=None, post_cb=None): """ Create a new result set iterator for a given stanza query. @@ -170,6 +170,6 @@ class XEP_0059(BasePlugin): results -- The name of the interface containing the query results (typically just 'substanzas'). """ - return ResultIterator(stanza, interface, results, reverse=reverse, + return ResultIterator(stanza, interface, results, amount, reverse=reverse, recv_interface=recv_interface, pre_cb=pre_cb, post_cb=post_cb) diff --git a/slixmpp/plugins/xep_0313/mam.py b/slixmpp/plugins/xep_0313/mam.py index f78b62fa..e3268e9b 100644 --- a/slixmpp/plugins/xep_0313/mam.py +++ b/slixmpp/plugins/xep_0313/mam.py @@ -53,9 +53,12 @@ class XEP_0313(BasePlugin): iq['mam']['start'] = start iq['mam']['end'] = end iq['mam']['with'] = with_jid + amount = 10 if rsm: for key, value in rsm.items(): iq['mam']['rsm'][key] = str(value) + if key is 'max': + amount = value cb_data = {} def pre_cb(query): @@ -72,8 +75,8 @@ class XEP_0313(BasePlugin): result['mam']['results'] = results if iterator: - return self.xmpp['xep_0059'].iterate(iq, 'mam', 'results', reverse=reverse, - recv_interface='mam_fin', + return self.xmpp['xep_0059'].iterate(iq, 'mam', 'results', amount=amount, + reverse=reverse, recv_interface='mam_fin', pre_cb=pre_cb, post_cb=post_cb) collector = Collector( |