jarvis/jarvis-bot
jarvis/jarvis-bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Change from CacheCog to Scale, fix permissions and formatting

Browse source
This commit is contained in:
Zeva Rose 2022-02-02 19:45:55 -07:00
parent cd49bd9e78
commit a10d297e4e
1 changed files with 3 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
jarvis/cogs/admin/kick.py
View file

@ -1,5 +1,5 @@
"""J.A.R.V.I.S. KickCog.""" """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.embed import EmbedField
from dis_snek.models.discord.user import User from dis_snek.models.discord.user import User
from dis_snek.models.snek.application_commands import ( 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.db.models import Kick
from jarvis.utils import build_embed from jarvis.utils import build_embed
from jarvis.utils.cachecog import CacheCog
from jarvis.utils.permissions import admin_or_permissions from jarvis.utils.permissions import admin_or_permissions
class KickCog(CacheCog): class KickCog(Scale):
"""J.A.R.V.I.S. KickCog.""" """J.A.R.V.I.S. KickCog."""
def __init__(self, bot: Snek):
super().__init__(bot)
@slash_command(name="kick", description="Kick a user") @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="user", description="User to kick", option_type=OptionTypes.USER, required=True)
@slash_option(name="reason", description="Kick reason", option_type=OptionTypes.STRING, 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: async def _kick(self, ctx: InteractionContext, user: User, reason: str) -> None:
if not user or user == ctx.author: if not user or user == ctx.author:
await ctx.send("You cannot kick yourself.", hidden=True) await ctx.send("You cannot kick yourself.", hidden=True)