aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar AvianAnalyst <[email protected]>2019-06-18 11:03:03 -0400
committerGravatar AvianAnalyst <[email protected]>2019-06-18 11:03:03 -0400
commitdb81de60117d72d056831fde5caa691fd02b9125 (patch)
treefed01ab10713012f7d8549dc4acc0823e422cfdb
parentFinished vetting song list and received mod approval (diff)
Added type hints
-rw-r--r--bot/seasons/pride/pride_anthem.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/bot/seasons/pride/pride_anthem.py b/bot/seasons/pride/pride_anthem.py
index 7a0125f7..0e2ee9c9 100644
--- a/bot/seasons/pride/pride_anthem.py
+++ b/bot/seasons/pride/pride_anthem.py
@@ -15,7 +15,7 @@ class PrideAnthem(commands.Cog):
self.bot = bot
self.anthems = self.load_vids()
- def get_video(self, genre: str = None):
+ def get_video(self, genre: str = None) -> dict:
if not genre:
return choice(self.anthems)
else:
@@ -26,13 +26,13 @@ class PrideAnthem(commands.Cog):
log.info('No videos for that genre.')
@staticmethod
- def load_vids():
+ def load_vids() -> list:
with open(Path('bot', 'resources', 'pride', 'anthems.json').absolute(), 'r') as f:
anthems = load(f)
return anthems
@commands.command(name='prideanthem')
- async def send_anthem(self, ctx, genre=None):
+ async def send_anthem(self, ctx, genre: str = None):
anthem = self.get_video(genre)
# embed = Embed(title='Pride Anthem',
# description="Here is a pride anthem to check out!")