helixbot.py

   1 #!/usr/bin/python
   2 # -*- encoding: utf-8 -*-
   3
   4 """ Codé par 20-100
   5
   6 Un bot IRC qui donne les réponses de Helix the fossil.
   7 """
   8
   9 import irclib
  10 import ircbot
  11 import random
  12 import time
  13 import re
  14 from commands import getstatusoutput as ex
  15
  16 # on récupère la config
  17 import config
  18
  19
  20 def get_config_logfile(serveur):
  21     """Renvoie le nom du fichier de log en fonction du serveur."""
  22     serveurs = {"acoeur.crans.org" : "acoeur", "irc.crans.org" : "crans"}
  23     return config.logfile_template % (serveurs[serveur])
  24
  25 def log(serveur, channel, auteur=None, message=None):
  26     f = open(get_config_logfile(serveur), "a")
  27     if auteur == message == None:
  28         # alors c'est que c'est pas un channel mais juste une ligne de log
  29         chain = "%s %s" % (time.strftime("%F %T"), channel)
  30     else:
  31         chain = "%s [%s:%s] %s" % (time.strftime("%F %T"), channel, auteur, message)
  32     f.write(chain+"\n")
  33     if config.debug_stdout:
  34         print chain
  35     f.close()
  36
  37 def is_something(chain, matches, avant=u".*(?:^| )", apres=u"(?:$|\.| |,|;).*", case_sensitive=False, debug=False):
  38     if case_sensitive:
  39         chain = unicode(chain, "utf8")
  40     else:
  41         chain = unicode(chain, "utf8").lower()
  42     allmatches = "("+"|".join(matches)+")"
  43     reg = (avant+allmatches+apres).lower()
  44     o = re.match(reg, chain)
  45     return o
  46
  47 def is_tag(chain):
  48     return is_something(chain, config.tag_triggers)
  49 def is_tesla(chain):
  50     return is_something(chain, config.tesla_triggers, avant=u"^", apres=u"$", debug=True)
  51
  52
  53 class UnicodeBotError(Exception):
  54     pass
  55 def bot_unicode(chain):
  56     try:
  57         unicode(chain, "utf8")
  58     except UnicodeDecodeError as exc:
  59         raise UnicodeBotError
  60
  61 class HelixBot(ircbot.SingleServerIRCBot):
  62     def __init__(self, serveur, debug=False):
  63         temporary_pseudo = config.irc_pseudo + str(random.randrange(10000, 100000))
  64         ircbot.SingleServerIRCBot.__init__(self, [(serveur, 6667)],
  65                               temporary_pseudo, config.ircname, 10)
  66         self.debug = debug
  67         self.serveur = serveur
  68         self.overops = config.overops
  69         self.ops = self.overops+config.ops
  70         self.chanlist = config.chanlist
  71         self.stay_channels = config.stay_channels
  72         self.quiet_channels = config.quiet_channels
  73
  74     def give_me_my_pseudo(self, serv):
  75         serv.privmsg("NickServ", "RECOVER %s %s" % (config.irc_pseudo, config.irc_password))
  76         serv.privmsg("NickServ", "RELEASE %s %s" % (config.irc_pseudo, config.irc_password))
  77         time.sleep(0.3)
  78         serv.nick(config.irc_pseudo)
  79
  80     def on_welcome(self, serv, ev):
  81         self.serv = serv # ça serv ira :)
  82         self.give_me_my_pseudo(serv)
  83         serv.privmsg("NickServ", "identify %s" % (config.irc_password))
  84         log(self.serveur, "Connected")
  85         if self.debug:
  86             self.chanlist = ["#bot"]
  87         for c in self.chanlist:
  88             log(self.serveur, "JOIN %s" % (c))
  89             serv.join(c)
  90
  91     def pourmoi(self, serv, message):
  92         """renvoie (False, lemessage) ou (True, le message amputé de "pseudo: ")"""
  93         pseudo = self.nick
  94         size = len(pseudo)
  95         if message[:size]  == pseudo and len(message)>size and message[size] == ":":
  96             return (True, message[size+1:].lstrip(" "))
  97         else:
  98             return (False, message)
  99
 100     def on_privmsg(self, serv, ev):
 101         message = ev.arguments()[0]
 102         auteur = irclib.nm_to_n(ev.source())
 103         try:
 104             bot_unicode(message)
 105         except UnicodeBotError:
 106             if config.utf8_trigger:
 107                 serv.privmsg(auteur, random.choice(config.utf8_fail_answers).encode("utf8"))
 108             return
 109         message = message.split()
 110         cmd = message[0].lower()
 111         notunderstood = False
 112         if cmd == "help":
 113             helpdico = {"help":["""HELP <commande>
 114  Affiche de l'aide sur la commande""", None, None],
 115  "join": [None, """JOIN <channel>
 116  Me fait rejoindre le channel""", None],
 117  "leave": [None, """LEAVE <channel>
 118  Me fait quitter le channel (sauf s'il est dans ma stay_list).""", None],
 119  "quiet": [None, """QUIET <channel>
 120  Me rend silencieux sur le channel.""", None],
 121  "noquiet": [None, """NOQUIET <channel>
 122  Me rend la parole sur le channel.""", None],
 123  "say": [None, None, """SAY <channel> <message>
 124  Me fait parler sur le channel."""],
 125  "do": [None, None, """DO <channel> <action>
 126  Me fait faitre une action (/me) sur le channel."""],
 127  "stay": [None, None, """STAY <channel>
 128  Ajoute le channel à ma stay_list."""],
 129  "nostay": [None, None, """NOSTAY <channel>
 130  Retire le channel de ma stay_list."""],
 131  "ops": [None, None, """OPS
 132  Affiche la liste des ops."""],
 133  "overops": [None, None, """OVEROPS
 134  Affiche la liste des overops."""],
 135  "kick": [None, None, """KICK <channel> <pseudo> [<raison>]
 136  Kicke <pseudo> du channel (Il faut bien entendu que j'y sois op)."""],
 137  "die": [None, None, """DIE
 138  Me déconnecte du serveur IRC."""]
 139  }
 140             helpmsg_default = "Liste des commandes disponibles :\nHELP"
 141             helpmsg_ops = " JOIN LEAVE QUIET NOQUIET RECONNECT"
 142             helpmsg_overops = " SAY DO STAY NOSTAY OPS OVEROPS KICK DIE"
 143             op, overop = auteur in self.ops, auteur in self.overops
 144             if len(message) == 1:
 145                 helpmsg = helpmsg_default
 146                 if op:
 147                     helpmsg += helpmsg_ops
 148                 if overop:
 149                     helpmsg += helpmsg_overops
 150             else:
 151                 helpmsgs = helpdico.get(message[1].lower(), ["Commande inconnue.", None, None])
 152                 helpmsg = helpmsgs[0]
 153                 if op and helpmsgs[1]:
 154                     if helpmsg:
 155                         helpmsg += "\n"+helpmsgs[1]
 156                     else:
 157                         helpmsg = helpmsgs[1]
 158                 if overop and helpmsgs[2]:
 159                     if helpmsg:
 160                         helpmsg += "\n"+helpmsgs[2]
 161                     else:
 162                         helpmsg = helpmsgs[2]
 163             for ligne in helpmsg.split("\n"):
 164                 serv.privmsg(auteur, ligne)
 165         elif cmd == "join":
 166             if auteur in self.ops:
 167                 if len(message)>1:
 168                     if message[1] in self.chanlist:
 169                         serv.privmsg(auteur, "Je suis déjà sur %s" % (message[1]))
 170                     else:
 171                         serv.join(message[1])
 172                         self.chanlist.append(message[1])
 173                         serv.privmsg(auteur, "Channels : "+" ".join(self.chanlist))
 174                         log(self.serveur, "priv", auteur, " ".join(message))
 175                 else:
 176                     serv.privmsg(auteur, "Channels : "+" ".join(self.chanlist))
 177             else:
 178                 notunderstood = True
 179         elif cmd == "leave":
 180             if auteur in self.ops and len(message)>1:
 181                 if message[1] in self.chanlist:
 182                     if not (message[1] in self.stay_channels) or auteur in self.overops:
 183                         self.quitter(message[1], " ".join(message[2:]))
 184                         self.chanlist.remove(message[1])
 185                         log(self.serveur, "priv", auteur, " ".join(message)+"[successful]")
 186                     else:
 187                         serv.privmsg(auteur, "Non, je reste !")
 188                         log(self.serveur, "priv", auteur, " ".join(message)+"[failed]")
 189                 else:
 190                     serv.privmsg(auteur, "Je ne suis pas sur %s" % (message[1]))
 191             else:
 192                 notunderstood = True
 193         elif cmd == "stay":
 194             if auteur in self.overops:
 195                 if len(message)>1:
 196                     if message[1] in self.stay_channels:
 197                         log(self.serveur, "priv", auteur, " ".join(message)+"[failed]")
 198                         serv.privmsg(auteur, "Je stay déjà sur %s." % (message[1]))
 199                     else:
 200                         log(self.serveur, "priv", auteur, " ".join(message)+"[successful]")
 201                         self.stay_channels.append(message[1])
 202                         serv.privmsg(auteur, "Stay channels : "+" ".join(self.stay_channels))
 203                 else:
 204                     serv.privmsg(auteur, "Stay channels : "+" ".join(self.stay_channels))
 205             else:
 206                 notunderstood = True
 207         elif cmd == "nostay":
 208             if auteur in self.overops:
 209                 if len(message)>1:
 210                     if message[1] in self.stay_channels:
 211                         log(self.serveur, "priv", auteur, " ".join(message)+"[successful]")
 212                         self.stay_channels.remove(message[1])
 213                         serv.privmsg(auteur, "Stay channels : "+" ".join(self.stay_channels))
 214                     else:
 215                         log(self.serveur, "priv", auteur, " ".join(message)+"[failed]")
 216                         serv.privmsg(auteur, "Je ne stay pas sur %s." % (message[1]))
 217
 218             else:
 219                 notunderstood = True
 220         elif cmd == "die":
 221             if auteur in self.overops:
 222                 log(self.serveur, "priv", auteur, " ".join(message)+"[successful]")
 223                 self.mourir()
 224             else:
 225                 notunderstood = True
 226         elif cmd == "quiet":
 227             if auteur in self.ops:
 228                 if len(message)>1:
 229                     if message[1] in self.quiet_channels:
 230                         serv.privmsg(auteur, "Je me la ferme déjà sur %s" % (message[1]))
 231                         log(self.serveur, "priv", auteur, " ".join(message)+"[failed]")
 232                     else:
 233                         self.quiet_channels.append(message[1])
 234                         serv.privmsg(auteur, "Quiet channels : "+" ".join(self.quiet_channels))
 235                         log(self.serveur, "priv", auteur, " ".join(message)+"[successful]")
 236                 else:
 237                     serv.privmsg(auteur, "Quiet channels : "+" ".join(self.quiet_channels))
 238             else:
 239                 notunderstood = True
 240         elif cmd == "noquiet":
 241             if auteur in self.ops:
 242                 if len(message)>1:
 243                     if message[1] in self.quiet_channels:
 244                         self.quiet_channels.remove(message[1])
 245                         serv.privmsg(auteur, "Quiet channels : "+" ".join(self.quiet_channels))
 246                         log(self.serveur, "priv", auteur, " ".join(message)+"[successful]")
 247                     else:
 248                         serv.privmsg(auteur, "Je ne me la ferme pas sur %s." % (message[1]))
 249                         log(self.serveur, "priv", auteur, " ".join(message)+"[failed]")
 250             else:
 251                 notunderstood = True
 252         elif cmd == "say":
 253             if auteur in self.overops and len(message)>2:
 254                 serv.privmsg(message[1], " ".join(message[2:]))
 255                 log(self.serveur, "priv", auteur, " ".join(message))
 256             elif len(message) <= 2:
 257                 serv.privmsg(auteur, "Syntaxe : SAY <channel> <message>")
 258             else:
 259                 notunderstood = True
 260         elif cmd == "do":
 261             if auteur in self.overops and len(message)>2:
 262                 serv.action(message[1], " ".join(message[2:]))
 263                 log(self.serveur, "priv", auteur, " ".join(message))
 264             elif len(message) <= 2:
 265                 serv.privmsg(auteur, "Syntaxe : DO <channel> <action>")
 266             else:
 267                 notunderstood = True
 268         elif cmd == "kick":
 269             if auteur in self.overops and len(message)>2:
 270                 serv.kick(message[1], message[2], " ".join(message[3:]))
 271                 log(self.serveur, "priv", auteur, " ".join(message))
 272             elif len(message) <= 2:
 273                 serv.privmsg(auteur, "Syntaxe : KICK <channel> <pseudo> [<raison>]")
 274             else:
 275                 notunderstood = True
 276         elif cmd == "ops":
 277             if auteur in self.overops:
 278                 serv.privmsg(auteur, " ".join(self.ops))
 279             else:
 280                 notunderstood = True
 281         elif cmd == "overops":
 282             if auteur in self.overops:
 283                 serv.privmsg(auteur, " ".join(self.overops))
 284             else:
 285                 notunderstood = True
 286         else:
 287             notunderstood = True
 288         if notunderstood:
 289             serv.privmsg(auteur, "Je n'ai pas compris. Essayez HELP…")
 290
 291     def on_pubmsg(self, serv, ev):
 292         auteur = irclib.nm_to_n(ev.source())
 293         canal = ev.target()
 294         message = ev.arguments()[0]
 295         try:
 296             bot_unicode(message)
 297         except UnicodeBotError:
 298             if config.utf8_trigger and not canal in self.quiet_channels:
 299                 serv.privmsg(canal, (u"%s: %s" % (auteur, random.choice(config.utf8_fail_answers))).encode("utf8"))
 300             return
 301         pour_moi, message = self.pourmoi(serv, message)
 302         if pour_moi and message.split() != []:
 303             cmd = message.split()[0].lower()
 304             try:
 305                 args = " ".join(message.split()[1:])
 306             except:
 307                 args = ""
 308             if cmd in ["meurs", "die", "crève"]:
 309                 if auteur in self.overops:
 310                     log(self.serveur, canal, auteur, message+"[successful]")
 311                     self.mourir()
 312                 else:
 313                     serv.privmsg(canal, ("%s: %s" % (auteur, random.choice(config.quit_fail_messages))).encode("utf8"))
 314                     log(self.serveur, canal, auteur, message+"[failed]")
 315
 316             elif cmd in ["part", "leave", "dégage", "va-t-en", "tut'tiresailleurs, c'estmesgalets"]:
 317                 if auteur in self.ops and (not (canal in self.stay_channels)
 318                                            or auteur in self.overops):
 319                     self.quitter(canal)
 320                     log(self.serveur, canal, auteur, message+"[successful]")
 321                     if canal in self.chanlist:
 322                         self.chanlist.remove(canal)
 323                 else:
 324                     serv.privmsg(canal, ("%s: %s" % (auteur, random.choice(config.leave_fail_messages))).encode("utf8"))
 325                     log(self.serveur, canal, auteur, message+"[failed]")
 326
 327             elif cmd in ["deviens", "pseudo"]:
 328                 if auteur in self.ops:
 329                     become = args
 330                     serv.nick(become)
 331                     log(self.serveur, canal, auteur, message+"[successful]")
 332
 333             elif cmd in ["ping"] and not canal in self.quiet_channels:
 334                 serv.privmsg(canal, "%s: pong" % (auteur))
 335             if is_tag(message) and not canal in self.quiet_channels:
 336                 if auteur in self.ops:
 337                     action = random.choice(config.tag_actions)
 338                     serv.action(canal, action.encode("utf8"))
 339                     self.quiet_channels.append(canal)
 340                 else:
 341                     answer = random.choice(config.tag_answers)
 342                     for ligne in answer.split("\n"):
 343                         serv.privmsg(canal, "%s: %s" % (auteur, ligne.encode("utf8")))
 344             elif (any([re.match(trig, message) for trig in config.fossil_triggers]) and
 345                   not canal in self.quiet_channels):
 346                 answer = random.choice(config.fossil_answers)
 347                 serv.privmsg(canal, "%s: %s" % (auteur, answer))
 348         else:
 349             pass
 350
 351     def on_action(self, serv, ev):
 352         action = ev.arguments()[0]
 353         auteur = irclib.nm_to_n(ev.source())
 354         channel = ev.target()
 355         try:
 356             bot_unicode(action)
 357         except UnicodeBotError:
 358             if config.utf8_trigger and not channel in self.quiet_channels:
 359                 serv.privmsg(channel, (u"%s: %s" % (auteur, random.choice(config.utf8_fail_answers))).encode("utf8"))
 360             return
 361         mypseudo = self.nick
 362
 363
 364     def on_kick(self, serv, ev):
 365         auteur = irclib.nm_to_n(ev.source())
 366         channel = ev.target()
 367         victime = ev.arguments()[0]
 368         raison = ev.arguments()[1]
 369         if victime == self.nick:
 370             log(self.serveur, "%s kické de %s par %s (raison : %s)" % (victime, channel, auteur, raison))
 371             time.sleep(2)
 372             serv.join(channel)
 373             return # pas de réaction verbale au kick
 374             l1, l2 = config.kick_answers, config.kick_actions
 375             n1, n2 = len(l1), len(l2)
 376             i = random.randrange(n1+n2)
 377             if i >= n1:
 378                 serv.action(channel, l2[i-n1].format(auteur).encode("utf8"))
 379             else:
 380                 serv.privmsg(channel, l1[i].format(auteur).encode("utf8"))
 381
 382     def quitter(self, chan, leave_message=None):
 383         if leave_message == None:
 384             leave_message = random.choice(config.leave_messages)
 385         self.serv.part(chan, message=leave_message.encode("utf8"))
 386
 387     def mourir(self):
 388         quit_message = random.choice(config.quit_messages)
 389         self.die(msg=quit_message.encode("utf8"))
 390
 391     def _getnick(self):
 392         return self.serv.get_nickname()
 393     nick = property(_getnick)
 394
 395
 396 if __name__ == "__main__":
 397     import sys
 398     if len(sys.argv) == 1:
 399         print "Usage : helixbot.py <serveur> [--debug]"
 400         exit(1)
 401     serveur = sys.argv[1]
 402     if "debug" in sys.argv or "--debug" in sys.argv:
 403         debug = True
 404     else:
 405         debug = False
 406     if "--quiet" in sys.argv:
 407         config.debug_stdout = False
 408     serveurs = {"a♡":"acoeur.crans.org", "acoeur":"acoeur.crans.org", "acoeur.crans.org":"acoeur.crans.org",
 409               "irc":"irc.crans.org", "crans":"irc.crans.org", "irc.crans.org":"irc.crans.org"}
 410     try:
 411         serveur = serveurs[serveur]
 412     except KeyError:
 413         print "Server Unknown : %s" % (serveur)
 414         exit(404)
 415     bot = HelixBot(serveur, debug)
 416     bot.start()