summaryrefslogtreecommitdiff
path: root/examples/disco_browser.py
diff options
context:
space:
mode:
authorLance Stout <lancestout@gmail.com>2012-08-07 16:44:52 -0700
committerLance Stout <lancestout@gmail.com>2012-08-07 16:44:52 -0700
commit295d23ccf37860245d0e86b38a3a2315532f2972 (patch)
tree4d9795172bd6d0177ea2b8c72736ba99cfe18ac4 /examples/disco_browser.py
parentaebcf6ff828bf179af95df38d7f8ffa54f289c36 (diff)
downloadslixmpp-295d23ccf37860245d0e86b38a3a2315532f2972.tar.gz
slixmpp-295d23ccf37860245d0e86b38a3a2315532f2972.tar.bz2
slixmpp-295d23ccf37860245d0e86b38a3a2315532f2972.tar.xz
slixmpp-295d23ccf37860245d0e86b38a3a2315532f2972.zip
Fix disco browser example to handle errors.
Diffstat (limited to 'examples/disco_browser.py')
-rwxr-xr-xexamples/disco_browser.py89
1 files changed, 47 insertions, 42 deletions
diff --git a/examples/disco_browser.py b/examples/disco_browser.py
index ff0af97d..67e12d19 100755
--- a/examples/disco_browser.py
+++ b/examples/disco_browser.py
@@ -15,6 +15,7 @@ import getpass
from optparse import OptionParser
import sleekxmpp
+from sleekxmpp.exceptions import IqError, IqTimeout
# Python versions before 3.0 do not use UTF-8 encoding
@@ -83,50 +84,54 @@ class Disco(sleekxmpp.ClientXMPP):
self.get_roster()
self.send_presence()
- if self.get in self.info_types:
- # By using block=True, the result stanza will be
- # returned. Execution will block until the reply is
- # received. Non-blocking options would be to listen
- # for the disco_info event, or passing a handler
- # function using the callback parameter.
- info = self['xep_0030'].get_info(jid=self.target_jid,
- node=self.target_node,
- block=True)
- if self.get in self.items_types:
- # The same applies from above. Listen for the
- # disco_items event or pass a callback function
- # if you need to process a non-blocking request.
- items = self['xep_0030'].get_items(jid=self.target_jid,
- node=self.target_node,
- block=True)
+ try:
+ if self.get in self.info_types:
+ # By using block=True, the result stanza will be
+ # returned. Execution will block until the reply is
+ # received. Non-blocking options would be to listen
+ # for the disco_info event, or passing a handler
+ # function using the callback parameter.
+ info = self['xep_0030'].get_info(jid=self.target_jid,
+ node=self.target_node,
+ block=True)
+ elif self.get in self.items_types:
+ # The same applies from above. Listen for the
+ # disco_items event or pass a callback function
+ # if you need to process a non-blocking request.
+ items = self['xep_0030'].get_items(jid=self.target_jid,
+ node=self.target_node,
+ block=True)
+ else:
+ logging.error("Invalid disco request type.")
+ return
+ except IqError as e:
+ logging.error("Entity returned an error: %s" % e.iq['error']['condition'])
+ except IqTimeout:
+ logging.error("No response received.")
else:
- logging.error("Invalid disco request type.")
- self.disconnect()
- return
-
- header = 'XMPP Service Discovery: %s' % self.target_jid
- print(header)
- print('-' * len(header))
- if self.target_node != '':
- print('Node: %s' % self.target_node)
+ header = 'XMPP Service Discovery: %s' % self.target_jid
+ print(header)
print('-' * len(header))
-
- if self.get in self.identity_types:
- print('Identities:')
- for identity in info['disco_info']['identities']:
- print(' - %s' % str(identity))
-
- if self.get in self.feature_types:
- print('Features:')
- for feature in info['disco_info']['features']:
- print(' - %s' % feature)
-
- if self.get in self.items_types:
- print('Items:')
- for item in items['disco_items']['items']:
- print(' - %s' % str(item))
-
- self.disconnect()
+ if self.target_node != '':
+ print('Node: %s' % self.target_node)
+ print('-' * len(header))
+
+ if self.get in self.identity_types:
+ print('Identities:')
+ for identity in info['disco_info']['identities']:
+ print(' - %s' % str(identity))
+
+ if self.get in self.feature_types:
+ print('Features:')
+ for feature in info['disco_info']['features']:
+ print(' - %s' % feature)
+
+ if self.get in self.items_types:
+ print('Items:')
+ for item in items['disco_items']['items']:
+ print(' - %s' % str(item))
+ finally:
+ self.disconnect()
if __name__ == '__main__':