diff --git a/jarvis/cogs/admin/kick.py b/jarvis/cogs/admin/kick.py index 28d2d1d..34f5ed2 100644 --- a/jarvis/cogs/admin/kick.py +++ b/jarvis/cogs/admin/kick.py @@ -1,5 +1,5 @@ """J.A.R.V.I.S. KickCog.""" -from dis_snek import InteractionContext, Snek +from dis_snek import InteractionContext, Permissions, Scale from dis_snek.models.discord.embed import EmbedField from dis_snek.models.discord.user import User from dis_snek.models.snek.application_commands import ( @@ -10,20 +10,16 @@ from dis_snek.models.snek.application_commands import ( from jarvis.db.models import Kick from jarvis.utils import build_embed -from jarvis.utils.cachecog import CacheCog from jarvis.utils.permissions import admin_or_permissions -class KickCog(CacheCog): +class KickCog(Scale): """J.A.R.V.I.S. KickCog.""" - def __init__(self, bot: Snek): - super().__init__(bot) - @slash_command(name="kick", description="Kick a user") @slash_option(name="user", description="User to kick", option_type=OptionTypes.USER, required=True) @slash_option(name="reason", description="Kick reason", option_type=OptionTypes.STRING, required=True) - @admin_or_permissions(kick_members=True) + @admin_or_permissions(Permissions.BAN_MEMBERS) async def _kick(self, ctx: InteractionContext, user: User, reason: str) -> None: if not user or user == ctx.author: await ctx.send("You cannot kick yourself.", hidden=True)