Merge pull request #329 from python-discord/seasonal-purge

Deseasonify: Make all cogs available year-round, and manage only branding by season.

1 files changed, 57 insertions, 0 deletions

diff --git a/bot/exts/pride/pride_anthem.py b/bot/exts/pride/pride_anthem.py
new file mode 100644
index 00000000..186c5fff
--- /dev/null
+++ b/bot/exts/pride/pride_anthem.py
@@ -0,0 +1,57 @@
+import json
+import logging
+import random
+from pathlib import Path
+
+from discord.ext import commands
+
+log = logging.getLogger(__name__)
+
+
+class PrideAnthem(commands.Cog):
+    """Embed a random youtube video for a gay anthem!"""
+
+    def __init__(self, bot: commands.Bot):
+        self.bot = bot
+        self.anthems = self.load_vids()
+
+    def get_video(self, genre: str = None) -> dict:
+        """
+        Picks a random anthem from the list.
+
+        If `genre` is supplied, it will pick from videos attributed with that genre.
+        If none can be found, it will log this as well as provide that information to the user.
+        """
+        if not genre:
+            return random.choice(self.anthems)
+        else:
+            songs = [song for song in self.anthems if genre.casefold() in song["genre"]]
+            try:
+                return random.choice(songs)
+            except IndexError:
+                log.info("No videos for that genre.")
+
+    @staticmethod
+    def load_vids() -> list:
+        """Loads a list of videos from the resources folder as dictionaries."""
+        with open(Path("bot/resources/pride/anthems.json"), "r", encoding="utf-8") as f:
+            anthems = json.load(f)
+        return anthems
+
+    @commands.command(name="prideanthem", aliases=["anthem", "pridesong"])
+    async def prideanthem(self, ctx: commands.Context, genre: str = None) -> None:
+        """
+        Sends a message with a video of a random pride anthem.
+
+        If `genre` is supplied, it will select from that genre only.
+        """
+        anthem = self.get_video(genre)
+        if anthem:
+            await ctx.send(anthem["url"])
+        else:
+            await ctx.send("I couldn't find a video, sorry!")
+
+
+def setup(bot: commands.Bot) -> None:
+    """Cog loader for pride anthem."""
+    bot.add_cog(PrideAnthem(bot))