refacotr: 清理# type ignore, 处理self.ws 可能为None的情况

Author: shuiping233

astrbot/core/platform/sources/kook/kook_client.py

@@ -1,6 +1,5 @@
 import asyncio
 import base64
-import json
 import os
 import random
 import time
@@ -188,7 +187,12 @@ async def listen(self):
         try:
             while self.running:
                 try:
-                    msg = await asyncio.wait_for(self.ws.recv(), timeout=10)  # type: ignore
+                    if self.ws is None:
+                        logger.warning(
+                            "[KOOK] 尚未连接kook WebSocket服务器, 跳过消息监听流程"
+                        )
+                        continue
+                    msg = await asyncio.wait_for(self.ws.recv(), timeout=10)
 
                     if isinstance(msg, bytes):
                         try:
@@ -326,9 +330,16 @@ async def _heartbeat_loop(self):
 
     async def _send_ping(self):
         """发送心跳PING"""
+        if self.ws is None:
+            logger.warning("[KOOK] 尚未连接kook WebSocket服务器, 跳过发送心跳包流程")
+            return
         try:
-            ping_data = {"s": 2, "sn": self.last_sn}
-            await self.ws.send(json.dumps(ping_data))  # type: ignore
+            ping_data = KookWebsocketEvent(
+                s=KookMessageSignal.PING,
+                d=None,
+                sn=self.last_sn,
+            )
+            await self.ws.send(ping_data.to_json())
         except Exception as e:
             logger.error(f"[KOOK] 发送心跳失败: {e}")