rsivakov / maury-bot Goto Github PK

class MauryBot(AbstractBotPersonality, AbstractBot):
    def __init__(self):
        AbstractBotPersonality.__init__(self)
        self.name = "Captain Maury"
        self.description = "The Desolate of Foggy Dock"
        self.adjectives = ["desolate", "grungy", "seafaring ghost captain"]
        self.verb = "haunting the docks"
        self.location = "fisherman's wharf"
        self.statuses = [
                "the lapping of the waves against the pier",
                "the snapping of a flag in the breeze",
                "the scuffle of feet from dock workers",
                "the clang of a boat's bell",
                "the rattle of the mooring chains",
                "the chatter of fishermen",
                "the low hum of boat engines",
                "the distant rumble of thunder",
                "the gentle clinking of fishing lines",
                "the thrum of heavy cargo machinery",
            ]
        AbstractBot.__init__(self)
        self.handler = PersonalityHandler(self)
        self.avatar_file = "maury_bot/avatars/maury.png"
        # load file into bytes
        with open(self.avatar_file, "rb") as f:
            self.avatar = f.read()

# EMOTION
if any([kwarg == emoji.name for kwarg in ["emoji_name1", "emoji_name2"]):
    prompt += "Respond with EMOTION to the message from the author, on behalf of yourself and the reactor."

# condemn, tread lightly
if any([kwarg == emoji.name for kwarg in ["judgement", "flip_off", "banned"]]):
    prompt += "Respond with a condemnation of the message from the author, on behalf of yourself and the reactor."

#say congratulations
elif any([kwarg == emoji.name for kwarg in ["sheeee", "flawless_victory", "ole", "pog"]]):
    prompt += "Respond with congratulations to the message from the author, on behalf of yourself and the reactor."

react_probability = 0.05

@bot.event
async def on_message(message: discord.Message) -> None:
    ...
    if "<trigger:emote>" in message.content:
        bot.high_activity = 2
        bot_activity_level.start()
        await message.add_reaction("<acknowledge:emote>")
    ...

# probability of reacting, banned has 1, otherwise .05
    react_probability = 1 #0.05
    if not bot.high_activity and np.random.random() > react_probability and reaction.emoji.name != "banned":
        return