From 876f20d10076c6f1dcfb4e21e88c7fd469eacc5d Mon Sep 17 00:00:00 2001 From: Luke Robles Date: Mon, 12 Mar 2018 13:00:51 -0700 Subject: [PATCH] renaming avatar file to make more sense --- app/dragon-bot.py | 2 +- app/set_avatar.py | 17 +++++++++++++++++ 2 files changed, 18 insertions(+), 1 deletion(-) create mode 100644 app/set_avatar.py diff --git a/app/dragon-bot.py b/app/dragon-bot.py index 3e7e1e26..65741e77 100644 --- a/app/dragon-bot.py +++ b/app/dragon-bot.py @@ -43,7 +43,7 @@ async def on_ready(): ) if not debug: - await client.edit_profile(avatar=avatar.change_avatar()) + await client.edit_profile(avatar=set_avatar.change_bots_avatar()) print("\n********************************") print("\nDRAGON BOT RUNNING IN {} MODE".format(dragon_environment.upper())) diff --git a/app/set_avatar.py b/app/set_avatar.py new file mode 100644 index 00000000..eef173e8 --- /dev/null +++ b/app/set_avatar.py @@ -0,0 +1,17 @@ +import requests + +import core_utils +import get_from_reddit + +def change_bots_avatar(): + image = None + while not image: + image = get_from_reddit.get_image(boards='smuganimegirls') + extension = image.split('.')[-1] + local_smug = "/tmp/smug.{}".format(extension) + + if extension.lower() in ['png', 'jpg']: + # save an image locally + return open(core_utils.download_image(image, local_smug), 'rb').read() + else: + image = None