diff --git a/app/dragon-bot.py b/app/dragon-bot.py index 2ef6359f..ee7329d3 100644 --- a/app/dragon-bot.py +++ b/app/dragon-bot.py @@ -8,6 +8,7 @@ import define_word import help_methods import discord import docker +import wallpaper from pybooru import Danbooru # Client object @@ -26,7 +27,7 @@ async def on_ready(): print("\n********************************") await client.change_presence(game=discord.Game(name='Type !help to see how to use me')) - + if debug: print("\nPress control+c to exit the bot") print("Followed by control+d or by typing") @@ -192,17 +193,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