155 lines
5.0 KiB
Python
155 lines
5.0 KiB
Python
from discord import (
|
|
ApplicationContext,
|
|
SlashCommandGroup,
|
|
User,
|
|
option,
|
|
)
|
|
from discord.ext.commands import Cog
|
|
from libbot.i18n import _, in_every_locale
|
|
|
|
from classes import PycordUser
|
|
from classes.pycord_bot import PycordBot
|
|
from modules.utils import is_operation_confirmed
|
|
|
|
|
|
class CogUser(Cog):
|
|
"""Cog with user management commands."""
|
|
|
|
def __init__(self, bot: PycordBot):
|
|
self.bot: PycordBot = bot
|
|
|
|
command_group: SlashCommandGroup = SlashCommandGroup(
|
|
"user",
|
|
description=_("description", "commands", "user"),
|
|
description_localizations=in_every_locale("description", "commands", "user"),
|
|
)
|
|
|
|
# TODO Implement the command
|
|
# @command_group.command(
|
|
# name="create_channel",
|
|
# description="Create channel for the user",
|
|
# )
|
|
# @option(
|
|
# "user",
|
|
# description="Selected user",
|
|
# )
|
|
# async def command_user_create_channel(self, ctx: ApplicationContext, user: User) -> None:
|
|
# await ctx.respond("Not implemented.")
|
|
|
|
# TODO Implement the command
|
|
# @command_group.command(
|
|
# name="update_channel",
|
|
# description="Update user's channel",
|
|
# )
|
|
# @option(
|
|
# "user",
|
|
# description="Selected user",
|
|
# )
|
|
# async def command_user_update_channel(self, ctx: ApplicationContext, user: User) -> None:
|
|
# await ctx.respond("Not implemented.")
|
|
|
|
# TODO Implement the command
|
|
# @command_group.command(
|
|
# name="delete_channel",
|
|
# description="Delete user's channel",
|
|
# )
|
|
# @option(
|
|
# "user",
|
|
# description="Selected user",
|
|
# )
|
|
# @option("confirm", description="Confirmation of the operation", required=False)
|
|
# async def command_user_delete_channel(
|
|
# self, ctx: ApplicationContext, user: User, confirm: bool = False
|
|
# ) -> None:
|
|
# await ctx.respond("Not implemented.")
|
|
|
|
@command_group.command(
|
|
name="jail",
|
|
description=_("description", "commands", "user_jail"),
|
|
description_localizations=in_every_locale("description", "commands", "user_jail"),
|
|
)
|
|
@option(
|
|
"user",
|
|
description=_("description", "commands", "user_jail", "options", "user"),
|
|
description_localizations=in_every_locale(
|
|
"description", "commands", "user_jail", "options", "user"
|
|
),
|
|
)
|
|
@option(
|
|
"confirm",
|
|
description=_("description", "commands", "user_jail", "options", "confirm"),
|
|
description_localizations=in_every_locale(
|
|
"description", "commands", "user_jail", "options", "confirm"
|
|
),
|
|
required=False,
|
|
)
|
|
async def command_user_jail(self, ctx: ApplicationContext, user: User, confirm: bool = False) -> None:
|
|
if not (await is_operation_confirmed(ctx, confirm)):
|
|
return
|
|
|
|
pycord_user: PycordUser = await self.bot.find_user(user, ctx.guild)
|
|
|
|
if pycord_user.is_jailed:
|
|
await ctx.respond(
|
|
self.bot._("user_jail_already_jailed", "messages", locale=ctx.locale).format(
|
|
display_name=user.display_name
|
|
),
|
|
ephemeral=True,
|
|
)
|
|
return
|
|
|
|
await pycord_user.jail(self.bot.cache)
|
|
|
|
await ctx.respond(
|
|
self.bot._("user_jail_successful", "messages", locale=ctx.locale).format(
|
|
display_name=user.display_name
|
|
)
|
|
)
|
|
|
|
@command_group.command(
|
|
name="unjail",
|
|
description=_("description", "commands", "user_unjail"),
|
|
description_localizations=in_every_locale("description", "commands", "user_unjail"),
|
|
)
|
|
@option(
|
|
"user",
|
|
description=_("description", "commands", "user_unjail", "options", "user"),
|
|
description_localizations=in_every_locale(
|
|
"description", "commands", "user_unjail", "options", "user"
|
|
),
|
|
)
|
|
@option(
|
|
"confirm",
|
|
description=_("description", "commands", "user_unjail", "options", "confirm"),
|
|
description_localizations=in_every_locale(
|
|
"description", "commands", "user_unjail", "options", "confirm"
|
|
),
|
|
required=False,
|
|
)
|
|
async def command_user_unjail(self, ctx: ApplicationContext, user: User, confirm: bool = False) -> None:
|
|
if not (await is_operation_confirmed(ctx, confirm)):
|
|
return
|
|
|
|
pycord_user: PycordUser = await self.bot.find_user(user, ctx.guild)
|
|
|
|
if not pycord_user.is_jailed:
|
|
await ctx.respond(
|
|
self.bot._("user_unjail_not_jailed", "messages", locale=ctx.locale).format(
|
|
display_name=user.display_name
|
|
),
|
|
ephemeral=True,
|
|
)
|
|
return
|
|
|
|
await pycord_user.unjail(self.bot.cache)
|
|
|
|
await ctx.respond(
|
|
self.bot._("user_unjail_successful", "messages", locale=ctx.locale).format(
|
|
display_name=user.display_name
|
|
)
|
|
)
|
|
|
|
|
|
def setup(bot: PycordBot) -> None:
|
|
bot.add_cog(CogUser(bot))
|