import random from edmond.plugin import Plugin from edmond.plugins.mood import Mood from edmond.utils import proc class InsultsPlugin(Plugin): REQUIRED_CONFIGS = ["commands", "sentences", "pissed_rate", "deny_message"] def __init__(self, bot): super().__init__(bot) self.mood_plugin = None def on_welcome(self, event): self.mood_plugin = self.bot.get_plugin("mood") def on_pubmsg(self, event): message = event.arguments[0] if self.should_handle_command(message): self.handle_insult_command(event.target) return True content = self.should_read_message(message) if content is None: return False if content in self.config["sentences"]: self.react_to_insult(event.source.nick, event.target) return True return False def handle_insult_command(self, target): if self.mood_plugin is not None: mood = self.get_runtime_value("mood", ns="mood") if mood == Mood.CALM: self.bot.say(target, self.config["deny_message"]) return insult_target = self.command.content if not insult_target: return insult = random.choice(self.config["sentences"]) reply = f"{insult_target} {insult}" self.bot.say(target, reply) def react_to_insult(self, sender, target): if self.mood_plugin is not None: mood = self.get_runtime_value("mood", ns="mood") if mood == Mood.CALM and proc(self.config["pissed_rate"]): self.mood_plugin.get_pissed(target) insult = random.choice(self.config["sentences"]) reply = f"{sender} {insult}" self.bot.say(target, reply)