liz/PluralKit
liz/PluralKit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bot: get avatar url from attachment if not given directly

Browse Source 
* Update member_commands.py

Add ability to get avatar image from an attachment

* Update system_commands.py

Adding ability to get avatar from an attachment
This commit is contained in:
Grey Himmel 2019-05-03 08:50:46 -04:00 committed by Astrid
parent 2c0a588fcc
commit 9cc1d899a6
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/pluralkit/bot/commands/member_commands.py
View File

@ -128,6 +128,9 @@ async def member_avatar(ctx: CommandContext, member: Member):
user = await utils.parse_mention(ctx.client, new_avatar_url) user = await utils.parse_mention(ctx.client, new_avatar_url)
if user: if user:
new_avatar_url = user.avatar_url_as(format="png") new_avatar_url = user.avatar_url_as(format="png")
if not new_avatar_url and ctx.message.attachments[0]:
new_avatar_url = ctx.message.attachments[0].url
await member.set_avatar(ctx.conn, new_avatar_url) await member.set_avatar(ctx.conn, new_avatar_url)
await ctx.reply_ok("Member avatar {}.".format("updated" if new_avatar_url else "cleared")) await ctx.reply_ok("Member avatar {}.".format("updated" if new_avatar_url else "cleared"))

3
src/pluralkit/bot/commands/system_commands.py
View File

@ -184,6 +184,9 @@ async def system_avatar(ctx: CommandContext):
user = await utils.parse_mention(ctx.client, new_avatar_url) user = await utils.parse_mention(ctx.client, new_avatar_url)
if user: if user:
new_avatar_url = user.avatar_url_as(format="png") new_avatar_url = user.avatar_url_as(format="png")
if not new_avatar_url and ctx.message.attachments[0]:
new_avatar_url = ctx.message.attachments[0].url
await system.set_avatar(ctx.conn, new_avatar_url) await system.set_avatar(ctx.conn, new_avatar_url)
await ctx.reply_ok("System avatar {}.".format("updated" if new_avatar_url else "cleared")) await ctx.reply_ok("System avatar {}.".format("updated" if new_avatar_url else "cleared"))