Mercurial > xib

annotate start_bots_from_xml_config.py @ 172:64a0e9636ae6

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
removed encoding.py, charsets handling is now in irclib and custom charsets can be set per server in the configuration file Signed-off-by: Charly COSTE <changaco@changaco.net>
author Charly COSTE <changaco@changaco.net>
date Tue, 02 Feb 2010 20:19:50 +0100
parents 489c157d9e82
children 102f895347ff
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
1 #!/usr/bin/env python
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
2 # -*- coding: utf-8 -*-
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
3
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
4
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
5 # *** LICENSE ***
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
6 # This program is free software: you can redistribute it and/or modify
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
7 # it under the terms of the GNU General Public License as published by
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
8 # the Free Software Foundation, either version 3 of the License, or
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
9 # (at your option) any later version.
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
10 #
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
11 # This program is distributed in the hope that it will be useful,
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
14 # GNU General Public License for more details.
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
15 #
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
16 # You should have received a copy of the GNU General Public License
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
18
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
19
124
99f3dee1fad7 code cleaning
Charly COSTE <changaco@changaco.net>
parents: 86
diff changeset
20 from xml.dom.minidom import parse
0
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
21 from time import sleep
13
a9077fa9cac9 Added optional parameter to start_bots_from_xml_config.py
Charly COSTE <changaco@changaco.net>
parents: 0
diff changeset
22 import sys
17
32a35f7eff70 Rewrote/modified many things, multiple bridges should now work and are preferred over multiple bots.
Charly COSTE <changaco@changaco.net>
parents: 13
diff changeset
23 import traceback
0
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
24
124
99f3dee1fad7 code cleaning
Charly COSTE <changaco@changaco.net>
parents: 86
diff changeset
25 from bot import Bot
0
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
26
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
27
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
28 try:
13
a9077fa9cac9 Added optional parameter to start_bots_from_xml_config.py
Charly COSTE <changaco@changaco.net>
parents: 0
diff changeset
29 if len(sys.argv) > 1:
a9077fa9cac9 Added optional parameter to start_bots_from_xml_config.py
Charly COSTE <changaco@changaco.net>
parents: 0
diff changeset
30 config = parse(sys.argv[1])
a9077fa9cac9 Added optional parameter to start_bots_from_xml_config.py
Charly COSTE <changaco@changaco.net>
parents: 0
diff changeset
31 else:
a9077fa9cac9 Added optional parameter to start_bots_from_xml_config.py
Charly COSTE <changaco@changaco.net>
parents: 0
diff changeset
32 config = parse('config.xml')
0
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
33 except IOError:
19
c1b84196c100 Changed format of non-debug error messages, fixed IRC namreply handling, prevented crash when receiving bad XMPP stanza.
Charly COSTE <changaco@changaco.net>
parents: 17
diff changeset
34 print '[Error] configuration file is missing or cannot be read'
0
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
35 quit(1)
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
36
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
37 bots_jids = []
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
38 for bot_el in config.getElementsByTagName('bot'):
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
39 if bot_el.getAttribute('jid') in bots_jids:
19
c1b84196c100 Changed format of non-debug error messages, fixed IRC namreply handling, prevented crash when receiving bad XMPP stanza.
Charly COSTE <changaco@changaco.net>
parents: 17
diff changeset
40 print '[Error] you cannot have two bots using the same JID'
0
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
41 quit(2)
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
42 bots_jids.append(bot_el.getAttribute('jid'))
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
43
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
44
171
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
45
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
46 bots = []
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
47 for bot_el in config.getElementsByTagName('bot'):
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
48 debug = False
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
49 if bot_el.hasAttribute('debug'):
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
50 if bot_el.getAttribute('debug') == 'true':
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
51 debug = True
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
52 admins_jid = []
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
53 for admin_el in bot_el.getElementsByTagName('admin'):
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
54 if admin_el.hasAttribute('jid'):
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
55 admins_jid.append(admin_el.getAttribute('jid'))
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
56 bot = Bot(bot_el.getAttribute('jid'), bot_el.getAttribute('password'), bot_el.getAttribute('nickname'), admins_jid=admins_jid, debug=debug)
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
57 bots.append(bot)
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
58 for bridge_el in bot_el.getElementsByTagName('bridge'):
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
59 xmpp_room = bridge_el.getElementsByTagName('xmpp-room')[0]
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
60 irc = bridge_el.getElementsByTagName('irc')[0]
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
61
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
62 irc_connection_interval = 1
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
63 if irc.hasAttribute('connection_interval'):
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
64 try:
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
65 irc_connection_interval = float(irc.getAttribute('connection_interval'))
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
66 except ValueError:
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
67 print '[Error] the value of connection_interval must be a number'
172
64a0e9636ae6 removed encoding.py, charsets handling is now in irclib and custom charsets can be set per server in the configuration file
Charly COSTE <changaco@changaco.net>
parents: 171
diff changeset
68 if irc.hasAttribute('charsets'):
64a0e9636ae6 removed encoding.py, charsets handling is now in irclib and custom charsets can be set per server in the configuration file
Charly COSTE <changaco@changaco.net>
parents: 171
diff changeset
69 irc_charsets = irc.getAttribute('charsets').split()
64a0e9636ae6 removed encoding.py, charsets handling is now in irclib and custom charsets can be set per server in the configuration file
Charly COSTE <changaco@changaco.net>
parents: 171
diff changeset
70 else:
64a0e9636ae6 removed encoding.py, charsets handling is now in irclib and custom charsets can be set per server in the configuration file
Charly COSTE <changaco@changaco.net>
parents: 171
diff changeset
71 irc_charsets = None
171
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
72
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
73 if bridge_el.hasAttribute('say_level'):
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
74 say_level = bridge_el.getAttribute('say_level')
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
75 else:
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
76 say_level = 'all'
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
77
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
78 if bridge_el.hasAttribute('mode'):
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
79 mode = bridge_el.getAttribute('mode')
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
80 else:
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
81 mode = 'normal'
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
82
172
64a0e9636ae6 removed encoding.py, charsets handling is now in irclib and custom charsets can be set per server in the configuration file
Charly COSTE <changaco@changaco.net>
parents: 171
diff changeset
83 bot.new_bridge(xmpp_room.getAttribute('jid'), irc.getAttribute('chan'), irc.getAttribute('server'), mode, say_level, irc_connection_interval=irc_connection_interval, irc_charsets=irc_charsets)
171
489c157d9e82 display error when bot creation fails
Charly COSTE <changaco@changaco.net>
parents: 161
diff changeset
84
0
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
85 try:
155
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
86 if len(bots) == 0:
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
87 print 'No bots in the configuration file, exiting ...'
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
88 exit(0)
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
89
0
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
90 while True:
155
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
91 for bot in bots:
161
c021656e4c54 minor fix in start_bots_from_xml_config.py
Charly COSTE <changaco@changaco.net>
parents: 155
diff changeset
92 if bot.halt and len(bot.xmpp_connections) == 0:
155
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
93 bots.remove(bot)
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
94 if len(bots) == 0:
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
95 raise Exception()
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
96 sleep(10)
0
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
97 except:
155
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
98 if len(bots) == 0:
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
99 print 'All bots have been shut down, exiting ...'
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
100 exit(0)
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
101
17
32a35f7eff70 Rewrote/modified many things, multiple bridges should now work and are preferred over multiple bots.
Charly COSTE <changaco@changaco.net>
parents: 13
diff changeset
102 for bot in bots:
155
63db565438bd fixed the halt command
Charly COSTE <changaco@changaco.net>
parents: 150
diff changeset
103 bots.remove(bot)
17
32a35f7eff70 Rewrote/modified many things, multiple bridges should now work and are preferred over multiple bots.
Charly COSTE <changaco@changaco.net>
parents: 13
diff changeset
104 del bot
32a35f7eff70 Rewrote/modified many things, multiple bridges should now work and are preferred over multiple bots.
Charly COSTE <changaco@changaco.net>
parents: 13
diff changeset
105 traceback.print_exc()
0
4c842d23d4ce Initial commit, version 0.1
Charly COSTE <changaco@changaco.net>
parents:
diff changeset
106 quit(3)