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