]> gitweb.pimeys.fr Git - bots/bbc.git/blob - skeleton.py
Ménage basique à partir de Skeleton
[bots/bbc.git] / skeleton.py
1 #!/usr/bin/python
2 # -*- encoding: utf-8 -*-
3
4 """ Codé par 20-100
5
6 Un bot IRC qui ne fait rien. Base pour en coder un vrai.
7 """
8
9 import irclib
10 import ircbot
11 import random
12 import time
13 import re
14 from commands import getstatusoutput as ex
15
16 # on récupère la config
17 import config
18
19
20 def get_config_logfile(serveur):
21 """Renvoie le nom du fichier de log en fonction du serveur."""
22 serveurs = {"acoeur.crans.org" : "acoeur", "irc.crans.org" : "crans"}
23 return config.logfile_template % (serveurs[serveur])
24
25 def log(serveur, channel, auteur=None, message=None):
26 f = open(get_config_logfile(serveur), "a")
27 if auteur == message == None:
28 # alors c'est que c'est pas un channel mais juste une ligne de log
29 chain = "%s %s" % (time.strftime("%F %T"), channel)
30 else:
31 chain = "%s [%s:%s] %s" % (time.strftime("%F %T"), channel, auteur, message)
32 f.write(chain + "\n")
33 if config.debug_stdout:
34 print chain
35 f.close()
36
37 class UnicodeBotError(Exception):
38 pass
39 def bot_unicode(chain):
40 try:
41 unicode(chain, "utf8")
42 except UnicodeDecodeError as exc:
43 raise UnicodeBotError
44
45 class Skeleton(ircbot.SingleServerIRCBot):
46 def __init__(self, serveur, debug=False):
47 temporary_pseudo = config.irc_pseudo + str(random.randrange(10000, 100000))
48 ircbot.SingleServerIRCBot.__init__(self, [(serveur, 6667)],
49 temporary_pseudo, config.ircname, 10)
50 self.debug = debug
51 self.serveur = serveur
52 self.overops = config.overops
53 self.ops = self.overops + config.ops
54 self.chanlist = config.chanlist
55 self.stay_channels = config.stay_channels
56 self.quiet_channels = config.quiet_channels
57 self.last_perdu = 0
58
59 def give_me_my_pseudo(self, serv):
60 serv.privmsg("NickServ", "RECOVER %s %s" % (config.irc_pseudo, config.irc_password))
61 serv.privmsg("NickServ", "RELEASE %s %s" % (config.irc_pseudo, config.irc_password))
62 time.sleep(0.3)
63 serv.nick(config.irc_pseudo)
64
65 def on_welcome(self, serv, ev):
66 self.serv = serv # ça serv ira :)
67 self.give_me_my_pseudo(serv)
68 serv.privmsg("NickServ", "identify %s" % (config.irc_password))
69 log(self.serveur, "Connected")
70 if self.debug:
71 self.chanlist = ["#bot"]
72 for c in self.chanlist:
73 log(self.serveur, "JOIN %s" % (c))
74 serv.join(c)
75
76 def pourmoi(self, serv, message):
77 """renvoie (False, lemessage) ou (True, le message amputé de "pseudo: ")"""
78 pseudo = self.nick
79 size = len(pseudo)
80 if message[:size] == pseudo and len(message) > size and message[size] == ":":
81 return (True, message[size + 1:].lstrip(" "))
82 else:
83 return (False, message)
84
85 def on_privmsg(self, serv, ev):
86 message = ev.arguments()[0]
87 auteur = irclib.nm_to_n(ev.source())
88 try:
89 bot_unicode(message)
90 except UnicodeBotError:
91 if config.utf8_trigger:
92 serv.privmsg(auteur, random.choice(config.utf8_fail_answers).encode("utf8"))
93 return
94 message = message.split()
95 cmd = message[0].lower()
96 notunderstood = False
97 if cmd == "help":
98 helpdico = {"help" : ["""HELP <commande>
99 Affiche de l'aide sur la commande""", None, None],
100 "join" : [None, """JOIN <channel>
101 Me fait rejoindre le channel""", None],
102 "leave" : [None, """LEAVE <channel>
103 Me fait quitter le channel (sauf s'il est dans ma stay_list).""", None],
104 "quiet" : [None, """QUIET <channel>
105 Me rend silencieux sur le channel.""", None],
106 "noquiet" : [None, """NOQUIET <channel>
107 Me rend la parole sur le channel.""", None],
108 "say" : [None, None, """SAY <channel> <message>
109 Me fait parler sur le channel."""],
110 "do" : [None, None, """DO <channel> <action>
111 Me fait faitre une action (/me) sur le channel."""],
112 "stay" : [None, None, """STAY <channel>
113 Ajoute le channel à ma stay_list."""],
114 "nostay" : [None, None, """NOSTAY <channel>
115 Retire le channel de ma stay_list."""],
116 "ops" : [None, None, """OPS
117 Affiche la liste des ops."""],
118 "overops" : [None, None, """OVEROPS
119 Affiche la liste des overops."""],
120 "kick" : [None, None, """KICK <channel> <pseudo> [<raison>]
121 Kicke <pseudo> du channel (Il faut bien entendu que j'y sois op)."""],
122 "die" : [None, None, """DIE
123 Me déconnecte du serveur IRC."""]
124 }
125 helpmsg_default = "Liste des commandes disponibles :\nHELP"
126 helpmsg_ops = " JOIN LEAVE QUIET NOQUIET LOST RECONNECT"
127 helpmsg_overops = " SAY DO STAY NOSTAY OPS OVEROPS KICK DIE"
128 op, overop = auteur in self.ops, auteur in self.overops
129 if len(message) == 1:
130 helpmsg = helpmsg_default
131 if op:
132 helpmsg += helpmsg_ops
133 if overop:
134 helpmsg += helpmsg_overops
135 else:
136 helpmsgs = helpdico.get(message[1].lower(), ["Commande inconnue.", None, None])
137 helpmsg = helpmsgs[0]
138 if op and helpmsgs[1]:
139 if helpmsg:
140 helpmsg += "\n" + helpmsgs[1]
141 else:
142 helpmsg = helpmsgs[1]
143 if overop and helpmsgs[2]:
144 if helpmsg:
145 helpmsg += "\n" + helpmsgs[2]
146 else:
147 helpmsg = helpmsgs[2]
148 for ligne in helpmsg.split("\n"):
149 serv.privmsg(auteur, ligne)
150 elif cmd == "join":
151 if auteur in self.ops:
152 if len(message) > 1:
153 if message[1] in self.chanlist:
154 serv.privmsg(auteur, "Je suis déjà sur %s" % (message[1]))
155 else:
156 serv.join(message[1])
157 self.chanlist.append(message[1])
158 serv.privmsg(auteur, "Channels : " + " ".join(self.chanlist))
159 log(self.serveur, "priv", auteur, " ".join(message))
160 else:
161 serv.privmsg(auteur, "Channels : " + " ".join(self.chanlist))
162 else:
163 notunderstood = True
164 elif cmd == "leave":
165 if auteur in self.ops and len(message) > 1:
166 if message[1] in self.chanlist:
167 if not (message[1] in self.stay_channels) or auteur in self.overops:
168 self.quitter(message[1], " ".join(message[2:]))
169 self.chanlist.remove(message[1])
170 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
171 else:
172 serv.privmsg(auteur, "Non, je reste !")
173 log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
174 else:
175 serv.privmsg(auteur, "Je ne suis pas sur %s" % (message[1]))
176 else:
177 notunderstood = True
178 elif cmd == "stay":
179 if auteur in self.overops:
180 if len(message) > 1:
181 if message[1] in self.stay_channels:
182 log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
183 serv.privmsg(auteur, "Je stay déjà sur %s." % (message[1]))
184 else:
185 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
186 self.stay_channels.append(message[1])
187 serv.privmsg(auteur, "Stay channels : " + " ".join(self.stay_channels))
188 else:
189 serv.privmsg(auteur, "Stay channels : " + " ".join(self.stay_channels))
190 else:
191 notunderstood = True
192 elif cmd == "nostay":
193 if auteur in self.overops:
194 if len(message) > 1:
195 if message[1] in self.stay_channels:
196 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
197 self.stay_channels.remove(message[1])
198 serv.privmsg(auteur, "Stay channels : " + " ".join(self.stay_channels))
199 else:
200 log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
201 serv.privmsg(auteur, "Je ne stay pas sur %s." % (message[1]))
202
203 else:
204 notunderstood = True
205 elif cmd == "die":
206 if auteur in self.overops:
207 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
208 self.mourir()
209 else:
210 notunderstood = True
211 elif cmd == "quiet":
212 if auteur in self.ops:
213 if len(message) > 1:
214 if message[1] in self.quiet_channels:
215 serv.privmsg(auteur, "Je me la ferme déjà sur %s" % (message[1]))
216 log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
217 else:
218 self.quiet_channels.append(message[1])
219 serv.privmsg(auteur, "Quiet channels : " + " ".join(self.quiet_channels))
220 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
221 else:
222 serv.privmsg(auteur, "Quiet channels : " + " ".join(self.quiet_channels))
223 else:
224 notunderstood = True
225 elif cmd == "noquiet":
226 if auteur in self.ops:
227 if len(message) > 1:
228 if message[1] in self.quiet_channels:
229 self.quiet_channels.remove(message[1])
230 serv.privmsg(auteur, "Quiet channels : " + " ".join(self.quiet_channels))
231 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
232 else:
233 serv.privmsg(auteur, "Je ne me la ferme pas sur %s." % (message[1]))
234 log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
235 else:
236 notunderstood = True
237 elif cmd == "say":
238 if auteur in self.overops and len(message) > 2:
239 serv.privmsg(message[1], " ".join(message[2:]))
240 log(self.serveur, "priv", auteur, " ".join(message))
241 elif len(message) <= 2:
242 serv.privmsg(auteur, "Syntaxe : SAY <channel> <message>")
243 else:
244 notunderstood = True
245 elif cmd == "do":
246 if auteur in self.overops and len(message) > 2:
247 serv.action(message[1], " ".join(message[2:]))
248 log(self.serveur, "priv", auteur, " ".join(message))
249 elif len(message) <= 2:
250 serv.privmsg(auteur, "Syntaxe : DO <channel> <action>")
251 else:
252 notunderstood = True
253 elif cmd == "kick":
254 if auteur in self.overops and len(message) > 2:
255 serv.kick(message[1], message[2], " ".join(message[3:]))
256 log(self.serveur, "priv", auteur, " ".join(message))
257 elif len(message) <= 2:
258 serv.privmsg(auteur, "Syntaxe : KICK <channel> <pseudo> [<raison>]")
259 else:
260 notunderstood = True
261 elif cmd == "ops":
262 if auteur in self.overops:
263 serv.privmsg(auteur, " ".join(self.ops))
264 else:
265 notunderstood = True
266 elif cmd == "overops":
267 if auteur in self.overops:
268 serv.privmsg(auteur, " ".join(self.overops))
269 else:
270 notunderstood = True
271 else:
272 notunderstood = True
273 if notunderstood:
274 serv.privmsg(auteur, "Je n'ai pas compris. Essayez HELP…")
275
276 def on_pubmsg(self, serv, ev):
277 auteur = irclib.nm_to_n(ev.source())
278 canal = ev.target()
279 message = ev.arguments()[0]
280 try:
281 bot_unicode(message)
282 except UnicodeBotError:
283 if config.utf8_trigger and not canal in self.quiet_channels:
284 serv.privmsg(canal, (u"%s: %s" % (auteur, random.choice(config.utf8_fail_answers))).encode("utf8"))
285 return
286 pour_moi, message = self.pourmoi(serv, message)
287 if pour_moi and message.split() != []:
288 cmd = message.split()[0].lower()
289 try:
290 args = " ".join(message.split()[1:])
291 except:
292 args = ""
293 if cmd in ["meurs", "die", "crève"]:
294 if auteur in self.overops:
295 log(self.serveur, canal, auteur, message + "[successful]")
296 self.mourir()
297 else:
298 serv.privmsg(canal, ("%s: %s" % (auteur, random.choice(config.quit_fail_messages))).encode("utf8"))
299 log(self.serveur, canal, auteur, message + "[failed]")
300
301 elif cmd in ["part", "leave", "dégage", "va-t-en", "tut'tiresailleurs, c'estmesgalets"]:
302 if auteur in self.ops and (not (canal in self.stay_channels)
303 or auteur in self.overops):
304 self.quitter(canal)
305 log(self.serveur, canal, auteur, message + "[successful]")
306 if canal in self.chanlist:
307 self.chanlist.remove(canal)
308 else:
309 serv.privmsg(canal, ("%s: %s" % (auteur, random.choice(config.leave_fail_messages))).encode("utf8"))
310 log(self.serveur, canal, auteur, message + "[failed]")
311
312 elif cmd in ["deviens", "pseudo"]:
313 if auteur in self.ops:
314 become = args
315 serv.nick(become)
316 log(self.serveur, canal, auteur, message + "[successful]")
317
318 elif cmd in ["ping"] and not canal in self.quiet_channels:
319 serv.privmsg(canal, "%s: pong" % (auteur))
320 else:
321 if not canal in self.quiet_channels:
322 pass
323
324 def on_action(self, serv, ev):
325 action = ev.arguments()[0]
326 auteur = irclib.nm_to_n(ev.source())
327 channel = ev.target()
328 try:
329 bot_unicode(action)
330 except UnicodeBotError:
331 if config.utf8_trigger and not channel in self.quiet_channels:
332 serv.privmsg(channel, (u"%s: %s" % (auteur, random.choice(config.utf8_fail_answers))).encode("utf8"))
333 return
334 mypseudo = self.nick
335
336 def on_kick(self, serv, ev):
337 auteur = irclib.nm_to_n(ev.source())
338 channel = ev.target()
339 victime = ev.arguments()[0]
340 raison = ev.arguments()[1]
341 if victime == self.nick:
342 log(self.serveur, "%s kické de %s par %s (raison : %s)" % (victime, channel, auteur, raison))
343 time.sleep(2)
344 serv.join(channel)
345 return
346 l1, l2 = config.kick_answers, config.kick_actions
347 n1, n2 = len(l1), len(l2)
348 i = random.randrange(n1 + n2)
349 if i >= n1:
350 serv.action(channel, l2[i - n1].format(auteur).encode("utf8"))
351 else:
352 serv.privmsg(channel, l1[i].format(auteur).encode("utf8"))
353
354 def quitter(self, chan, leave_message=None):
355 if leave_message == None:
356 leave_message = random.choice(config.leave_messages)
357 self.serv.part(chan, message=leave_message.encode("utf8"))
358
359 def mourir(self):
360 quit_message = random.choice(config.quit_messages)
361 self.die(msg=quit_message.encode("utf8"))
362
363 def _getnick(self):
364 return self.serv.get_nickname()
365 nick = property(_getnick)
366
367
368 if __name__ == "__main__":
369 import sys
370 if len(sys.argv) == 1:
371 print "Usage : bbc.py <serveur> [--debug]"
372 exit(1)
373 serveur = sys.argv[1]
374 if "debug" in sys.argv or "--debug" in sys.argv:
375 debug = True
376 else:
377 debug = False
378 if "--quiet" in sys.argv:
379 config.debug_stdout = False
380 serveurs = {"irc" : "irc.crans.org", "crans" : "irc.crans.org", "irc.crans.org" : "irc.crans.org"}
381 try:
382 serveur = serveurs[serveur]
383 except KeyError:
384 print "Server Unknown : %s" % (serveur)
385 exit(404)
386 bot = Skeleton(serveur, debug)
387 bot.start()