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