from bson.errors import InvalidId from discord import ApplicationContext, Cog, option, slash_command from discord.utils import basic_autocomplete from classes import PycordEvent, PycordGuild, PycordUser from classes.pycord_bot import PycordBot from modules.utils import autocomplete_user_registered_events, is_operation_confirmed class CogUnregister(Cog): """Cog with the event unregistration command.""" def __init__(self, bot: PycordBot): self.bot: PycordBot = bot # TODO Introduce i18n @slash_command( name="unregister", description="Leave the selected event", ) @option( "event", description="Name of the event", autocomplete=basic_autocomplete(autocomplete_user_registered_events), ) @option("confirm", description="Confirmation of the operation", required=False) async def command_unregister(self, ctx: ApplicationContext, event: str, confirm: bool = False) -> None: if not (await is_operation_confirmed(ctx, confirm)): return guild: PycordGuild = await self.bot.find_guild(ctx.guild.id) try: pycord_event: PycordEvent = await self.bot.find_event(event_id=event) except (InvalidId, RuntimeError): # TODO Make a nice message await ctx.respond("Event was not found.") return if not guild.is_configured(): await ctx.respond(self.bot._("guild_unconfigured", "messages", locale=ctx.locale)) return user: PycordUser = await self.bot.find_user(ctx.author, ctx.guild) if user.is_jailed: # TODO Make a nice message await ctx.respond("You are jailed and cannot interact with events. Please, contact the administrator.") return if pycord_event._id not in user.registered_event_ids: # TODO Make a nice message await ctx.respond("You are not registered for this event.") return await user.event_unregister(pycord_event._id, cache=self.bot.cache) # TODO Text channel must be locked and updated await ctx.respond("Ok.") def setup(bot: PycordBot) -> None: bot.add_cog(CogUnregister(bot))