# 安装核心库
# pip install python-telegram-bot

import logging
from telegram import Update
from telegram.ext import Application, CommandHandler, MessageHandler, filters, CallbackContext

# 启用日志
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
logger = logging.getLogger(__name__)

# 替换为你的Bot Token
TOKEN = "YOUR_BOT_TOKEN_HERE"

async def start(update: Update, context: CallbackContext) -> None:
    """处理 /start 命令"""
    user = update.effective_user
    await update.message.reply_html(f"你好，{user.mention_html()}！\n我是你的测试助手Bot。\n发送 /help 查看可用命令。")

async def help_command(update: Update, context: CallbackContext) -> None:
    """处理 /help 命令"""
    help_text = """
    可用命令：
    /start - 开始使用
    /help - 显示此帮助信息
    /echo <文本> - 回显你发送的文本
    直接发送任何消息，我会重复它。
    """
    await update.message.reply_text(help_text)

async def echo(update: Update, context: CallbackContext) -> None:
    """回显用户消息"""
    # 提取命令后的文本
    text_to_echo = ' '.join(context.args)
    if not text_to_echo:
        text_to_echo = update.message.text
    await update.message.reply_text(f"你说了：{text_to_echo}")

def main() -> None:
    """主函数，启动Bot"""
    # 创建Application实例
    application = Application.builder().token(TOKEN).build()

    # 注册命令处理器
    application.add_handler(CommandHandler("start", start))
    application.add_handler(CommandHandler("help", help_command))
    application.add_handler(CommandHandler("echo", echo))

    # 注册一个消息处理器，处理非命令的普通文本消息
    application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, echo))

    # 启动Bot（使用长轮询模式）
    application.run_polling(allowed_updates=Update.ALL_TYPES)

if __name__ == '__main__':
    main()

from telegram import InlineKeyboardButton, InlineKeyboardMarkup

async def menu(update: Update, context: CallbackContext) -> None:
    """发送一个带内联键盘的菜单"""
    keyboard = [
        [InlineKeyboardButton("选项 1", callback_data='1'),
         InlineKeyboardButton("选项 2", callback_data='2')],
        [InlineKeyboardButton("关于", callback_data='about')],
    ]
    reply_markup = InlineKeyboardMarkup(keyboard)
    await update.message.reply_text('请选择一个选项：', reply_markup=reply_markup)

async def button_callback(update: Update, context: CallbackContext) -> None:
    """处理内联键盘按钮的回调查询"""
    query = update.callback_query
    await query.answer() # 确认回调，防止客户端旋转加载图标
    data = query.data
    if data == '1':
        text = "你选择了选项 1！"
    elif data == '2':
        text = "你选择了选项 2！"
    elif data == 'about':
        text = "这是一个演示Bot。"
    # 编辑原始消息，更新文本并移除键盘
    await query.edit_message_text(text=text)

# 在主函数中注册处理器
# application.add_handler(CommandHandler("menu", menu))
# application.add_handler(CallbackQueryHandler(button_callback))

from telegram import InlineQueryResultArticle, InputTextMessageContent

async def inline_query(update: Update, context: CallbackContext) -> None:
    """处理内联查询"""
    query = update.inline_query.query
    if not query:
        return
    results = [
        InlineQueryResultArticle(
            id='1',
            title="回显",
            input_message_content=InputTextMessageContent(f"内联查询结果：{query}"),
            description=f"回显你输入的内容：{query}",
        )
    ]
    await update.inline_query.answer(results, cache_time=1)

# 在主函数中注册内联查询处理器
# application.add_handler(InlineQueryHandler(inline_query))

async def handle_document(update: Update, context: CallbackContext) -> None:
    """处理用户发送的文档"""
    document = update.message.document
    file = await document.get_file()
    file_path = f"downloads/{document.file_name}"
    await file.download_to_drive(file_path)
    await update.message.reply_text(f"文件 '{document.file_name}' 已接收并保存。")

# 注册文档处理器
# application.add_handler(MessageHandler(filters.DOCUMENT, handle_document))

from apscheduler.schedulers.asyncio import AsyncIOScheduler
from apscheduler.triggers.cron import CronTrigger

async def send_daily_reminder(context: CallbackContext) -> None:
    """定时发送提醒"""
    job_context = context.job
    chat_id = job_context.data
    await context.bot.send_message(chat_id=chat_id, text="⏰ 每日提醒：该喝水休息一下了！")

async def set_reminder(update: Update, context: CallbackContext) -> None:
    """用户命令设置每日提醒"""
    chat_id = update.effective_chat.id
    # 添加一个每天上午10点执行的任务
    context.application.job_queue.run_daily(send_daily_reminder, time=datetime.time(hour=10, minute=0), data=chat_id)
    await update.message.reply_text("已设置每日10点提醒。")

# 在主函数中初始化调度器并注册命令
# application.job_queue.scheduler = AsyncIOScheduler()
# application.add_handler(CommandHandler("setreminder", set_reminder))

import aiohttp

async def weather(update: Update, context: CallbackContext) -> None:
    """查询天气"""
    city = ' '.join(context.args) or "Beijing"
    api_key = "YOUR_WEATHER_API_KEY"
    url = f"http://api.openweathermap.org/data/2.5/weather?q={city}&appid={api_key}&units=metric"

    async with aiohttp.ClientSession() as session:
        async with session.get(url) as resp:
            if resp.status == 200:
                data = await resp.json()
                temp = data['main']['temp']
                desc = data['weather'][0]['description']
                reply = f"{city}的天气：{desc}，温度 {temp}°C。"
            else:
                reply = "无法获取天气信息。"
    await update.message.reply_text(reply)

# 使用内存字典的简单示例
user_states = {}

async def start_survey(update: Update, context: CallbackContext) -> None:
    chat_id = update.effective_chat.id
    user_states[chat_id] = {'step': 1, 'answers': {}}
    await update.message.reply_text("问题1：你的名字是？")

async def handle_survey_answer(update: Update, context: CallbackContext) -> None:
    chat_id = update.effective_chat.id
    state = user_states.get(chat_id)
    if not state:
        return
    answer = update.message.text
    if state['step'] == 1:
        state['answers']['name'] = answer
        state['step'] = 2
        await update.message.reply_text("问题2：你的年龄是？")
    elif state['step'] == 2:
        state['answers']['age'] = answer
        # 完成调查，处理答案
        await update.message.reply_text(f"感谢参与，{state['answers']['name']}！")
        del user_states[chat_id] # 清除状态

# 需要更精细的消息路由来调用此处理器

# 假设使用FastAPI作为Web框架
from fastapi import FastAPI, Request
from telegram import Update
from telegram.ext import Application, CallbackContext

app = FastAPI()
# 初始化你的Bot Application，注意这里不调用 run_polling
application = Application.builder().token(TOKEN).build()
# ... 注册所有处理器 ...

@app.post("/webhook")
async def webhook(request: Request):
    """接收Telegram推送的更新"""
    json_data = await request.json()
    update = Update.de_json(json_data, application.bot)
    await application.process_update(update)
    return {"ok": True}

# 部署后，你需要调用Telegram Bot API的 `setWebhook` 方法，将你的公开HTTPS URL（如 https://yourdomain.com/webhook）设置上去。
# 可以使用 curl 或编写一个简单的脚本：
# curl -F "url=https://yourdomain.com/webhook" https://api.telegram.org/bot<YOUR_TOKEN>/setWebhook