init

wechat-bot/main.py

@@ -0,0 +1,86 @@
+"""
+Wechat-Bot 主入口
+启动企业微信 WebSocket 长连接，接收消息后转发给远程 Agent。
+"""
+import asyncio
+import logging
+import sys
+
+from config import Config
+from bot.wecom_bot import create_bot_client
+from bot.agent_client import close_http_client
+
+# 配置日志
+logging.basicConfig(
+    level=getattr(logging, Config.LOG_LEVEL, logging.INFO),
+    format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
+    datefmt="%Y-%m-%d %H:%M:%S",
+)
+logger = logging.getLogger(__name__)
+
+
+async def run_bot():
+    """启动企业微信机器人 WebSocket 客户端"""
+    # 创建机器人客户端并注册事件处理器
+    ws_client = create_bot_client()
+    logger.info("正在连接企业微信...")
+
+    # 用 Event 等待认证完成
+    authenticated = asyncio.Event()
+
+    @ws_client.on("authenticated")
+    def _on_authenticated():
+        logger.info("企业微信认证成功，机器人已上线")
+        authenticated.set()
+
+    @ws_client.on("error")
+    def _on_error(err):
+        logger.error("SDK 错误: %s", err)
+
+    try:
+        await ws_client.connect()
+        # 等待认证结果（最多 30 秒）
+        try:
+            await asyncio.wait_for(authenticated.wait(), timeout=30)
+        except asyncio.TimeoutError:
+            logger.error("认证超时（30秒），请检查 BOT_ID 和 BOT_SECRET 是否正确")
+            return
+
+        # 认证成功，保持事件循环运行以持续接收消息
+        logger.info("正在监听消息...")
+        await asyncio.Future()  # 永远挂起，直到事件循环被取消
+
+    except asyncio.CancelledError:
+        logger.info("任务被取消")
+    except Exception as e:
+        logger.error("企业微信连接异常: %s", e)
+    finally:
+        await close_http_client()
+        logger.info("机器人已停止")
+
+
+def main():
+    """主函数"""
+    # 校验配置
+    missing = Config.validate()
+    if missing:
+        logger.error("缺少必要配置项: %s", ", ".join(missing))
+        logger.error("请复制 .env.example 为 .env 并填写配置")
+        sys.exit(1)
+
+    logger.info("=" * 50)
+    logger.info("  Wechat-Bot 启动中")
+    logger.info("  Agent 地址: %s", Config.AGENT_BASE_URL)
+    logger.info("  日志级别: %s", Config.LOG_LEVEL)
+    logger.info("=" * 50)
+
+    try:
+        asyncio.run(run_bot())
+    except KeyboardInterrupt:
+        logger.info("收到中断信号，正在停止...")
+
+    logger.info("Wechat-Bot 已退出")
+
+
+if __name__ == "__main__":
+    main()