Sign in, Send a Message, and Disconnect¶
A common use case for SleekXMPP is to send one-off messages from time to time. For example, one use case could be sending out a notice when a shell script finishes a task.
We will create our one-shot bot based on the pattern explained in SleekXMPP Quickstart - Echo Bot. To start, we create a client class based on ClientXMPP and register a handler for the session_start event. We will also accept parameters for the JID that will receive our message, and the string content of the message.
import sleekxmpp class SendMsgBot(sleekxmpp.ClientXMPP): def __init__(self, jid, password, recipient, msg): super(SendMsgBot, self).__init__(jid, password) self.recipient = recipient self.msg = msg self.add_event_handler('session_start', self.start) def start(self, event): self.send_presence() self.get_roster()
def start(self, event): self.send_presence() self.get_roster() self.send_message(mto=self.recipient, mbody=self.msg)
Finally, we need to disconnect the client using disconnect. Now, sent stanzas are placed in a queue to pass them to the send thread. If we were to call disconnect without any parameters, then it is possible for the client to disconnect before the send queue is processed and the message is actually sent on the wire. To ensure that our message is processed, we use disconnect(wait=True).
def start(self, event): self.send_presence() self.get_roster() self.send_message(mto=self.recipient, mbody=self.msg) self.disconnect(wait=True)
If you happen to be adding stanzas to the send queue faster than the send thread can process them, then disconnect(wait=True) will block and not disconnect.
The final step is to create a small runner script for initialising our SendMsgBot class and adding some basic configuration options. By following the basic boilerplate pattern in SleekXMPP Quickstart - Echo Bot, we arrive at the code below. To experiment with this example, you can use:
python send_client.py -d -j email@example.com -t firstname.lastname@example.org -m "This is a message"
which will prompt for the password and then log in, send your message, and then disconnect. To test, open your regular IM client with the account you wish to send messages to. When you run the send_client.py example and instruct it to send your IM client account a message, you should receive the message you gave. If the two JIDs you use also have a mutual presence subscription (they’re on each other’s buddy lists) then you will also see the SendMsgBot client come online and then go offline.
#!/usr/bin/env python # -*- coding: utf-8 -*- """ SleekXMPP: The Sleek XMPP Library Copyright (C) 2010 Nathanael C. Fritz This file is part of SleekXMPP. See the file LICENSE for copying permission. """ import sys import logging import getpass from optparse import OptionParser import sleekxmpp # Python versions before 3.0 do not use UTF-8 encoding # by default. To ensure that Unicode is handled properly # throughout SleekXMPP, we will set the default encoding # ourselves to UTF-8. if sys.version_info < (3, 0): reload(sys) sys.setdefaultencoding('utf8') else: raw_input = input class SendMsgBot(sleekxmpp.ClientXMPP): """ A basic SleekXMPP bot that will log in, send a message, and then log out. """ def __init__(self, jid, password, recipient, message): sleekxmpp.ClientXMPP.__init__(self, jid, password) # The message we wish to send, and the JID that # will receive it. self.recipient = recipient self.msg = message # The session_start event will be triggered when # the bot establishes its connection with the server # and the XML streams are ready for use. We want to # listen for this event so that we we can initialize # our roster. self.add_event_handler("session_start", self.start) def start(self, event): """ Process the session_start event. Typical actions for the session_start event are requesting the roster and broadcasting an initial presence stanza. Arguments: event -- An empty dictionary. The session_start event does not provide any additional data. """ self.send_presence() self.get_roster() self.send_message(mto=self.recipient, mbody=self.msg, mtype='chat') # Using wait=True ensures that the send queue will be # emptied before ending the session. self.disconnect(wait=True) if __name__ == '__main__': # Setup the command line arguments. optp = OptionParser() # Output verbosity options. optp.add_option('-q', '--quiet', help='set logging to ERROR', action='store_const', dest='loglevel', const=logging.ERROR, default=logging.INFO) optp.add_option('-d', '--debug', help='set logging to DEBUG', action='store_const', dest='loglevel', const=logging.DEBUG, default=logging.INFO) optp.add_option('-v', '--verbose', help='set logging to COMM', action='store_const', dest='loglevel', const=5, default=logging.INFO) # JID and password options. optp.add_option("-j", "--jid", dest="jid", help="JID to use") optp.add_option("-p", "--password", dest="password", help="password to use") optp.add_option("-t", "--to", dest="to", help="JID to send the message to") optp.add_option("-m", "--message", dest="message", help="message to send") opts, args = optp.parse_args() # Setup logging. logging.basicConfig(level=opts.loglevel, format='%(levelname)-8s %(message)s') if opts.jid is None: opts.jid = raw_input("Username: ") if opts.password is None: opts.password = getpass.getpass("Password: ") if opts.to is None: opts.to = raw_input("Send To: ") if opts.message is None: opts.message = raw_input("Message: ") # Setup the EchoBot and register plugins. Note that while plugins may # have interdependencies, the order in which you register them does # not matter. xmpp = SendMsgBot(opts.jid, opts.password, opts.to, opts.message) xmpp.register_plugin('xep_0030') # Service Discovery xmpp.register_plugin('xep_0199') # XMPP Ping # If you are working with an OpenFire server, you may need # to adjust the SSL version used: # xmpp.ssl_version = ssl.PROTOCOL_SSLv3 # If you want to verify the SSL certificates offered by a server: # xmpp.ca_certs = "path/to/ca/cert" # Connect to the XMPP server and start processing XMPP stanzas. if xmpp.connect(): # If you do not have the dnspython library installed, you will need # to manually specify the name of the server if it does not match # the one in the JID. For example, to use Google Talk you would # need to use: # # if xmpp.connect(('talk.google.com', 5222)): # ... xmpp.process(block=True) print("Done") else: print("Unable to connect.")