Restrict nsfw subreddit(s) or similar (subreddits that require you to be over 18).

Changed the return format a little bit for the fetch_posts() function, instead of returning an empty list, it returns a list with a dict holding the error message.

1 files changed, 18 insertions, 7 deletions

diff --git a/bot/cogs/reddit.py b/bot/cogs/reddit.py
index 5d9e2c20b..0b002f9b6 100644
--- a/bot/cogs/reddit.py
+++ b/bot/cogs/reddit.py
@@ -141,12 +141,27 @@ class Reddit(Cog):
                 # Got appropriate response - process and return.
                 content = await response.json()
                 posts = content["data"]["children"]
+                if posts[0]["data"]["over_18"]:
+                    resp_not_allowed = [
+                        {
+                            "error": "Oops ! Looks like this subreddit, doesn't fit in the scope of the server."
+                        }
+                    ]
+                    return resp_not_allowed
                 return posts[:amount]
 
             await asyncio.sleep(3)
 
         log.debug(f"Invalid response from: {url} - status code {response.status}, mimetype {response.content_type}")
-        return list()  # Failed to get appropriate response within allowed number of retries.
+        resp_failed = [
+            {
+                "error": (
+                    "Sorry! We couldn't find any posts from that subreddit. "
+                    "If this problem persists, please let us know."
+                )
+            }
+        ]
+        return resp_failed  # Failed to get appropriate response within allowed number of retries.
 
     async def get_top_posts(self, subreddit: Subreddit, time: str = "all", amount: int = 5) -> Embed:
         """
@@ -164,14 +179,10 @@ class Reddit(Cog):
             amount=amount,
             params={"t": time}
         )
-
-        if not posts:
+        if "error" in posts[0]:
             embed.title = random.choice(ERROR_REPLIES)
             embed.colour = Colour.red()
-            embed.description = (
-                "Sorry! We couldn't find any posts from that subreddit. "
-                "If this problem persists, please let us know."
-            )
+            embed.description = posts[0]["error"]
 
             return embed