Set _input_chat on custom.Message creation/fixup input_chat

This commit is contained in:
Lonami Exo 2018-06-14 23:40:44 +02:00
parent 091b03c635
commit 24d2074e0c
2 changed files with 6 additions and 1 deletions

View File

@ -127,7 +127,8 @@ class EventCommon(abc.ABC):
except ValueError:
ch = isinstance(self._chat_peer, types.PeerChannel)
if not ch and self._message_id is not None:
msg = await self._client.get_messages(ids=self._message_id)
msg = await self._client.get_messages(
None, ids=self._message_id)
self._chat = msg._chat
self._input_chat = msg._input_chat
else:

View File

@ -37,7 +37,11 @@ class Message:
self._input_sender = get_input_peer(self._sender)
else:
self._input_sender = None
self._input_chat = input_chat
if not self._input_chat and self._chat:
self._input_chat = get_input_peer(self._chat)
self._fwd_from_entity = None
if getattr(self.original_message, 'fwd_from', None):
fwd = self.original_message.fwd_from