Add PEP slash command

bot/exts/info/pep.py

@@ -1,12 +1,13 @@
 from datetime import UTC, datetime, timedelta
 from typing import TypedDict
+from rapidfuzz import process
 
-from discord import Colour, Embed
+from discord import Colour, Embed, app_commands, Interaction
 from discord.ext.commands import Cog, Context, command
 
 from bot.bot import Bot
 from bot.log import get_logger
 from bot.utils.messages import send_or_reply
 
 log = get_logger(__name__)
 
@@ -72,10 +73,7 @@
 
         return embed
 
-    @command(name="pep", aliases=("get_pep", "p"))
-    async def pep_command(self, ctx: Context, pep_number: int) -> None:
-        """Fetches information about a PEP and sends it to the channel."""
-        # Refresh the PEP data up to every hour, as e.g. the PEP status might have changed.
+    async def get_pep_embed(self, pep_number: int) -> Embed:
         if (
             self.last_refreshed_peps is None or (
                 (self.last_refreshed_peps + timedelta(hours=1)) <= datetime.now(tz=UTC)
@@ -93,9 +91,34 @@
                 description=f"PEP {pep_number} does not exist.",
                 colour=Colour.red(),
             )
+        return embed
 
+    @command(name="pep", aliases=("get_pep", "p"))
+    async def pep_command(self, ctx: Context, pep_number: int) -> None:
+        """Fetches information about a PEP and sends it to the channel."""
+        # Refresh the PEP data up to every hour, as e.g. the PEP status might have changed.
+        embed = await self.get_pep_embed(pep_number)
         await send_or_reply(ctx, embed)
 
+    @app_commands.command(name="pep")
+    @app_commands.guild_only()
+    @app_commands.describe(pep_number="The pep number or the autocompleted pep")
+    async def pep_slash_command(self, interaction: Interaction, pep_number: int) -> bool:
+        embed = await self.get_pep_embed(pep_number)
+        await interaction.response.send_message(embed=embed)
+
+    @pep_slash_command.autocomplete("pep_number")
+    async def pep_slash_command_autocomplete(self, interaction: Interaction, current: str) -> list[app_commands.Choice]:
+        if self.last_refreshed_peps is None:
+            await self.refresh_pep_data()
+
+        choices = {pep_id: pep_info["title"] for pep_id, pep_info in self.peps.items()}
+
+        # list[('pep_title', similarity, pep_number)]
+        result = process.extract(query=current, choices=choices, limit=10)
+        return [app_commands.Choice(name=f"{pep[2]} - {pep[0]}", value=pep[2]) for pep in result]
+
+
 async def setup(bot: Bot) -> None:
     """Load the PEP cog."""
     await bot.add_cog(PythonEnhancementProposals(bot))