151 lines
4.3 KiB
Python
151 lines
4.3 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
Telethon Session 创建脚本
|
||
|
||
这个脚本会帮助你创建 Telegram session 文件
|
||
运行后按照提示操作即可
|
||
"""
|
||
|
||
import asyncio
|
||
from telethon import TelegramClient
|
||
from telethon.errors import SessionPasswordNeededError
|
||
|
||
# 你的 API 凭证
|
||
API_ID = 24660516
|
||
API_HASH = "eae564578880a59c9963916ff1bbbd3a"
|
||
|
||
# Session 文件名
|
||
SESSION_NAME = "funstat_bot_session"
|
||
|
||
async def create_session():
|
||
"""创建 Telegram session 文件"""
|
||
|
||
print("=" * 60)
|
||
print("🚀 Telegram Session 创建工具")
|
||
print("=" * 60)
|
||
print()
|
||
|
||
# 创建客户端
|
||
client = TelegramClient(SESSION_NAME, API_ID, API_HASH)
|
||
|
||
print("📱 正在连接到 Telegram...")
|
||
await client.connect()
|
||
|
||
if not await client.is_user_authorized():
|
||
print()
|
||
print("=" * 60)
|
||
print("需要登录到你的 Telegram 账号")
|
||
print("=" * 60)
|
||
print()
|
||
|
||
# 请求手机号
|
||
phone = input("请输入你的手机号 (格式: +86xxxxxxxxxx): ")
|
||
|
||
try:
|
||
await client.send_code_request(phone)
|
||
print()
|
||
print("✅ 验证码已发送到你的 Telegram 客户端")
|
||
print(" (请检查你的 Telegram 应用)")
|
||
print()
|
||
|
||
# 请求验证码
|
||
code = input("请输入收到的验证码: ")
|
||
|
||
try:
|
||
await client.sign_in(phone, code)
|
||
except SessionPasswordNeededError:
|
||
# 如果账号设置了两步验证
|
||
print()
|
||
print("⚠️ 你的账号启用了两步验证")
|
||
password = input("请输入你的两步验证密码: ")
|
||
await client.sign_in(password=password)
|
||
|
||
except Exception as e:
|
||
print(f"❌ 登录失败: {e}")
|
||
await client.disconnect()
|
||
return False
|
||
|
||
# 验证登录成功
|
||
me = await client.get_me()
|
||
print()
|
||
print("=" * 60)
|
||
print("✅ 登录成功!")
|
||
print("=" * 60)
|
||
print(f"账号信息:")
|
||
print(f" - 用户名: @{me.username if me.username else '未设置'}")
|
||
print(f" - 姓名: {me.first_name} {me.last_name if me.last_name else ''}")
|
||
print(f" - 手机号: {me.phone}")
|
||
print(f" - ID: {me.id}")
|
||
print()
|
||
print(f"✅ Session 文件已创建: {SESSION_NAME}.session")
|
||
print()
|
||
|
||
# 测试与 @openaiw_bot 的连接
|
||
print("=" * 60)
|
||
print("🔍 正在测试与 @openaiw_bot 的连接...")
|
||
print("=" * 60)
|
||
|
||
try:
|
||
# 查找 bot
|
||
bot_entity = await client.get_entity("@openaiw_bot")
|
||
print(f"✅ 找到 BOT: {bot_entity.first_name}")
|
||
print(f" BOT ID: {bot_entity.id}")
|
||
print()
|
||
|
||
# 发送测试消息
|
||
print("📤 发送测试消息: /start")
|
||
await client.send_message(bot_entity, "/start")
|
||
|
||
print("⏳ 等待 BOT 响应 (最多 10 秒)...")
|
||
|
||
# 等待响应
|
||
async def wait_for_response():
|
||
async for message in client.iter_messages(bot_entity, limit=1):
|
||
return message
|
||
|
||
try:
|
||
response = await asyncio.wait_for(wait_for_response(), timeout=10.0)
|
||
if response:
|
||
print()
|
||
print("=" * 60)
|
||
print("✅ 收到 BOT 响应:")
|
||
print("=" * 60)
|
||
print(response.text[:500]) # 显示前500个字符
|
||
print()
|
||
if len(response.text) > 500:
|
||
print(f"... (还有 {len(response.text) - 500} 个字符)")
|
||
print()
|
||
except asyncio.TimeoutError:
|
||
print("⚠️ 10秒内未收到响应，但连接正常")
|
||
print()
|
||
|
||
except Exception as e:
|
||
print(f"❌ 测试失败: {e}")
|
||
print()
|
||
|
||
await client.disconnect()
|
||
|
||
print("=" * 60)
|
||
print("🎉 完成!")
|
||
print("=" * 60)
|
||
print()
|
||
print(f"Session 文件位置: ./{SESSION_NAME}.session")
|
||
print("现在你可以在 MCP 服务器中使用这个 session 文件了")
|
||
print()
|
||
|
||
return True
|
||
|
||
if __name__ == "__main__":
|
||
# 检查是否已安装 telethon
|
||
try:
|
||
import telethon
|
||
print(f"✅ Telethon 版本: {telethon.__version__}")
|
||
print()
|
||
except ImportError:
|
||
print("❌ 未安装 Telethon")
|
||
print("请先运行: pip install telethon")
|
||
exit(1)
|
||
|
||
# 运行创建流程
|
||
asyncio.run(create_session())
|