chore: initial commit

core/test_text_search.py

@@ -0,0 +1,73 @@
+#!/usr/bin/env python3
+"""测试 funstat text 搜索功能"""
+
+import asyncio
+import time
+from datetime import datetime
+from telethon import TelegramClient
+
+# 配置
+API_ID = 24660516
+API_HASH = "eae564578880a59c9963916ff1bbbd3a"
+SESSION_PATH = "/Users/lucas/telegram_sessions/funstat_bot"
+BOT_USERNAME = "@openaiw_bot"
+
+
+async def send_command_and_wait(client, bot_entity, command, timeout=15):
+    """发送命令到 BOT 并等待响应"""
+    print(f"📤 发送命令: {command}")
+
+    # 记录发送前的最新消息 ID
+    last_message_id = 0
+    async for message in client.iter_messages(bot_entity, limit=1):
+        last_message_id = message.id
+        break
+
+    # 发送消息
+    await client.send_message(bot_entity, command)
+
+    # 等待响应
+    await asyncio.sleep(2)
+
+    # 获取新消息
+    start_time = time.time()
+    while time.time() - start_time < timeout:
+        async for message in client.iter_messages(bot_entity, limit=5):
+            if message.id > last_message_id:
+                if not message.out and message.text:
+                    print(f"\n✅ 收到响应 ({len(message.text)} 字符):\n")
+                    print("=" * 60)
+                    print(message.text)
+                    print("=" * 60)
+                    return message.text
+
+        await asyncio.sleep(0.5)
+
+    raise TimeoutError(f"等待 BOT 响应超时 ({timeout}秒)")
+
+
+async def main():
+    """主函数"""
+    print("初始化 Telegram 客户端...")
+
+    # 创建客户端
+    client = TelegramClient(SESSION_PATH, API_ID, API_HASH)
+    await client.start()
+
+    # 获取 bot 实体
+    bot_entity = await client.get_entity(BOT_USERNAME)
+    print(f"✅ 已连接到: {bot_entity.first_name}\n")
+
+    # 发送 text 搜索命令
+    try:
+        await send_command_and_wait(client, bot_entity, "/text 翻译")
+    except Exception as e:
+        print(f"❌ 错误: {e}")
+
+    # 断开连接
+    await client.disconnect()
+    print("\n✅ 完成")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())