Pyrogram 中正确处理和转发媒体组消息的教程

pip install pyrogram pyyaml

id: YOUR_API_ID
hash: YOUR_API_HASH
channels:
  - SOURCE_CHAT_ID_1 # 来源聊天ID，可以是数字或用户名
  - SOURCE_CHAT_ID_2
my_channel: TARGET_CHANNEL_ID # 目标频道ID

import pyrogram
from pyrogram import filters
import yaml
import asyncio

# 加载配置
try:
    with open("config.yaml", "r") as f:
        config = yaml.safe_load(f)
except FileNotFoundError:
    print("Error: config.yaml not found. Please create one with your API ID, API HASH, and channel IDs.")
    exit()

api_id = config["id"]
api_hash = config["hash"]
source_channels = config["channels"]
target_channel = config["my_channel"]

# 初始化 Pyrogram 客户端
app = pyrogram.Client(
    'my_account',
    api_id=api_id,
    api_hash=api_hash
)

# 用于存储已处理的媒体组ID，避免重复处理
processed_media_groups = set()

@app.on_message(filters.chat(source_channels))
async def copy_to_my_channel(client, message):
    # 如果消息没有 media_group_id，则它是单条消息
    if message.media_group_id is None:
        print(f"Received single message (ID: {message.id}). Copying...")
        await client.copy_message(
            from_chat_id=message.chat.id,
            chat_id=target_channel,
            message_id=message.id
        )
        print(f"Single message (ID: {message.id}) copied.")
    else:
        # 这是一个媒体组消息
        media_group_id = message.media_group_id

        # 检查是否已经处理过这个媒体组，如果是，则跳过
        if media_group_id in processed_media_groups:
            # print(f"Media group {media_group_id} already processed. Skipping message {message.id}.")
            return

        print(f"Received message (ID: {message.id}) belonging to media group {media_group_id}.")

        # 获取媒体组中的所有消息
        # 注意：get_media_group 可能会在媒体组消息尚未完全到达时被调用，
        # 导致返回不完整的列表。可以添加一个短暂停顿或重试机制来提高健壮性。
        await asyncio.sleep(0.5) # 给Telegram一点时间来聚合所有消息

        full_media_group = await client.get_media_group(
            chat_id=message.chat.id,
            message_id=message.id # 任何属于该媒体组的消息ID都可以
        )

        # 查找当前消息在完整媒体组中的索引
        current_message_index = -1
        for i, msg in enumerate(full_media_group):
            if msg.id == message.id:
                current_message_index = i
                break

        # 只有当当前消息是媒体组中的最后一条消息时，才执行转发
        # 这里的判断逻辑需要更严谨，因为消息到达的顺序不总是线性的。
        # 最可靠的方法是等待一段时间，然后一次性处理。
        # 但为了演示，我们先使用这种基于“最后一条消息”的简化逻辑。
        # 更好的做法是收集所有消息ID，然后在检测到“完成”时发送。
        # 对于Pyrogram的get_media_group，它通常会返回完整的组，
        # 所以我们只需确保只执行一次copy_media_group。

        # 标记此媒体组为已处理，防止后续触发重复发送
        processed_media_groups.add(media_group_id)

        # 提取所有消息的ID
        message_ids_to_copy = [msg.id for msg in full_media_group]

        print(f"Detected complete media group {media_group_id} with IDs: {message_ids_to_copy}. Copying...")

        # 使用 copy_media_group 转发整个媒体组
        await client.copy_media_group(
            chat_id=target_channel,
            from_chat_id=message.chat.id,
            message_ids=message_ids_to_copy
        )
        print(f"Media group {media_group_id} published to {target_channel}.")

# 运行客户端
if __name__ == '__main__':
    print("Bot started. Listening for messages...")
    app.run()