Add check if user is muted for userinfo

This commit is contained in:
Zeva Rose 2022-08-16 12:50:45 -06:00
parent 4660bab812
commit 4acb37a566

View file

@ -8,13 +8,6 @@ from io import BytesIO
import numpy as np import numpy as np
from dateparser import parse from dateparser import parse
from PIL import Image
from tzlocal import get_localzone
from jarvis import const as jconst
from jarvis.data import pigpen
from jarvis.data.robotcamo import emotes, hk, names
from jarvis.utils import build_embed, get_repo_hash
from naff import Client, Extension, InteractionContext, const from naff import Client, Extension, InteractionContext, const
from naff.models.discord.channel import GuildCategory, GuildText, GuildVoice from naff.models.discord.channel import GuildCategory, GuildText, GuildVoice
from naff.models.discord.embed import EmbedField from naff.models.discord.embed import EmbedField
@ -32,6 +25,13 @@ from naff.models.naff.application_commands import (
) )
from naff.models.naff.command import cooldown from naff.models.naff.command import cooldown
from naff.models.naff.cooldowns import Buckets from naff.models.naff.cooldowns import Buckets
from PIL import Image
from tzlocal import get_localzone
from jarvis import const as jconst
from jarvis.data import pigpen
from jarvis.data.robotcamo import emotes, hk, names
from jarvis.utils import build_embed, get_repo_hash
JARVIS_LOGO = Image.open("jarvis_small.png").convert("RGBA") JARVIS_LOGO = Image.open("jarvis_small.png").convert("RGBA")
@ -187,6 +187,8 @@ class UtilCog(Extension):
if not await ctx.guild.fetch_member(user.id): if not await ctx.guild.fetch_member(user.id):
await ctx.send("That user isn't in this guild.", ephemeral=True) await ctx.send("That user isn't in this guild.", ephemeral=True)
return return
muted = user.communication_disabled_until is not None
user_roles = user.roles user_roles = user.roles
if user_roles: if user_roles:
user_roles = sorted(user.roles, key=lambda x: -x.position) user_roles = sorted(user.roles, key=lambda x: -x.position)
@ -215,7 +217,8 @@ class UtilCog(Extension):
) )
embed.set_author( embed.set_author(
name=f"{user.display_name}#{user.discriminator}", icon_url=user.display_avatar.url name=f"{'🔇 ' if muted else ''}{user.display_name}#{user.discriminator}",
icon_url=user.display_avatar.url,
) )
embed.set_thumbnail(url=user.display_avatar.url) embed.set_thumbnail(url=user.display_avatar.url)
embed.set_footer(text=f"ID: {user.id}") embed.set_footer(text=f"ID: {user.id}")