]> gitweb.pimeys.fr Git - bots/basile.git/blob - basile.py
66dab704126a61c3198c55baa7b18b438d0df4f2
[bots/basile.git] / basile.py
1 #!/usr/bin/python
2 # -*- coding:utf8 -*-
3
4 # Codé par 20-100 (commencé le 23/04/12)
5
6 """ Un bot IRC destiné à s'interfacer avec la Note Kfet 2015 """
7
8 import threading
9 import random
10 import time
11 import json
12 import re
13 import os
14 import signal
15 import sys
16
17 # Oui, j'ai recodé ma version d'irclib pour pouvoir rattrapper les SIGHUP
18 sys.path.insert(0, "/home/vincent/scripts/python-myirclib")
19 import irclib
20 import ircbot
21
22 from commands import getstatusoutput as ex
23
24 #: Config de basile
25 import config
26 #: Module responsable du dialogue avec la NoteKfet2015
27 import nk
28
29 # la partie qui réfère au fichier lui-même est mieux ici
30 # sinon on réfère la config et pas le fichier lui-même
31 import os
32 config.thisfile = os.path.realpath(__file__)
33
34 def get_config_logfile(serveur):
35 """Renvoie le nom du fichier de log en fonction du ``serveur`` et de la config."""
36 serveurs = {"acoeur.crans.org" : "acoeur",
37 "irc.crans.org" : "crans"}
38 return config.logfile_template % (serveurs[serveur],)
39
40 def get_filesize():
41 """Récupère la taille de ce fichier."""
42 return ex("ls -s %s" % (config.thisfile))[1].split()[0]
43
44 def log(serveur, channel, auteur=None, message=None):
45 """Enregistre une ligne de log."""
46 if auteur == message == None:
47 # alors c'est que c'est pas un channel mais juste une ligne de log
48 chain = u"%s %s" % (time.strftime("%F %T"), channel)
49 else:
50 chain = u"%s [%s:%s] %s" % (time.strftime("%F %T"), channel, auteur, message)
51 f = open(get_config_logfile(serveur), "a")
52 f.write((chain + u"\n").encode("utf-8"))
53 f.close()
54 if config.debug_stdout:
55 print chain.encode("utf-8")
56
57 def ignore_event(serv, ev):
58 """Retourne ``True`` si il faut ignorer cet évènement."""
59 for (blackmask, exceptmask) in config.blacklisted_masks:
60 usermask = ev.source()
61 if exceptmask is None:
62 exceptit = False
63 else:
64 exceptit = bool(irclib.mask_matches(usermask, exceptmask))
65 blackit = bool(irclib.mask_matches(usermask, blackmask))
66 return blackit and not exceptit
67
68 def regex_join(liste, avant=u".*(?:^| )", apres=u"(?:$|\.| |,|;).*"):
69 """Fabrique une regexp à partir d'une liste d'éléments à matcher."""
70 return avant + u"(" + u"|".join(liste) + u")" + apres
71
72 def is_something(chain, regexp=None, matches=[], avant=u".*(?:^| )", apres=u"(?:$|\.| |,|;).*",
73 case_sensitive=False):
74 """Vérifie si chain contient un des éléments de ``matches``.
75 Si ``regexp`` est fournie, c'est simplement elle qui est testée"""
76 if case_sensitive:
77 chain = chain.lower()
78 if regexp == None:
79 regexp = regex_join(matches, avant, apres)
80 regexp = re.compile(regexp)
81 o = regexp.match(chain)
82 return o
83
84 def regexp_compile():
85 """Compilation des regexp à partir de la conf.
86 Place les résultats dans le namespace de ``config``"""
87 config.insult_regexp = regex_join(config.insultes, avant=u".*(?:^| |')")
88 config.insult_regexp_compiled = re.compile(config.insult_regexp)
89
90 config.not_insult_regexp = u".*pas %s%s" % (config.amplifier_regexp, config.insult_regexp)
91 config.not_insult_regexp_compiled = re.compile(config.not_insult_regexp)
92
93 config.compliment_regexp = regex_join(config.compliment_triggers, avant=u".*(?:^| |')")
94 config.compliment_regexp_compiled = re.compile(config.compliment_regexp)
95
96 config.perdu_regexp = regex_join(config.perdu)
97 config.perdu_regexp_compiled = re.compile(config.perdu_regexp)
98
99 config.tag_regexp = regex_join(config.tag_triggers)
100 config.tag_regexp_compiled = re.compile(config.tag_regexp)
101
102 config.gros_regexp = regex_join(config.gros)
103 config.gros_regexp_compiled = re.compile(config.gros_regexp)
104
105 config.tesla_regexp = regex_join(config.tesla_triggers, avant=u"^", apres="$")
106 config.tesla_regexp_compiled = re.compile(config.tesla_regexp)
107
108 config.merci_regexp = regex_join(config.merci_triggers)
109 config.merci_regexp_compiled = re.compile(config.merci_regexp)
110
111 config.tamere_regexp = regex_join(config.tamere_triggers)
112 config.tamere_regexp_compiled = re.compile(config.tamere_regexp)
113
114 config.bonjour_regexp = regex_join(config.bonjour_triggers, avant=u"^")
115 config.bonjour_regexp_compiled = re.compile(config.bonjour_regexp)
116
117 config.bonne_nuit_regexp = regex_join(config.bonne_nuit_triggers, avant=u"^")
118 config.bonne_nuit_regexp_compiled = re.compile(config.bonne_nuit_regexp)
119
120 config.pan_regexp = regex_join(config.pan_triggers, avant=".*", apres=".*")
121 config.pan_regexp_compiled = re.compile(config.pan_regexp)
122
123
124 regexp_compile()
125 def is_insult(chain, debug=True):
126 """Vérifie si ``chain`` contient une insulte."""
127 return is_something(chain, config.insult_regexp_compiled)
128 def is_not_insult(chain):
129 """Vérifie si ``chain`` contient une insulte à la forme négative."""
130 return is_something(chain, config.not_insult_regexp_compiled)
131 def is_compliment(chain, debug=True):
132 """Vérifie si ``chain`` contient un compliment."""
133 return is_something(chain, config.compliment_regexp_compiled)
134 def is_perdu(chain):
135 """Vérifie si ``chain`` contient une raison de perdre."""
136 return is_something(chain, config.perdu_regexp_compiled)
137 def is_tag(chain):
138 """Vérifie si ``chain`` demande de fermer sa gueule."""
139 return is_something(chain, config.tag_regexp_compiled)
140 def is_gros(chain):
141 """Vérifie si ``chain`` traite de gros."""
142 return is_something(chain, config.gros_regexp_compiled)
143 def is_tesla(chain):
144 """Vérifie si ``chain`` est un ping."""
145 return is_something(chain, config.tesla_regexp_compiled)
146 def is_merci(chain):
147 """Vérifie si ``chain`` contient un remerciement."""
148 return is_something(chain, config.merci_regexp_compiled)
149 def is_tamere(chain):
150 """Vérifie si ``chain`` traite ma mère."""
151 return is_something(chain, config.tamere_regexp_compiled)
152 def is_bad_action_trigger(chain,pseudo):
153 """Vérifie si ``chain`` est une action méchante.
154 On a besoin d'une regexp dynamique à cause du pseudo qui peut changer."""
155 return is_something(chain, matches=config.bad_action_triggers, avant=u"^",
156 apres="(?: [a-z]*ment)? %s($|\.| |,|;).*" % (pseudo))
157 def is_good_action_trigger(chain,pseudo):
158 """Vérifie si ``chain`` est une action gentille.
159 On a besoin d'une regexp dynamique à cause du pseudo qui peut changer."""
160 return is_something(chain, matches=config.good_action_triggers, avant=u"^",
161 apres="(?: [a-z]*ment)? %s($|\.| |,|;).*" % (pseudo))
162 def is_bonjour(chain):
163 """Vérifie si ``chain`` contient un bonjour."""
164 return is_something(chain, config.bonjour_regexp_compiled)
165 def is_bonne_nuit(chain):
166 """Vérifie si ``chain`` contient un bonne nuit."""
167 return is_something(chain, config.bonne_nuit_regexp_compiled)
168 def is_pan(chain):
169 """Vérifie si ``chain`` contient un pan."""
170 return is_something(chain, config.pan_regexp_compiled)
171
172 def is_time(conf):
173 """Vérifie si l'heure actuelle est entre les deux heures ``conf[0]`` et ``conf[1]``"""
174 _, _, _, h, m, s, _, _, _ = time.localtime()
175 return (conf[0], 0, 0) < (h, m, s) < (conf[1], 0, 0)
176 def is_day():
177 """Vérifie si on est le jour."""
178 return is_time(config.daytime)
179 def is_night():
180 """Vérifie si on est la nuit."""
181 return is_time(config.nighttime)
182
183
184 class UnicodeBotError(Exception):
185 """Erreur levée si quelqu'un fait du caca avec son encodage."""
186 pass
187
188 class CrashError(Exception):
189 """Pour pouvoir faire crasher Basile, parce que ça a l'air drôle."""
190 def __init__(self, msg=""):
191 Exception.__init__(self, msg)
192
193 def bot_unicode(chain):
194 """Essaye de décoder ``chain`` en UTF-8.
195 Lève une py:class:`UnicodeBotError` en cas d'échec."""
196 try:
197 return chain.decode("utf8")
198 except UnicodeDecodeError as exc:
199 raise UnicodeBotError
200
201
202 class Basile(ircbot.SingleServerIRCBot):
203 """Classe principale : définition du bot Basile."""
204 def __init__(self, serveur, debug=False):
205 temporary_pseudo = config.irc_pseudo + str(random.randrange(10000,100000))
206 ircbot.SingleServerIRCBot.__init__(self, [(serveur, 6667)],
207 temporary_pseudo, "Basile, le bot irc. [Codé par 20-100]", 10)
208 self.debug = debug
209 self.serveur = serveur
210 self.overops = config.overops
211 self.ops = self.overops + config.ops
212 self.report_bugs_to = config.report_bugs_to
213 self.chanlist = config.chanlist
214 self.identities = json.load(open(config.identities_file, "r"))
215 self.stay_channels = config.stay_channels
216 self.quiet_channels = config.quiet_channels
217 self.last_perdu = 0
218
219 ### Communication NK
220 def new_connection_NK(self, serv, username, password, typ="bdd"):
221 """Renvoie (``True``, <une socket ouverte et authentifiée sur la NoteKfet2015>)
222 ou bien (``False``, None)"""
223 try:
224 login_result, sock = nk.login(username, password, typ)
225 info, retcode, errmsg = login_result["msg"], login_result["retcode"], login_result["errmsg"]
226 except nk.NKRefused as exc:
227 for report in self.report_bugs_to:
228 serv.privmsg(report, "Le Serveur NK2015 est down.")
229 return (False, None, None)
230 except nk.NKHelloFailed as exc:
231 for report in self.report_bugs_to:
232 serv.privmsg(report,
233 "La version du protocole utilisée n'est pas supportée par le serveur NK2015.")
234 return (False, None, None)
235 except nk.NKUnknownError as exc:
236 erreurs = ["Une fucking erreur inconnue s'est produite"]
237 erreurs += str(exc).split("\n")
238 for report in self.report_bugs_to:
239 for err in erreurs:
240 serv.privmsg(report, err)
241 return (False, None, None)
242 except Exception as exc:
243 # Exception qui ne vient pas de la communication avec le serveur NK2015
244 log(self.serveur, "Erreur dans new_connection_NK\n" + str(exc))
245 return (False, None, None)
246 if retcode == 0:
247 return (True, info, sock)
248 else:
249 return (False, None, None)
250
251 ### Utilitaires
252 def _getnick(self):
253 """Récuère le nick effectif du bot sur le serveur."""
254 return self.serv.get_nickname()
255 nick = property(_getnick)
256
257 def give_me_my_pseudo(self, serv):
258 """Récupère le pseudo auprès de NickServ."""
259 serv.privmsg("NickServ", "RECOVER %s %s" % (config.irc_pseudo, config.irc_password))
260 serv.privmsg("NickServ", "RELEASE %s %s" % (config.irc_pseudo, config.irc_password))
261 time.sleep(0.3)
262 serv.nick(config.irc_pseudo)
263
264 def pourmoi(self, serv, message):
265 """Renvoie (False, lemessage) ou (True, le message amputé de "pseudo: ")"""
266 pseudo = self.nick
267 pseudo = pseudo.decode("utf-8")
268 size = len(pseudo)
269 if message[:size] == pseudo and len(message) > size and message[size] == ":":
270 return (True, message[size+1:].lstrip(" "))
271 else:
272 return (False, message)
273
274 ### Exécution d'actions
275 def lost(self, serv, channel, forced=False):
276 """Réaction à un trigger de perdu.
277 Annonce "J'ai perdu" sur le channel si on n'a pas perdu depuis un certain temps."""
278 if self.last_perdu + config.time_between_perdu < time.time() or forced:
279 if not channel in self.quiet_channels or forced:
280 serv.privmsg(channel, "J'ai perdu !")
281 self.last_perdu=time.time()
282 delay=config.time_between_perdu_trigger
283 delta=config.time_between_perdu_trigger_delta
284 serv.execute_delayed(random.randrange(delay-delta,delay+delta),self.lost,(serv,channel))
285
286 def quitter(self, chan, leave_message=None):
287 """Quitter un channel avec un message customisable."""
288 if leave_message == None:
289 leave_message = random.choice(config.leave_messages)
290 self.serv.part(chan, message=leave_message.encode("utf8"))
291
292 def mourir(self):
293 """Se déconnecter du serveur IRC avec un message customisable."""
294 quit_message = random.choice(config.quit_messages)
295 self.die(msg=quit_message.encode("utf8"))
296
297 def execute_reload(self, auteur=None):
298 """Recharge la config."""
299 reload(config)
300 regexp_compile()
301 if auteur in [None, "SIGHUP"]:
302 towrite = "Config reloaded" + " (SIGHUP received)" * (auteur == "SIGHUP")
303 for to in config.report_bugs_to:
304 self.serv.privmsg(to, towrite)
305 log(self.serveur, towrite)
306 return True, None
307 else:
308 return True, u"Config reloaded"
309
310 def crash(self, chan="nowhere", who="nobody"):
311 """Fait crasher le bot."""
312 where = "en privé" if chan == "priv" else "sur le chan %s" % chan
313 raise CrashError("Crash demandé par %s %s" % (who, where))
314
315 ACTIONS = {
316 "reload" : execute_reload,
317 }
318
319 def execute_something(self, something, params, place=None, auteur=None):
320 """Exécute une action et répond son résultat à ``auteur``
321 sur un chan ou en privé en fonction de ``place``"""
322 action = self.ACTIONS[something]
323 success, message = action(self, **params)
324 if message:
325 if irclib.is_channel(place):
326 message = "%s: %s" % (auteur, message.encode("utf-8"))
327 self.serv.privmsg(place, message)
328 log(self.serveur, place, auteur, something + "%r" % params + ("[successful]" if success else "[failed]"))
329
330 ### Surcharge des events du Bot
331 def on_welcome(self, serv, ev):
332 """À l'arrivée sur le serveur."""
333 self.serv = serv # ça serv ira :)
334 self.give_me_my_pseudo(serv)
335 serv.privmsg("NickServ", "IDENTIFY %s" % (config.irc_password))
336 log(self.serveur, "Connected")
337 if self.debug:
338 self.chanlist = ["#bot"]
339 for c in self.chanlist:
340 log(self.serveur, "JOIN %s" % (c))
341 serv.join(c)
342 # on ouvre la connexion note de Basile, special user
343 self.nk = self.new_connection_NK(serv, config.note_pseudo, config.note_password, "special")[2]
344 if self.nk == None:
345 for report in self.report_bugs_to:
346 serv.privmsg(report, "Connection to NK2015 failed, invalid password ?")
347
348 def on_privmsg(self, serv, ev):
349 """À la réception d'un message en privé."""
350 if ignore_event(serv, ev):
351 return
352 message = ev.arguments()[0]
353 auteur = irclib.nm_to_n(ev.source())
354 try:
355 message = bot_unicode(message)
356 except UnicodeBotError:
357 if config.utf8_trigger:
358 serv.privmsg(auteur, random.choice(config.utf8_fail_answers).encode("utf8"))
359 return
360 message = message.split()
361 cmd = message[0].lower()
362 notunderstood = False
363 if cmd == u"help":
364 op,overop=auteur in self.ops, auteur in self.overops
365 if len(message)==1:
366 helpmsg = config.helpmsg_default
367 if op:
368 helpmsg += config.helpmsg_ops
369 if overop:
370 helpmsg += config.helpmsg_overops
371 else:
372 helpmsgs = config.helpdico.get(message[1].lower(), ["Commande inconnue.", None, None])
373 helpmsg = helpmsgs[0]
374 if op and helpmsgs[1]:
375 if helpmsg:
376 helpmsg += "\n" + helpmsgs[1]
377 else:
378 helpmsg = helpmsgs[1]
379 if overop and helpmsgs[2]:
380 if helpmsg:
381 helpmsg += "\n" + helpmsgs[2]
382 else:
383 helpmsg = helpmsgs[2]
384 for ligne in helpmsg.split("\n"):
385 serv.privmsg(auteur, ligne.encode("utf-8"))
386 elif cmd == u"identify":
387 if len(message) == 1:
388 if self.identities.has_key(auteur):
389 serv.privmsg(auteur, "Je vous connais sous le pseudo note %s." % (
390 self.identities[auteur]["pseudo"].encode("utf8")))
391 else:
392 serv.privmsg(auteur, "Je ne connais pas votre pseudo note.")
393 elif len(message) >= 3:
394 username, password = message[1], " ".join(message[2:])
395 success, info, _ = self.new_connection_NK(serv, username, password)
396 if success:
397 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
398 serv.privmsg(auteur, "Identité enregistrée.")
399 self.identities[auteur] = info
400 json.dump(self.identities, open(config.identities_file,"w"))
401 else:
402 log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
403 serv.privmsg(auteur, "Mot de passe invalide. (ou serveur down)")
404 else:
405 serv.privmsg(auteur, "Syntaxe : IDENTIFY [<username> <password>]")
406 elif cmd == u"drop":
407 if len(message) > 1:
408 if self.identities.has_key(auteur):
409 password = " ".join(message[1:])
410 success, _, _ = self.new_connection_NK(serv, self.identities[auteur], password)
411 if success:
412 del self.identities[auteur]
413 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
414 json.dump(self.identities, open(config.identities_file, "w"))
415 serv.privmsg(auteur, "Identité oubliée.")
416 else:
417 log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
418 serv.privmsg(auteur, "Mot de passe invalide. (ou serveur down)")
419 else:
420 serv.privmsg(auteur, "Je ne connais pas ton pseudo note.")
421 else:
422 serv.privmsg(auteur, "Syntaxe : DROP <password>")
423 elif cmd == u"join":
424 if auteur in self.ops:
425 if len(message) > 1:
426 if message[1] in self.chanlist:
427 serv.privmsg(auteur, "Je suis déjà sur %s" % (message[1]))
428 else:
429 serv.join(message[1])
430 self.chanlist.append(message[1])
431 serv.privmsg(auteur, "Channels : " + " ".join(self.chanlist))
432 log(self.serveur, "priv", auteur, " ".join(message))
433 else:
434 serv.privmsg(auteur, "Channels : " + " ".join(self.chanlist))
435 else:
436 notunderstood = True
437 elif cmd == u"leave":
438 if auteur in self.ops and len(message) > 1:
439 if message[1] in self.chanlist:
440 if not (message[1] in self.stay_channels) or auteur in self.overops:
441 self.quitter(message[1].encode("utf-8"), " ".join(message[2:]))
442 self.chanlist.remove(message[1])
443 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
444 else:
445 serv.privmsg(auteur, "Non, je reste !")
446 log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
447 else:
448 serv.privmsg(auteur, "Je ne suis pas sur %s" % (message[1]))
449 else:
450 notunderstood = True
451 elif cmd == u"stay":
452 if auteur in self.overops:
453 if len(message) > 1:
454 if message[1] in self.stay_channels:
455 log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
456 serv.privmsg(auteur, "Je stay déjà sur %s." % (message[1]))
457 else:
458 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
459 self.stay_channels.append(message[1])
460 serv.privmsg(auteur, "Stay channels : " + " ".join(self.stay_channels))
461 else:
462 serv.privmsg(auteur, "Stay channels : " + " ".join(self.stay_channels))
463 else:
464 notunderstood = True
465 elif cmd == u"nostay":
466 if auteur in self.overops:
467 if len(message) > 1:
468 if message[1] in self.stay_channels:
469 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
470 self.stay_channels.remove(message[1])
471 serv.privmsg(auteur, "Stay channels : " + " ".join(self.stay_channels))
472 else:
473 log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
474 serv.privmsg(auteur, "Je ne stay pas sur %s." % (message[1]))
475 else:
476 notunderstood = True
477 elif cmd == u"die":
478 if auteur in self.overops:
479 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
480 self.mourir()
481 else:
482 notunderstood = True
483 elif cmd == u"crash":
484 if auteur in self.overops:
485 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
486 self.crash("priv", auteur)
487 else:
488 notunderstood = True
489 elif cmd == u"reload":
490 if auteur in self.ops:
491 self.execute_something("reload", {"auteur" : auteur}, place=auteur, auteur=auteur)
492 else:
493 notunderstood = True
494 elif cmd == u"reconnect":
495 if auteur in self.ops:
496 try:
497 self.nk = self.new_connection_NK(serv, config.note_pseudo,
498 config.note_password, "special")[2]
499 except Exception as exc:
500 self.nk = None
501 log(self.serveur, 'Erreur dans on_pubmsg/"cmd in ["reconnect"]\n' + str(exc))
502 if self.nk != None:
503 serv.privmsg(auteur, "%s: done" % (auteur))
504 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
505 else:
506 serv.privmsg(auteur, "%s: failed" % (auteur))
507 log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
508 for report in self.report_bugs_to:
509 serv.privmsg(report, "Connection to NK2015 failed, invalid password ? Server dead ?")
510 else:
511 notunderstood = True
512 elif cmd == u"quiet":
513 if auteur in self.ops:
514 if len(message) > 1:
515 if message[1] in self.quiet_channels:
516 serv.privmsg(auteur, "Je me la ferme déjà sur %s" % (message[1]))
517 log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
518 else:
519 self.quiet_channels.append(message[1])
520 serv.privmsg(auteur, "Quiet channels : " + " ".join(self.quiet_channels))
521 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
522 else:
523 serv.privmsg(auteur, "Quiet channels : " + " ".join(self.quiet_channels))
524 else:
525 notunderstood = True
526 elif cmd == u"noquiet":
527 if auteur in self.ops:
528 if len(message) > 1:
529 if message[1] in self.quiet_channels:
530 self.quiet_channels.remove(message[1])
531 serv.privmsg(auteur, "Quiet channels : " + " ".join(self.quiet_channels))
532 log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
533 else:
534 serv.privmsg(auteur, "Je ne me la ferme pas sur %s." % (message[1]))
535 log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
536 else:
537 notunderstood = True
538 elif cmd == u"say":
539 if auteur in self.overops and len(message) > 2:
540 serv.privmsg(message[1], " ".join(message[2:]))
541 log(self.serveur, "priv", auteur, " ".join(message))
542 elif len(message) <= 2:
543 serv.privmsg(auteur, "Syntaxe : SAY <channel> <message>")
544 else:
545 notunderstood = True
546 elif cmd == u"do":
547 if auteur in self.overops and len(message) > 2:
548 serv.action(message[1], " ".join(message[2:]))
549 log(self.serveur, "priv", auteur, " ".join(message))
550 elif len(message) <= 2:
551 serv.privmsg(auteur, "Syntaxe : DO <channel> <action>")
552 else:
553 notunderstood = True
554 elif cmd == u"kick":
555 if auteur in self.overops and len(message) > 2:
556 serv.kick(message[1], message[2], " ".join(message[3:]))
557 log(self.serveur, "priv", auteur, " ".join(message))
558 elif len(message) <= 2:
559 serv.privmsg(auteur, "Syntaxe : KICK <channel> <pseudo> [<raison>]")
560 else:
561 notunderstood = True
562 elif cmd == u"lost":
563 if auteur in self.ops and len(message) > 1:
564 serv.privmsg(message[1], "J'ai perdu !")
565 log(self.serveur, "priv", auteur, " ".join(message))
566 elif len(message) <= 1:
567 serv.privmsg(auteur, "Syntaxe : LOST <channel>")
568 else:
569 notunderstood = True
570 elif cmd == u"solde":
571 if len(message) == 1:
572 if self.identities.has_key(auteur):
573 success, solde, pseudo = nk.get_solde(self.nk, self.identities[auteur]["idbde"], serv, auteur)
574 if success:
575 serv.privmsg(auteur, "%.2f (%s)" % (solde/100.0, pseudo.encode("utf8")))
576 log(self.serveur, "priv", auteur, " ".join(message) + ("[successful]" if success else "[failed]"))
577 else:
578 serv.privmsg(canal, "Je ne connais pas ton pseudo note.")
579 elif cmd == u"ops":
580 if auteur in self.overops:
581 serv.privmsg(auteur, " ".join(self.ops))
582 else:
583 notunderstood = True
584 elif cmd == u"overops":
585 if auteur in self.overops:
586 serv.privmsg(auteur, " ".join(self.overops))
587 else:
588 notunderstood = True
589 else:
590 notunderstood = True
591 if notunderstood:
592 serv.privmsg(auteur, "Je n'ai pas compris. Essayez HELP…")
593
594 def on_pubmsg(self, serv, ev):
595 """À la réception d'un message sur un channel."""
596 if ignore_event(serv, ev):
597 return
598 auteur = irclib.nm_to_n(ev.source())
599 canal = ev.target()
600 message = ev.arguments()[0]
601 try:
602 message = bot_unicode(message)
603 except UnicodeBotError:
604 if config.utf8_trigger and not canal in self.quiet_channels:
605 serv.privmsg(canal, (u"%s: %s"% ( auteur, random.choice(config.utf8_fail_answers))).encode("utf8"))
606 return
607 pour_moi, message = self.pourmoi(serv, message)
608 if pour_moi and message.split()!=[]:
609 cmd = message.split()[0].lower()
610 try:
611 args = " ".join(message.split()[1:])
612 except:
613 args = ""
614 if cmd in [u"meurs", u"die", u"crève"]:
615 if auteur in self.overops:
616 log(self.serveur, canal, auteur, message + "[successful]")
617 self.mourir()
618 else:
619 serv.privmsg(canal,(u"%s: %s"%(auteur, random.choice(config.quit_fail_messages))).encode("utf8"))
620 log(self.serveur, canal, auteur, message + "[failed]")
621 elif cmd == u"reload":
622 if auteur in self.ops:
623 self.execute_something("reload", {"auteur" : auteur}, place=canal, auteur=auteur)
624 elif cmd == u"crash":
625 if auteur in self.overops:
626 self.crash(auteur, message)
627 elif cmd in [u"part", u"leave", u"dégage", u"va-t-en", u"tut'tiresailleurs,c'estmesgalets"]:
628 if auteur in self.ops and (not (canal in self.stay_channels)
629 or auteur in self.overops):
630 self.quitter(canal)
631 log(self.serveur, canal, auteur, message + "[successful]")
632 if canal in self.chanlist:
633 self.chanlist.remove(canal)
634 else:
635 serv.privmsg(canal,(u"%s: %s" % (auteur, random.choice(config.leave_fail_messages))).encode("utf8"))
636 log(self.serveur, canal, auteur, message + "[failed]")
637
638 elif cmd == u"reconnect":
639 if auteur in self.ops:
640 try:
641 self.nk = self.new_connection_NK(serv, config.note_pseudo,
642 config.note_password, "special")[2]
643 except Exception as exc:
644 self.nk = None
645 log(self.serveur, 'Erreur dans on_pubmsg/"cmd in ["reconnect"]\n' + str(exc))
646 if self.nk != None:
647 serv.privmsg(canal, "%s: done" % (auteur))
648 log(self.serveur, canal, auteur, message + "[successful]")
649 else:
650 serv.privmsg(canal, "%s: failed" % (auteur))
651 log(self.serveur, canal, auteur, message + "[failed]")
652 for report in self.report_bugs_to:
653 serv.privmsg(report, "Connection to NK2015 failed, invalid password ? Server dead ?")
654 else:
655 serv.privmsg(canal, "%s: %s" % (auteur, random.choice(config.pas_programme_pour_tobeir).encode("utf8")))
656 log(self.serveur, canal, auteur, message + "[failed]")
657
658 elif cmd in [u"deviens", u"pseudo"]:
659 if auteur in self.ops:
660 become = args
661 serv.nick(become)
662 log(self.serveur, canal, auteur, message + "[successful]")
663
664 if cmd in [u"meur", u"meurt", u"meurre", u"meurres"] and not canal in self.quiet_channels:
665 serv.privmsg(canal, '%s: Mourir, impératif, 2ème personne du singulier : "meurs" (de rien)' % (auteur))
666 elif cmd in [u"ping"] and not canal in self.quiet_channels:
667 serv.privmsg(canal, "%s: pong" % (auteur))
668
669 elif cmd in [u"solde", u"!solde", u"!coca"] or cmd.startswith("!"):
670 if self.identities.has_key(auteur):
671 idbde = self.identities[auteur]["idbde"]
672 if cmd in [u"solde", u"!solde"]:
673 success, solde, pseudo = nk.get_solde(self.nk, self.identities[auteur]["idbde"], serv, canal)
674 if success:
675 serv.privmsg(canal, "%s: %s (%s)" % (auteur, float(solde)/100, pseudo.encode("utf8")))
676 elif cmd in [u"!coca"] or cmd.startswith("!"):
677 success = nk.consomme(self.nk, self.identities[auteur]["idbde"], message[1:], serv, canal)
678 log(self.serveur, canal, auteur, message + ("[successful]" if success else "[failed]"))
679 else:
680 serv.privmsg(canal, "%s: Je ne connais pas votre pseudo note." % (auteur))
681 log(self.serveur, canal, auteur, message + "[unknown]")
682 elif (re.match("!?(pain au chocolat|chocolatine)", message.lower())
683 and not canal in self.quiet_channels):
684 serv.action(canal, "sert un pain au chocolat à %s" % (auteur))
685 elif re.match("!?manzana",message.lower()) and not canal in self.quiet_channels:
686 if auteur in config.manzana:
687 serv.action(canal, "sert une bouteille de manzana à %s" % (auteur))
688 elif auteur in config.manzana_bis:
689 serv.action(canal, "sert un grand verre de jus de pomme à %s : tout le monde sait qu'il ne boit pas." % (auteur))
690 else:
691 serv.action(canal, "sert un verre de manzana à %s" % (auteur))
692 if is_insult(message) and not canal in self.quiet_channels:
693 if is_not_insult(message):
694 answer = random.choice(config.compliment_answers)
695 for ligne in answer.split("\n"):
696 serv.privmsg(canal, "%s: %s" % (auteur, ligne.encode("utf8")))
697 else:
698 answer = random.choice(config.insultes_answers)
699 for ligne in answer.split("\n"):
700 serv.privmsg(canal, "%s: %s" % (auteur, ligne.encode("utf8")))
701 elif is_compliment(message) and not canal in self.quiet_channels:
702 answer = random.choice(config.compliment_answers)
703 for ligne in answer.split("\n"):
704 serv.privmsg(canal, "%s: %s" % (auteur,ligne.encode("utf8")))
705 gros_match = is_gros(message)
706 if gros_match and not canal in self.quiet_channels:
707 taille = get_filesize()
708 answer = u"Mais non, je ne suis pas %s, %sKo tout au plus…" % (gros_match.groups()[0], taille)
709 serv.privmsg(canal, "%s: %s"%(auteur, answer.encode("utf8")))
710 if is_tesla(message) and not canal in self.quiet_channels:
711 l1, l2 = config.tesla_answers, config.tesla_actions
712 n1, n2 = len(l1), len(l2)
713 i = random.randrange(n1 + n2)
714 if i >= n1:
715 serv.action(canal, l2[i - n1].encode("utf8"))
716 else:
717 serv.privmsg(canal, "%s: %s" % (auteur, l1[i].encode("utf8")))
718 if is_tamere(message) and not canal in self.quiet_channels:
719 answer = random.choice(config.tamere_answers)
720 for ligne in answer.split("\n"):
721 serv.privmsg(canal, "%s: %s"%(auteur, ligne.encode("utf8")))
722 if is_tag(message) and not canal in self.quiet_channels:
723 if auteur in self.ops:
724 action = random.choice(config.tag_actions)
725 serv.action(canal, action.encode("utf8"))
726 self.quiet_channels.append(canal)
727 else:
728 answer = random.choice(config.tag_answers)
729 for ligne in answer.split("\n"):
730 serv.privmsg(canal, "%s: %s" % (auteur, ligne.encode("utf8")))
731 if is_merci(message):
732 answer = random.choice(config.merci_answers)
733 for ligne in answer.split("\n"):
734 serv.privmsg(canal, "%s: %s"%(auteur, ligne.encode("utf8")))
735 out = re.match(ur"^([A-Z[]|\\|[0-9]+|(¹|²|³|⁴|⁵|⁶|⁷|⁸|⁹|⁰)+)(?:| \?| !)$", message.upper())
736 if re.match("ma bite dans ton oreille", message) and not canal in self.quiet_channels:
737 serv.privmsg(canal, "%s: Seul un olasd peut imiter un olasd dans un de ses grands jours !" % (auteur))
738 if out and not canal in self.quiet_channels:
739 out = out.groups()[0]
740 try:
741 iout = int(out)
742 serv.privmsg(canal, "%s: %s !" % (auteur, iout + 1))
743 if iout == 2147483647:
744 serv.privmsg(canal, "%s: Ciel, un maxint ! Heureusement que je suis en python…" % (auteur))
745 return
746 if iout + 1 > 1000 and random.randrange(4) == 0:
747 serv.privmsg(canal, "%s: Vous savez, moi et les chiffres…" % (auteur))
748 return
749 except Exception as exc:
750 pass
751 if re.match("[A-Y]", out):
752 alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
753 serv.privmsg(canal, "%s: %s !"%(auteur, alphabet[alphabet.index(out) + 1]))
754 elif out == "Z":
755 serv.privmsg(canal, "%s: Je ne vous remercie pas, j'ai l'air idiot ainsi… [ ?" % (auteur))
756 elif out in "[\\":
757 serv.privmsg(canal, "%s: Nous devrions nous en tenir là, ça va finir par poser des problèmes…" % (auteur))
758 elif re.match(ur"(¹|²|³|⁴|⁵|⁶|⁷|⁸|⁹|⁰)+", out):
759 def translate(mess):
760 return "".join([{u"⁰¹²³⁴⁵⁶⁷⁸⁹0123456789"[i]:u"0123456789⁰¹²³⁴⁵⁶⁷⁸⁹"[i]
761 for i in range(20)}[j]
762 for j in mess])
763 out = int(translate(out))
764 serv.privmsg(canal,"%s: %s !" % (auteur, translate(str(out + 1)).encode("utf8")))
765 if is_bonjour(message) and not canal in self.quiet_channels:
766 if is_night():
767 answer = random.choice(config.night_answers)
768 elif is_day():
769 answer = random.choice(config.bonjour_answers)
770 else:
771 answer = random.choice(config.bonsoir_answers)
772 serv.privmsg(canal, answer.format(auteur).encode("utf8"))
773 if is_bonne_nuit(message) and not canal in self.quiet_channels:
774 answer = random.choice(config.bonne_nuit_answers)
775 serv.privmsg(canal, answer.format(auteur).encode("utf8"))
776 if is_pan(message) and not canal in self.quiet_channels:
777 serv.privmsg(canal, "%s: ce n'est pas sur moi qu'il faut tirer, même si je sais que j'attire l'œil !" % (auteur))
778 else:
779 if message in [u"!pain au chocolat", u"!chocolatine"] and not canal in self.quiet_channels:
780 serv.action(canal, "sert un pain au chocolat à %s" % (auteur))
781 if message in [u"!manzana"] and not canal in self.quiet_channels:
782 if auteur in config.manzana:
783 serv.action(canal, "sert une bouteille de manzana à %s" % (auteur))
784 elif auteur in config.manzana_bis:
785 serv.action(canal, "sert un grand verre de jus de pomme à %s : tout le monde sait qu'il ne boit pas." % (auteur))
786 else:
787 serv.action(canal, "sert un verre de manzana à %s" % (auteur))
788 if re.match(u'^ *(.|§|!|/|/|:|)(w|b) [0-9]+$', message) and not canal in self.quiet_channels:
789 failanswers = config.buffer_fail_answers
790 answer = random.choice(failanswers)
791 serv.privmsg(canal, ("%s: %s"%(auteur,answer)).encode("utf8"))
792 if not canal in self.quiet_channels:
793 mypseudo = self.nick
794 if re.match((u"^(" + u"|".join(config.bonjour_triggers)
795 + ur")( {}| all| tout le monde| (à )?tous)(\.| ?!)?$"
796 ).format(mypseudo).lower(), message.strip().lower()):
797 answer = random.choice(config.bonjour_answers)
798 serv.privmsg(canal, answer.format(auteur).encode("utf8"))
799 if (is_perdu(message) and not canal in self.quiet_channels):
800 # proba de perdre sur trigger :
801 # avant 30min (enfin, config) : 0
802 # ensuite, +25%/30min, linéairement
803 deltat = time.time() - self.last_perdu
804 barre = (deltat - config.time_between_perdu)/(2*3600.0)
805 if random.uniform(0, 1) < barre:
806 serv.privmsg(canal, "%s: J'ai perdu !" % (auteur))
807 self.last_perdu = time.time()
808
809 def on_action(self, serv, ev):
810 """À la réception d'une action."""
811 if ignore_event(serv, ev):
812 return
813 action = ev.arguments()[0]
814 auteur = irclib.nm_to_n(ev.source())
815 channel = ev.target()
816 try:
817 action = bot_unicode(action)
818 except UnicodeBotError:
819 if config.utf8_trigger and not channel in self.quiet_channels:
820 serv.privmsg(channel, (u"%s: %s"%(auteur,random.choice(config.utf8_fail_answers))).encode("utf8"))
821 return
822 mypseudo = self.nick
823
824 if is_bad_action_trigger(action, mypseudo) and not channel in self.quiet_channels:
825 l1, l2 = config.bad_action_answers, config.bad_action_actions
826 n1, n2 = len(l1), len(l2)
827 i = random.randrange(n1 + n2)
828 if i >= n1:
829 serv.action(channel, l2[i - n1].format(auteur).encode("utf8"))
830 else:
831 serv.privmsg(channel, l1[i].format(auteur).encode("utf8"))
832 if is_good_action_trigger(action, mypseudo) and not channel in self.quiet_channels:
833 l1, l2 = config.good_action_answers, config.good_action_actions
834 n1, n2 = len(l1), len(l2)
835 i = random.randrange(n1 + n2)
836 if i >= n1:
837 serv.action(channel, l2[i-n1].format(auteur).encode("utf8"))
838 else:
839 serv.privmsg(channel, l1[i].format(auteur).encode("utf8"))
840
841 def on_kick(self, serv, ev):
842 """À la réception d'une action."""
843 auteur = irclib.nm_to_n(ev.source())
844 channel = ev.target()
845 victime = ev.arguments()[0]
846 raison = ev.arguments()[1]
847 if victime == self.nick:
848 log(self.serveur, "%s kické de %s par %s (raison : %s)" % (victime, channel, auteur, raison))
849 time.sleep(2)
850 serv.join(channel)
851 l1, l2 = config.kick_answers, config.kick_actions
852 n1, n2 = len(l1), len(l2)
853 i = random.randrange(n1 + n2)
854 if i >= n1:
855 serv.action(channel, l2[i-n1].format(auteur).encode("utf8"))
856 else:
857 serv.privmsg(channel, l1[i].format(auteur).encode("utf8"))
858
859 ### .fork trick
860 def start_as_daemon(self, outfile):
861 sys.stderr = Logger(outfile)
862 self.start()
863
864
865 class Logger(object):
866 """Pour écrire ailleurs que sur stdout"""
867 def __init__(self, filename="basile.full.log"):
868 self.filename = filename
869
870 def write(self, message):
871 f = open(self.filename, "a")
872 f.write(message)
873 f.close()
874
875 def main():
876 """Exécution principal : lecture des paramètres et lancement du bot."""
877 if len(sys.argv) == 1:
878 print "Usage : basile.py <serveur> [--debug] [--no-output] [--daemon [--pidfile]] [--outfile]"
879 print " --outfile sans --no-output ni --daemon n'a aucun effet"
880 exit(1)
881 serveur = sys.argv[1]
882 if "--daemon" in sys.argv:
883 thisfile = os.path.realpath(__file__)
884 thisdirectory = thisfile.rsplit("/", 1)[0]
885 os.chdir(thisdirectory)
886 daemon = True
887 else:
888 daemon = False
889 if "debug" in sys.argv or "--debug" in sys.argv:
890 debug = True
891 else:
892 debug = False
893 if "--quiet" in sys.argv:
894 config.debug_stdout = False
895 serveurs = {"a♡" : "acoeur.crans.org",
896 "acoeur" : "acoeur.crans.org",
897 "acoeur.crans.org" : "acoeur.crans.org",
898 "irc" : "irc.crans.org",
899 "crans" : "irc.crans.org",
900 "irc.crans.org" : "irc.crans.org"}
901 if "--no-output" in sys.argv or "--daemon" in sys.argv:
902 outfile = "/var/log/bots/basile.full.log"
903 for arg in sys.argv:
904 arg = arg.split("=")
905 if arg[0].strip('-') in ["out", "outfile", "logfile"]:
906 outfile = arg[1]
907 sys.stdout = Logger(outfile)
908 try:
909 serveur = serveurs[serveur]
910 except KeyError:
911 print "Server Unknown : %s" % (serveur)
912 exit(404)
913 basile = Basile(serveur,debug)
914 # Si on reçoit un SIGHUP, on reload la config
915 def sighup_handler(signum, frame):
916 basile.execute_reload(auteur="SIGHUP")
917 signal.signal(signal.SIGHUP, sighup_handler)
918 # Daemonization
919 if daemon:
920 child_pid = os.fork()
921 if child_pid == 0:
922 os.setsid()
923 basile.start_as_daemon(outfile)
924 else:
925 # on enregistre le pid de basile
926 pidfile = "/var/run/bots/basile.pid"
927 for arg in sys.argv:
928 arg = arg.split("=")
929 if arg[0].strip('-') in ["pidfile"]:
930 pidfile = arg[1]
931 f = open(pidfile, "w")
932 f.write("%s\n" % child_pid)
933 f.close()
934 else:
935 basile.start()
936
937 if __name__ == "__main__":
938 main()