basile.py

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