3 # JabberBot: A simple jabber/xmpp bot framework
4 # Copyright (c) 2007 Thomas Perl <thp@thpinfo.com>
6 # This program is free software; you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 3 of the License, or
9 # (at your option) any later version.
11 # This program is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 # GNU General Public License for more details.
16 # You should have received a copy of the GNU General Public License
17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
20 # This is an example JabberBot that serves as broadcasting server.
21 # Users can subscribe/unsubscribe to messages and send messages
22 # by using "broadcast". It also shows how to send message from
23 # outside the main loop, so you can inject messages into the bot
24 # from other threads or processes, too.
27 from jabberbot
import JabberBot
32 # Fill in the JID + Password of your JabberBot here...
33 (JID
, PASSWORD
) = ('my-jabber-id@jabberserver.example.org','my-password')
35 class BroadcastingJabberBot(JabberBot
):
36 """This is a simple broadcasting client. Use "subscribe" to subscribe to broadcasts, "unsubscribe" to unsubscribe and "broadcast" + message to send out a broadcast message. Automatic messages will be sent out all 60 seconds."""
38 def __init__( self
, jid
, password
, res
= None):
39 super( BroadcastingJabberBot
, self
).__init
__( jid
, password
, res
)
42 self
.message_queue
= []
43 self
.thread_killed
= False
45 def bot_subscribe( self
, mess
, args
):
46 """Subscribe to the broadcast list"""
48 if user
in self
.users
:
49 return 'You are already subscribed.'
51 self
.users
.append( user
)
52 self
.log( '%s subscribed to the broadcast.' % user
)
53 return 'You are now subscribed.'
55 def bot_unsubscribe( self
, mess
, args
):
56 """Unsubscribe from the broadcast list"""
58 if not user
in self
.users
:
59 return 'You are not subscribed!'
61 self
.users
.remove( user
)
62 self
.log( '%s unsubscribed from the broadcast.' % user
)
63 return 'You are now unsubscribed.'
65 def bot_broadcast( self
, mess
, args
):
66 """Sends out a broadcast, supply message as arguments (e.g. broadcast hello)"""
67 self
.message_queue
.append( 'broadcast: %s (from %s)' % ( args
, str(mess
.getFrom()), ))
68 self
.log( '%s sent out a message to %d users.' % ( str(mess
.getFrom()), len(self
.users
),))
71 if not len(self
.message_queue
):
74 # copy the message queue, then empty it
75 messages
= self
.message_queue
76 self
.message_queue
= []
78 for message
in messages
:
80 self
.log('sending "%s" to %d user(s).' % ( message
, len(self
.users
), ))
81 for user
in self
.users
:
82 self
.send( user
, message
)
84 def thread_proc( self
):
85 while not self
.thread_killed
:
86 self
.message_queue
.append('this is an automatic message, sent all 60 seconds :)')
89 if self
.thread_killed
:
93 bc
= BroadcastingJabberBot( JID
, PASSWORD
)
95 th
= threading
.Thread( target
= bc
.thread_proc
)
96 bc
.serve_forever( connect_callback
= lambda: th
.start())
97 bc
.thread_killed
= True