Make get_roster(block=False) work properly.

Fixes issue #136
2012-01-10 19:57:38 -08:00 · 2012-01-10 19:57:38 -08:00 · a86935a42f
commit a86935a42f
parent 6b9a55e62d
1 changed files with 5 additions and 1 deletions
--- a/sleekxmpp/clientxmpp.py
+++ b/sleekxmpp/clientxmpp.py
@ -237,9 +237,13 @@ class ClientXMPP(BaseXMPP):
        iq = self.Iq()
        iq['type'] = 'get'
        iq.enable('roster')
+
+        if not block and callback is None:
+            callback = lambda resp: self._handle_roster(resp, request=True)
+
        response = iq.send(block, timeout, callback)

-        if callback is None:
+        if block: 
            return self._handle_roster(response, request=True)

    def _handle_connected(self, event=None):