From 8fbf7adb6ccb08d661ddb0ff4c922351b8acc6a4 Mon Sep 17 00:00:00 2001 From: apepenkov Date: Fri, 10 Nov 2023 23:26:49 +0300 Subject: [PATCH] update get_message to use InputMessageCallbackQuery --- client/src/telethon/_impl/client/events/queries.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/client/src/telethon/_impl/client/events/queries.py b/client/src/telethon/_impl/client/events/queries.py index acb66f3a..074cf812 100644 --- a/client/src/telethon/_impl/client/events/queries.py +++ b/client/src/telethon/_impl/client/events/queries.py @@ -6,6 +6,7 @@ from ...tl import abcs, functions, types from ..types import Chat, Message from .event import Event from ..types.chat import peer_id +from ..client.messages import CherryPickedList if TYPE_CHECKING: from ..client.client import Client @@ -76,12 +77,16 @@ class ButtonCallback(Event): If the message is too old and is no longer accessible, :data:`None` is returned instead. """ + pid = peer_id(self._raw.peer) chat = self._chat_map.get(pid) if not chat: chat = await self._client._resolve_to_packed(pid) - message = (await self._client.get_messages_with_ids(chat=chat, message_ids=[self._raw.msg_id]))[0] + lst = CherryPickedList(self._client, chat, []) + lst._ids.append(types.InputMessageCallbackQuery(id=self._raw.msg_id, query_id=self._raw.query_id)) + + message = (await lst)[0] return message or None