修正個人檔案錯誤處理

Author: IanDesuyo

cogs/profileCard/__init__.py

@@ -1,4 +1,5 @@
 import logging
+import aiohttp
 from discord_slash.context import ComponentContext
 from discord_slash.model import ButtonStyle
 from discord_slash.utils.manage_components import create_actionrow, create_button
@@ -41,7 +42,10 @@ async def get_profile(self, server: int, uid: int, cache: bool = True):
             f"{self.apiUrl}/profile",
             params={"server": server, "uid": str(uid).zfill(9), "cache": "true" if cache else "false"},
         ) as resp:
-            data = await resp.json()
+            try:
+                data = await resp.json()
+            except:
+                raise errors.GameApiError
 
         self.logger.info(f"get_profile /{server}/{uid:09d}: {resp.status}")
 

main.py

@@ -15,7 +15,7 @@
 from utils.state_manger import StateManger, State
 
 __author__ = "IanDesuyo"
-__version__ = "3.1.8"
+__version__ = "3.1.9"
 
 
 class AIKyaru(AutoShardedBot):

utils/errors.py

@@ -10,6 +10,7 @@
 import uuid
 import utils
 import traceback
+import asyncio
 
 
 class GameApiError(Exception):
@@ -56,6 +57,10 @@ class RecordDeleted(Exception):
     pass
 
 
+class GameProfileNotLinked(Exception):
+    pass
+
+
 async def handle_error(bot: Bot, ctx: Union[Context, SlashContext, ComponentContext], error):
     if isinstance(ctx, ComponentContext):
         # don't send new message, edit the origin one.
@@ -78,9 +83,14 @@ async def do_edit_origin(content: str = "", embed=None, components=None):
         return await ctx.send(f"我沒有權限... 需要`{','.join(error.missing_perms)}`")
     if isinstance(error, commands.NoPrivateMessage):
         return await ctx.send("此功能僅限群組使用")
+    if isinstance(error, asyncio.TimeoutError):
+        return await ctx.send("與伺服器連線超時 :(")
+
 
     # game api errors
 
+    if isinstance(error, GameProfileNotLinked):
+        return await ctx.send("此用戶尚未綁定遊戲ID")
     if isinstance(error, ProfileNotFound):
         return ctx.send("找不到此玩家... 你還記得你的ID嗎?")
     if isinstance(error, GameApiError):