aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar Chris <[email protected]>2021-04-22 20:25:38 +0100
committerGravatar Chris <[email protected]>2021-04-22 20:25:38 +0100
commit29084d1576e1435a5a4a32071a79b6af28acd362 (patch)
tree63e5ecd6004d0a9c59510999f3fe701bc8b6c2f0
parentMerge pull request #697 from ToxicKidz/fix_catify (diff)
Fix errors when a subreddit has <5 posts.
If a subreddit has <2 posts, the posts[1] check would fail with an IndexError. If the subreddit had less that 5 posts, then the k=5 check would also error. These changes harden the command for these edge cases.
-rw-r--r--bot/exts/evergreen/reddit.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/bot/exts/evergreen/reddit.py b/bot/exts/evergreen/reddit.py
index 49127bea..4fdb6fca 100644
--- a/bot/exts/evergreen/reddit.py
+++ b/bot/exts/evergreen/reddit.py
@@ -54,7 +54,7 @@ class Reddit(commands.Cog):
if not posts:
return await ctx.send('No posts available!')
- if posts[1]["data"]["over_18"] is True:
+ if posts[0]["data"]["over_18"] is True:
return await ctx.send(
"You cannot access this Subreddit as it is ment for those who "
"are 18 years or older."
@@ -63,7 +63,7 @@ class Reddit(commands.Cog):
embed_titles = ""
# Chooses k unique random elements from a population sequence or set.
- random_posts = random.sample(posts, k=5)
+ random_posts = random.sample(posts, k=min(len(posts), 5))
# -----------------------------------------------------------
# This code below is bound of change when the emojis are added.