diff --git a/cogs/staffings.py b/cogs/staffings.py index e38a106..2e43683 100644 --- a/cogs/staffings.py +++ b/cogs/staffings.py @@ -37,12 +37,30 @@ def cog_unload(self): # # A function to dynamically fetch and return a list of titles as choices - async def get_title_choices(self) -> List[app_commands.Choice[str]]: + async def get_title_choices(self, interaction: Interaction, current: str) -> List[app_commands.Choice[str]]: + # Fetch the latest available titles from the database titles = StaffingAsync._get_titles() - return [app_commands.Choice(name=title, value=title) for title in titles] + + # Return titles that match the user's input + return [ + app_commands.Choice(name=title, value=title) + for title in titles if current.lower() in title.lower() + ] + + # Autocomplete function to fetch and filter titles in real-time + async def avail_title_autocomplete(self, interaction: Interaction, current: str): + # Fetch the latest available titles from the database + titles = StaffingAsync._get_avail_titles() + + # Return titles that match the user's input + return [ + app_commands.Choice(name=title, value=title) + for title in titles if current.lower() in title.lower() + ] @app_commands.command(name="setupstaffing", description="Bot setups staffing information") @app_commands.describe(title="What should the title of the staffing be?", week_int="What should the week interval be? eg. 1 then the date will be selected each week.", section_amount="What should the section amount be? eg. 3 then there will be 3 sections.", restrict_booking="Should the staffing restrict booking to first section before allowing other sections too?") + @app_commands.autocomplete(title=get_title_choices) @app_commands.checks.has_any_role(*staff_roles()) async def setup_staffing(self, interaction: Interaction, title: str, week_int: app_commands.Range[int, 1, 4], section_amount: app_commands.Range[int, 1, 4], restrict_booking: Literal["Yes", "No"], channel: TextChannel): ctx: commands.Context = await self.bot.get_context(interaction) @@ -100,23 +118,6 @@ async def setup_staffing(self, interaction: Interaction, title: str, week_int: a DB.insert(self=self, table="positions", columns=['position', 'user', 'type', 'local_booking', 'start_time', 'end_time', 'event'], values=[pos, "", j, local[section_positions[x][pos]['local_booking']], section_positions[x][pos]['start_time'], section_positions[x][pos]['end_time'], event]) j += 1 - # Link this command to custom autocomplete for titles - @setup_staffing.autocomplete("title") - async def title_autocomplete(self, interaction: Interaction, current: str): - titles = await self.get_title_choices() - return [choice for choice in titles if current.lower() in choice.name.lower()][:25] # Limit to 25 choices (Discord limit) - - # Autocomplete function to fetch and filter titles in real-time - async def avail_title_autocomplete(self, interaction: Interaction, current: str): - # Fetch the latest available titles from the database - titles = StaffingAsync._get_avail_titles() - - # Return titles that match the user's input - return [ - app_commands.Choice(name=title, value=title) - for title in titles if current.lower() in title.lower() - ] - @app_commands.command(name="refreshevent", description="Bot refreshes selected event") @app_commands.describe(title="Which staffing would you like to refresh?") @app_commands.autocomplete(title=avail_title_autocomplete) diff --git a/cogs/tasks.py b/cogs/tasks.py index a330048..bbb8602 100644 --- a/cogs/tasks.py +++ b/cogs/tasks.py @@ -100,7 +100,7 @@ async def user_check(self, interaction: discord.Integration): async def sync_commands(self, override=False): now = datetime.now().isoformat() - guild = self.bot.get_guild(GUILD_ID) + guild = discord.Object(id=GUILD_ID) try: if DEBUG == True and override == False: @@ -111,7 +111,6 @@ async def sync_commands(self, override=False): try: if guild: await self.bot.tree.sync(guild=guild) # Sync commands to a specific guild for faster deployment - print(f'Commands has been synced for guild {guild}', flush=True) else: await self.bot.tree.sync() # Sync global commands (might take up to 1 hour to reflect globally) except discord.HTTPException as e: