From fc7c65a6d96e089d13eba282ab3d0fdbaae5d8dc Mon Sep 17 00:00:00 2001
From: Unknown <kylerjuresic@gmail.com>
Date: Wed, 16 Aug 2017 21:00:32 -0500
Subject: [PATCH] wallpaper moved to isolated module

also allows !wallpaper help
see issue #7 for details
---
 app/dragon-bot.py | 16 +++++-----------
 app/wallpaper.py  | 29 +++++++++++++++++++++++++++++
 2 files changed, 34 insertions(+), 11 deletions(-)
 create mode 100644 app/wallpaper.py

diff --git a/app/dragon-bot.py b/app/dragon-bot.py
index 57aa3772..a19a12a2 100644
--- a/app/dragon-bot.py
+++ b/app/dragon-bot.py
@@ -10,6 +10,7 @@ import define_word
 import help_methods
 import discord
 import docker
+import wallpaper
 from pybooru import Danbooru
 
 # Client object
@@ -191,17 +192,10 @@ async def on_message(message):
         # await client.send_message(message.channel, 'Deleted {} message(s)'.format(len(deleted)))
 
     if message.content.startswith('!wallpaper'):
-        url = 'https://source.unsplash.com/3840x2160/'
-        if len(message.content.split()) > 1:
-            keyword = message.content.split()[1]
-            url = "?".join((url, keyword))
-
-            if 'waifu' in keyword:
-                url = 'https://media0.giphy.com/media/C79RKZ7nOcK8U/giphy.gif'
-
-        results = requests.get(url).url
-        await client.send_message(message.channel, results)
-
+        await client.send_message(
+            message.channel,
+            wallpaper.getselect(message.content)
+    )
 
     ###################################
     ###### +-------------------+ ######
diff --git a/app/wallpaper.py b/app/wallpaper.py
new file mode 100644
index 00000000..26d7af2d
--- /dev/null
+++ b/app/wallpaper.py
@@ -0,0 +1,29 @@
+import help_methods
+
+def getselect(select):
+    url = 'https://source.unsplash.com/3840x2160/'
+
+    if len(select.split()) > 2:
+        keyword = str(select.split('-'))
+        keyword = keyword.replace('[', '')
+        keyword = keyword.replace(']', '')
+        keyword = keyword.replace('\'', '')
+        keyword = keyword.replace(' ', '&')
+        keyword = keyword.replace('!wallpaper&', '')
+        #print(url)
+        url = "?".join((url, keyword))
+        if 'waifu' in keyword:
+            url = 'https://media0.giphy.com/media/C79RKZ7nOcK8U/giphy.gif'
+
+        return url
+
+    if len(select.split()) > 1:
+        keyword = select.split()[1]
+        #print (url)
+        url = "?".join((url, keyword))
+        if 'waifu' in keyword:
+            url = 'https://media0.giphy.com/media/C79RKZ7nOcK8U/giphy.gif'
+        if 'help' in keyword:
+            return help_methods.get_help_message('wallpaper')
+
+        return url