bbc.py

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