Fixing the lewd error message if someone tries to use it in a not lewd channel

app/dragon-bot.py

@@ -226,6 +226,7 @@ async def on_message(message):
         )
 
     if message.content.startswith('!lewd'):
+        if 'nsfw' in message.channel.name:
             await client.send_message(
                 message.channel,
                 embed=generate_embed(
@@ -233,6 +234,11 @@ async def on_message(message):
                     embed_title="{} is being lewd".format(message.author.name)
                 )
             )
+        else:
+            await client.send_message(
+                message.channel,
+                'You can only use this command in NSFW channels'
+            )
 
     if message.content.startswith('!weather'):
         await client.send_message(

app/lewds.py

@@ -41,8 +41,15 @@ def get_from_danbooru(boards, nsfw=True):
     return get_from_danbooru(boards)
 
 
-def get_lewd(channel_name):
+def get_lewd():
-    if 'nsfw' in channel_name:
+    """
+    get_lewd()
+
+    Chooses between getting from reddit or danbooru
+    If reddit is chosen, picks a random subreddit from the list of boards and
+    grabs a random image using the get_from_reddit() method
+    """
+    # List of subreddits that have the lewds we seek
     boards = [
         '2Booty',
         'ahegao',
@@ -62,4 +69,3 @@ def get_lewd(channel_name):
 
     # This is really bad practice but pass boards to get_from_reddit AND danbooru so it doesnt error
     return random.choice([get_from_reddit.get_image, get_from_danbooru])(boards=boards, nsfw=True)
-    return 'You can only use this command in NSFW channels'