diff options
author | 2019-10-01 15:44:40 -0400 | |
---|---|---|
committer | 2019-10-01 15:44:40 -0400 | |
commit | a1c9078e69ddaec24c632bd1bd85826d6abe2d65 (patch) | |
tree | 6f0949d140db8b1b93bcc014d4583222dfb15e79 | |
parent | Add drag queen names (diff) |
Add drag queen name generator
Could be improved by writing a message that says:
"@command_invoker's drag queen name is <generated name>"
-rw-r--r-- | bot/seasons/pride/drag_queen_name.py | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/bot/seasons/pride/drag_queen_name.py b/bot/seasons/pride/drag_queen_name.py new file mode 100644 index 00000000..f34da00e --- /dev/null +++ b/bot/seasons/pride/drag_queen_name.py @@ -0,0 +1,32 @@ +import logging +import random +from pathlib import Path + +from discord.ext import commands + +log = logging.getLogger(__name__) + + +class DragNames(commands.Cog): + """Gives a random drag queen name""" + + def __init__(self, bot: commands.Bot): + self.bot = bot + self.names = self.load_names() + + @staticmethod + def load_names() -> list: + """Loads a list of drag queen names.""" + with open(Path("bot/resources/pride/drag_queen_names.txt"), "r", encoding="utf-8") as f: + return f.readlines() + + @commands.command(name="dragname", aliases=["dragqueenname", "queenme"]) + async def dragname(self, ctx: commands.Context) -> None: + """Sends a message with a drag queen name.""" + await ctx.send(random.choice(self.names)) + + +def setup(bot: commands.Bot) -> None: + """Cog loader for drag queen name generator.""" + bot.add_cog(DragNames(bot)) + log.info("Drag queen name generator cog loaded!") |