Как узнать, что отправлен последний файл медиагруппы в Телеграм?

Question

[EugeneVKruglov]

Пользователь в ответ на запрос бота может отправить в чат (или переслать из другого чата) не одно фото или видео, а сразу несколько - в виде альбома (медиагруппы). Телеграм представляет их в виде отдельных сообщений с общим идентификатором (media_group_id). Каждое сообщение содержит свой файл. Как мне узнать, что загружено последнее сообщение (файл) из медиагруппы, и только после этого предпринимать дальнейшие действия?
Сохранять сообщения и восстанавливать медиагруппу я умею, проблема только в том, как понять, что все файлы отправлены.
Работаю с pyTelegramBotApi.

Comments

[Дмитрий Касаткин]

Никак. Это одно из проблемных мест телеграм. Как вариант можно ждать какое-то время и если по истечении ничего больше не пришло то работать дальше.

Answer 1

[EugeneVKruglov]

Решил вопрос так: через bot.middleware_handler отлавливаю первое сообщение медиагруппы, после чего в этом хэндлере продолжаю работу нужных функций. Остальные сообщения приходят, но в эти функции не попадают и обрабатываются отдельно. При этом можно не ждать, когда придут все сообщения из медиагруппы.

Answer 2

[Foldren]

Вот так я проверяю на последний файл, нигде не нашел решения пришел сам в итоге

async def check_last_media_file(message: Message):
        last_message_flag = False
        try:
            await message.bot.edit_message_text(
                text="  ",
                chat_id=message.from_user.id,
                message_id=message.message_id + 1
            )
        except TelegramBadRequest as exc:
            if (exc.message == "Bad Request: message must be non-empty" or
                    exc.message == "Bad Request: message to edit not found"):
                last_message_flag = True
        return last_message_flag

Comments

[timplay42]

Foldren, спасибо. Решил адаптировать под middelware

import asyncio
from typing import *
from aiogram import BaseMiddleware
from aiogram.exceptions import TelegramBadRequest
from aiogram.types import Message, CallbackQuery

class MediaMiddleware(BaseMiddleware):
    def __init__(self, latency: Union[int, float] = 0.6):
        self.medias = {}
        self.latency = latency
        super(MediaMiddleware, self).__init__()


    async def __call__(
        self,
        handler: Callable[[Union[Message, CallbackQuery], Dict[str, Any]], Awaitable[Any]],
        event: Union[Message, CallbackQuery],
        data: Dict[str, Any]
    ) -> Any:

        if isinstance(event, Message) and event.media_group_id:
            try:
                self.medias[event.media_group_id].append(event)

                last_message_flag = False
                try:
                    await event.bot.edit_message_text(
                        text="  ",
                        chat_id=event.from_user.id,
                        message_id=event.message_id + 1
                    )
                except TelegramBadRequest as exc:
                    if (exc.message == "Bad Request: message must be non-empty" or
                            exc.message == "Bad Request: message to edit not found"):
                        last_message_flag = True

                if last_message_flag:
                    data["media_events"] = self.medias.pop(event.media_group_id)
                    return await handler(event, data)

                return

            except KeyError:
                self.medias[event.media_group_id] = [event]
                return

        return await handler(event, data)