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