aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar Will Da Silva <[email protected]>2019-10-01 15:44:40 -0400
committerGravatar Will Da Silva <[email protected]>2019-10-01 15:44:40 -0400
commita1c9078e69ddaec24c632bd1bd85826d6abe2d65 (patch)
tree6f0949d140db8b1b93bcc014d4583222dfb15e79
parentAdd 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.py32
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!")