From 9f381ab1a959f4ce594825275865adeedcbb5633 Mon Sep 17 00:00:00 2001
From: UnCLAS-Prommer <gitunclasprommer@gmail.com>
Date: Tue, 13 May 2025 20:12:14 +0800
Subject: [PATCH] refactor

---
 main.py             |  6 +-----
 src/config.py       |  2 +-
 src/send_handler.py | 10 +++++++---
 3 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/main.py b/main.py
index c3d96a9..f15fb81 100644
--- a/main.py
+++ b/main.py
@@ -17,11 +17,7 @@ async def message_recv(server_connection: Server.ServerConnection):
         logger.debug(f"{raw_message[:80]}..." if len(raw_message) > 80 else raw_message)
         decoded_raw_message: dict = json.loads(raw_message)
         post_type = decoded_raw_message.get("post_type")
-        if post_type == "meta_event":
-            await message_queue.put(decoded_raw_message)
-        elif post_type == "message":
-            await message_queue.put(decoded_raw_message)
-        elif post_type == "notice":
+        if post_type in ["meta_event","message","notice"]:
             await message_queue.put(decoded_raw_message)
         elif post_type is None:
             await put_response(decoded_raw_message)
diff --git a/src/config.py b/src/config.py
index 920b6d6..3be3d5a 100644
--- a/src/config.py
+++ b/src/config.py
@@ -22,7 +22,7 @@ class Config:
         self.root_path = os.path.join(src_path, "..")
         self.config_path = os.path.join(self.root_path, "config.toml")
 
-    def load_config(self):
+    def load_config(self):  # sourcery skip: extract-method, move-assign
         include_configs = ["Nickname", "Napcat_Server", "MaiBot_Server", "Debug", "Voice"]
         if os.path.exists(self.config_path):
             with open(self.config_path, "rb") as f:
diff --git a/src/send_handler.py b/src/send_handler.py
index 247b512..b7f7d07 100644
--- a/src/send_handler.py
+++ b/src/send_handler.py
@@ -2,7 +2,7 @@ import json
 import websockets as Server
 import uuid
 
-# from .config import global_config
+from .config import global_config
 # 白名单机制不启用
 from .message_queue import get_response
 from .logger import logger
@@ -86,6 +86,7 @@ class SendHandler:
         return payload
 
     def process_message_by_type(self, seg: Seg, payload: list) -> list:
+        # sourcery skip: reintroduce-else, swap-if-else-branches, use-named-expression
         new_payload = payload
         if seg.type == "reply":
             target_id = seg.data
@@ -111,6 +112,7 @@ class SendHandler:
         return new_payload
 
     def build_payload(self, payload: list, addon: dict, is_reply: bool = False) -> list:
+        # sourcery skip: for-append-to-extend, merge-list-append, simplify-generator
         """构建发送的消息体"""
         if is_reply:
             temp_list = []
@@ -128,8 +130,7 @@ class SendHandler:
 
     def handle_text_message(self, message: str) -> dict:
         """处理文本消息"""
-        ret = {"type": "text", "data": {"text": message}}
-        return ret
+        return {"type": "text", "data": {"text": message}}
 
     def handle_image_message(self, encoded_image: str) -> dict:
         """处理图片消息"""
@@ -158,6 +159,9 @@ class SendHandler:
 
     def handle_voice_message(self, encoded_voice: str) -> dict:
         """处理语音消息"""
+        if not global_config.use_tts:
+            logger.warning("未启用语音消息处理")
+            return {}
         if not encoded_voice:
             return {}
         return {