diff options
author | 2021-04-22 20:25:38 +0100 | |
---|---|---|
committer | 2021-04-22 20:25:38 +0100 | |
commit | 29084d1576e1435a5a4a32071a79b6af28acd362 (patch) | |
tree | 63e5ecd6004d0a9c59510999f3fe701bc8b6c2f0 | |
parent | Merge 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.py | 4 |
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. |