feat: 計算処理を非同期に変更し、タイムアウトエラーハンドリングを追加

Author: techfish-11

cogs/commands/calculator.py

@@ -2,6 +2,7 @@
 from discord.ext import commands
 from discord import app_commands
 import numpy as np
+import asyncio
 
 class Calculator(commands.Cog):
     def __init__(self, bot):
@@ -10,19 +11,30 @@ def __init__(self, bot):
     @app_commands.command(name="calculator", description="数式を計算します")
     @app_commands.describe(expression="計算したい数式を入力してください")
     async def calculator(self, interaction: discord.Interaction, expression: str):
-            await interaction.response.defer(thinking=True)
-            allowed_names = {k: v for k, v in np.__dict__.items() if not k.startswith("__")}
-            allowed_names.update({"abs": abs, "round": round})
-            result = eval(expression, {"__builtins__": {}}, allowed_names)
-
+        await interaction.response.defer(thinking=True)
+        allowed_names = {k: v for k, v in np.__dict__.items() if not k.startswith("__")}
+        allowed_names.update({"abs": abs, "round": round})
+        
+        try:
+            result = await asyncio.wait_for(
+                asyncio.to_thread(eval, expression, {"__builtins__": {}}, allowed_names),
+                timeout=5.0
+            )
             embed = discord.Embed(
                 title="計算結果",
                 description=f"数式: `{expression}`",
                 color=discord.Color.green()
             )
             embed.add_field(name="結果", value=f"`{result}`", inline=False)
             embed.set_footer(text="計算完了 🎉")
-            await interaction.followup.send(embed=embed)
+        except asyncio.TimeoutError:
+            embed = discord.Embed(
+                title="エラー",
+                description="計算がタイムアウトしました。5秒以内に完了しませんでした。",
+                color=discord.Color.red()
+            )
+        
+        await interaction.followup.send(embed=embed)
 
 async def setup(bot):
     await bot.add_cog(Calculator(bot))