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