import config
#: Module définissant les erreurs
import errors
+#: Module de gestion des quotes
+import quotes
+# Je veux pouvoir éditer ce que crée ce bot
+os.umask(002)
def get_config_logfile(serveur):
"""Renvoie le nom du fichier de log en fonction du ``serveur`` et de la config."""
"""Essaye de décoder ``chain`` en UTF-8.
Lève une py:class:`errors.UnicodeBotError` en cas d'échec."""
try:
- return chain.decode("utf8")
+ return chain.decode("utf-8")
except UnicodeDecodeError as exc:
raise errors.UnicodeBotError
self.stay_channels = config.stay_channels
self.quiet_channels = config.quiet_channels
self.last_perdu = 0
+
+ self.quotedb = quotes.QuoteDB()
+ self.reload_quotes()
+
### Utilitaires
def _getnick(self):
def give_me_my_pseudo(self, serv):
"""Récupère le pseudo auprès de NickServ."""
serv.privmsg("NickServ", "RECOVER %s %s" % (config.irc_pseudo, config.irc_password))
- serv.privmsg("NickServ", "RELEASE %s %s" % (config.irc_pseudo, config.irc_password))
time.sleep(0.3)
serv.nick(config.irc_pseudo)
"""Quitter un channel avec un message customisable."""
if leave_message == None:
leave_message = random.choice(config.leave_messages)
- self.serv.part(chan, message=leave_message.encode("utf8"))
+ self.serv.part(chan, message=leave_message.encode("utf-8"))
def mourir(self):
"""Se déconnecter du serveur IRC avec un message customisable."""
quit_message = random.choice(config.quit_messages)
- self.die(msg=quit_message.encode("utf8"))
+ self.die(msg=quit_message.encode("utf-8"))
+ def reload_quotes(self):
+ """ Recharge la base de données des quotes et recompile la regexp de quote """
+ self.quotedb.load()
+ self.quote_pattern = re.compile(config.quote_regexp, flags=re.UNICODE)
+
def execute_reload(self, auteur=None):
"""Recharge la config."""
reload(config)
- isit.regexp_compile()
+ self.reload_quotes()
if auteur in [None, "SIGHUP"]:
towrite = "Config reloaded" + " (SIGHUP received)" * (auteur == "SIGHUP")
for to in config.report_bugs_to:
self.serv.privmsg(place, message)
log(self.serveur, place, auteur, something + "%r" % params + ("[successful]" if success else "[failed]"))
- ### Gestion des quotes
+ def acknowledge(self, asked_by, asked_where, message):
+ """Répond quelque chose au demandeur d'une action.
+ ``asked_where=None`` signifie en privé."""
+ if asked_where is None:
+ self.serv.privmsg(asked_by, message)
+ else:
+ self.serv.privmsg(asked_where, "%s: %s" % (asked_by, message))
+
+ def dump(self, asked_by, asked_where=None):
+ """Dumpe les quotes. ``asked_where=None`` signifie en privé."""
+ quotes.dump(self.quotedb)
+ self.acknowledge(asked_by, asked_where, "Quotes dumpées")
+
+ def restore(self, asked_by, asked_where=None):
+ """Restaure les quotes à partir du dump. ``asked_where=None`` signifie en privé."""
+ self.quotedb = quotes.restore()
+ self.acknowledge(asked_by, asked_where, "Quotes restaurées à partir du dump (pas de backup effectué).")
+ many = self.quotedb.get_clash_authors()
+ if many:
+ self.acknowledge(asked_by, asked_where, "Auteurs de casse différente : %s" % (many))
### Surcharge des events du Bot
def on_welcome(self, serv, ev):
message = bot_unicode(message)
except errors.UnicodeBotError:
if config.utf8_trigger:
- serv.privmsg(auteur, random.choice(config.utf8_fail_answers).encode("utf8"))
+ serv.privmsg(auteur, random.choice(config.utf8_fail_answers).encode("utf-8"))
return
message = message.split()
cmd = message[0].lower()
serv.privmsg(auteur, " ".join(self.overops))
else:
notunderstood = True
+ elif cmd == u"dump" and auteur in self.ops:
+ self.dump(asked_by=auteur)
+ elif cmd == u"restore" and auteur in self.overops:
+ self.restore(asked_by=auteur)
else:
notunderstood = True
if notunderstood:
message = bot_unicode(message)
except errors.UnicodeBotError:
if config.utf8_trigger and not canal in self.quiet_channels:
- serv.privmsg(canal, (u"%s: %s"% ( auteur, random.choice(config.utf8_fail_answers))).encode("utf8"))
+ serv.privmsg(canal, (u"%s: %s"% ( auteur, random.choice(config.utf8_fail_answers))).encode("utf-8"))
return
pour_moi, message = self.pourmoi(serv, message)
if pour_moi and message.split()!=[]:
log(self.serveur, canal, auteur, message + "[successful]")
self.mourir()
else:
- serv.privmsg(canal,(u"%s: %s"%(auteur, random.choice(config.quit_fail_messages))).encode("utf8"))
+ serv.privmsg(canal,(u"%s: %s"%(auteur, random.choice(config.quit_fail_messages))).encode("utf-8"))
log(self.serveur, canal, auteur, message + "[failed]")
elif cmd == u"reload":
if auteur in self.ops:
if canal in self.chanlist:
self.chanlist.remove(canal)
else:
- serv.privmsg(canal,(u"%s: %s" % (auteur, random.choice(config.leave_fail_messages))).encode("utf8"))
+ serv.privmsg(canal,(u"%s: %s" % (auteur, random.choice(config.leave_fail_messages))).encode("utf-8"))
log(self.serveur, canal, auteur, message + "[failed]")
elif cmd in [u"ping"] and not canal in self.quiet_channels:
serv.privmsg(canal, "%s: pong" % (auteur))
+ elif cmd in [u"dump"]:
+ self.dump(asked_by=auteur, asked_where=canal)
+ elif cmd in [u"restore"] and auteur in self.overops:
+ self.restore(asked_by=auteur, asked_where=canal)
+ elif cmd in [u"display", u"link", u"url"]:
+ self.serv.privmsg(canal, "%s: %s" % (auteur, config.quote_display_url.encode("utf-8")))
else:
# Vu que ce bot est prévu pour parser des quotes il va falloir bosser ici
- pass
+ match = self.quote_pattern.match(message)
+ if match:
+ d = match.groupdict()
+ # On n'autorise pas les gens à déclarer le quoter
+ d["quoter"] = auteur.decode("utf-8")
+ if self.quotedb.store(**d):
+ serv.privmsg(canal, (u"%s: Ce sera retenu, répété, amplifié" % (auteur,)).encode("utf-8"))
+ self.quotedb.save()
+ else:
+ serv.privmsg(canal, (u"%s: Je le savais déjà." % (auteur,)).encode("utf-8"))
+ # Whou, attention, hack dégueu
+ # on enlève context- au début des !commands si il y est,
+ # et on passe à True le paramètre show_context pour s'en souvenir
+ show_context = False
+ if message.startswith(u"!context-"):
+ show_context = True
+ message = u"!" + message[9:]
+ if message.startswith(u"!quote"):
+ if message.strip() == u"!quote":
+ q = self.quotedb.random()
+ serv.privmsg(canal, q.display(show_context))
+ elif message.startswith("!quote "):
+ author = message[7:].strip()
+ try:
+ q = self.quotedb.randomfrom(author)
+ except IndexError:
+ serv.privmsg(canal, (u"Pas de quote de %s en mémoire." % author).encode("utf-8"))
+ return
+ serv.privmsg(canal, q.display(show_context))
+ elif message.startswith(u"!author") or message.startswith(u"!from"):
+ words = message.split()
+ cmd = words[0].lstrip("!")
+ regexp = any([cmd.endswith(suffix) for suffix in config.regex_suffixes])
+ search = u" ".join(words[1:])
+ authors = self.quotedb.search_authors(search, regexp)
+ if not authors:
+ serv.privmsg(canal, "%s: Pas d'auteur correspondant à la recherche." % (auteur,))
+ return
+ if cmd.startswith("author"):
+ if len(authors) > config.search_max_authors:
+ authors = authors[:config.search_max_authors+1] + ["+%s" % (len(authors) - config.search_max_authors)]
+ serv.privmsg(canal, "%s: %s" % (auteur, (u", ".join(authors)).encode("utf-8")))
+ elif cmd.startswith("from"):
+ quotes = sum([self.quotedb.quotesfrom(a) for a in authors], [])
+ q = random.choice(quotes)
+ serv.privmsg(canal, q.display(show_context))
+ elif message.startswith(u"!search"):
+ words = message.split()
+ cmd = words[0].lstrip("!")
+ regexp = cmd in ["search" + suffix for suffix in config.regex_suffixes]
+ search = u" ".join(words[1:])
+ quotes = self.quotedb.search(inquote=search, regexp=regexp)
+ # On recherche également sur le contexte si on est en !context-search
+ if show_context:
+ quotes += self.quotedb.search(place=search, regexp=regexp)
+ # Pour pas biaiser le choix aléatoire, on enlève les doublons
+ quotes = list(set(quotes))
+ if quotes:
+ q = random.choice(quotes)
+ serv.privmsg(canal, q.display(show_context))
+ else:
+ serv.privmsg(canal, "%s: Pas de quotes correspondant à la recherche." % (auteur,))
def on_action(self, serv, ev):
"""À la réception d'une action."""
action = bot_unicode(action)
except errors.UnicodeBotError:
if config.utf8_trigger and not channel in self.quiet_channels:
- serv.privmsg(channel, (u"%s: %s"%(auteur,random.choice(config.utf8_fail_answers))).encode("utf8"))
+ serv.privmsg(channel, (u"%s: %s"%(auteur,random.choice(config.utf8_fail_answers))).encode("utf-8"))
return
mypseudo = self.nick
victime = ev.arguments()[0]
raison = ev.arguments()[1]
if victime == self.nick:
- log(self.serveur, u"%s kické de %s par %s (raison : %s)" % (victime, channel.decode("utf-8"), auteur, raison))
+ log(self.serveur, ("%s kické de %s par %s (raison : %s)" % (victime, channel, auteur, raison)).decode("utf-8"))
time.sleep(2)
serv.join(channel)
parrot.start_as_daemon(outfile)
else:
# on enregistre le pid de parrot
- pidfile = "/var/run/bots/parror.pid"
+ pidfile = "/var/run/bots/parrot.pid"
for arg in sys.argv:
arg = arg.split("=")
if arg[0].strip('-') in ["pidfile"]:
projects / bots / parrot.git / blobdiff