From cc8c808e10b484fb7593d8eb84ac11fcd6925c5c Mon Sep 17 00:00:00 2001 From: Madhur Garg Date: Sat, 20 Jul 2019 01:49:03 +0530 Subject: Added await on the get_info call so that MAM support is checked first before proceeding further. --- poezio/mam.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/poezio/mam.py b/poezio/mam.py index 93e76d34..28173994 100644 --- a/poezio/mam.py +++ b/poezio/mam.py @@ -42,10 +42,9 @@ async def query(self, remote_jid, start, end, top): self.start_date = start self.end_date = end text_buffer = self._text_buffer - def callback(iq): - if 'urn:xmpp:mam:2' not in iq['disco_info'].get_features(): - return self.core.information("This MUC doesn't support MAM.", "Error") - self.core.xmpp.plugin['xep_0030'].get_info(jid=remote_jid, callback=callback) + iq = await self.core.xmpp.plugin['xep_0030'].get_info(jid=remote_jid) + if 'urn:xmpp:mam:2' not in iq['disco_info'].get_features(): + return self.core.information("This MUC doesn't support MAM.", "Error") if top: results = self.core.xmpp['xep_0313'].retrieve(jid=self.remote_jid, iterator=True, reverse=top, end=self.end_date) -- cgit v1.2.3