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