From b7af111207b2fa507493015a70004bcde587dbbd Mon Sep 17 00:00:00 2001 From: Luke Robles Date: Fri, 9 Mar 2018 17:32:50 -0800 Subject: [PATCH] Return the users avatar if no username is specified --- app/dragon-bot.py | 23 +++++++++++++---------- 1 file changed, 13 insertions(+), 10 deletions(-) diff --git a/app/dragon-bot.py b/app/dragon-bot.py index c6ff2a3b..89d51ba7 100644 --- a/app/dragon-bot.py +++ b/app/dragon-bot.py @@ -81,17 +81,20 @@ async def on_message(message): eight_ball.check_8ball(message.content) ) - if (message.content.startswith('!avatar') and message.mentions.__len__() > 0 ): + if message.content.startswith('!avatar'): + profile = message.author + if len(message.mentions): + profile = message.mentions[0] + # Code stolen from this reddit post - # https://www.reddit.com/r/discordapp/comments/74bb4z/retrieve_a_mentioned_users_avatar_using_discordpy/ - for user in message.mentions: - embed = discord.Embed( - title="{}#{}".format(user.name,user.discriminator), - description="[Direct Link]({})".format(user.avatar_url), - type='rich' - ) - embed.set_image(url=user.avatar_url) - await client.send_message(message.channel, embed=embed) + # https://www.reddit.com/r/discordapp/comments/74bb4z/retrieve_a_mentioned_users_avatar_using_discordpy + embed = discord.Embed( + title="{}#{}".format(profile.name, profile.discriminator), + description="[Direct Link]({})".format(profile.avatar_url), + type='rich' + ) + embed.set_image(url=profile.avatar_url) + await client.send_message(message.channel, embed=embed) if message.content.startswith('!cleanup'): if not role_check.cleanup_permissions(message.author.roles):