blob: 33cb2a9dff7b69f9099c2ae8cfeb13a246f24ce5 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
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="utf8") 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))
|
