fix commands

app/modules/telegram/__init__.py

@@ -8,12 +8,14 @@ from app.modules.telegram.telegram import Telegram
 
 
 class TelegramModule(_ModuleBase):
-
     telegram: Telegram = None
 
     def init_module(self) -> None:
         self.telegram = Telegram()
 
+    def stop(self):
+        self.telegram.stop()
+
     def init_setting(self) -> Tuple[str, Union[str, bool]]:
         return "MESSAGER", "telegram"
 
@@ -51,13 +53,16 @@ class TelegramModule(_ModuleBase):
                 }
             }
         """
+        # 校验token
+        token = args.get("token")
+        if not token or token != settings.API_TOKEN:
+            return None
         try:
-            msg_json: dict = json.loads(body)
+            message: dict = json.loads(body)
         except Exception as err:
             logger.error(f"解析Telegram消息失败：{err}")
             return None
-        if msg_json:
-            message = msg_json.get("message", {})
+        if message:
             text = message.get("text")
             user_id = message.get("from", {}).get("id")
             # 获取用户名
@@ -117,3 +122,10 @@ class TelegramModule(_ModuleBase):
         :return: 成功或失败
         """
         return self.telegram.send_torrents_msg(title=title, torrents=items, userid=userid)
+
+    def register_commands(self, commands: dict):
+        """
+        注册命令，实现这个函数接收系统可用的命令菜单
+        :param commands: 命令字典
+        """
+        self.telegram.register_commands(commands)