eightball/PluralKit
1
0
Fork 0
mirror of https://github.com/PluralKit/PluralKit.git synced 2026-02-04 04:56:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

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
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)
if user:
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 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)
if user:
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 ctx.reply_ok("System avatar {}.".format("updated" if new_avatar_url else "cleared"))