Add missing await to mention generation in _parse_message_text (#634)

This commit is contained in:
Tulir Asokan 2018-02-22 21:57:40 +02:00 committed by Lonami
parent 6e854325a8
commit 7998fd59f7

View File

@ -587,7 +587,7 @@ class TelegramClient(TelegramBareClient):
if request.id == update.message.id: if request.id == update.message.id:
return update.message return update.message
def _parse_message_text(self, message, parse_mode): async def _parse_message_text(self, message, parse_mode):
""" """
Returns a (parsed message, entities) tuple depending on parse_mode. Returns a (parsed message, entities) tuple depending on parse_mode.
""" """
@ -608,7 +608,7 @@ class TelegramClient(TelegramBareClient):
if m: if m:
try: try:
msg_entities[i] = InputMessageEntityMentionName( msg_entities[i] = InputMessageEntityMentionName(
e.offset, e.length, self.get_input_entity( e.offset, e.length, await self.get_input_entity(
int(m.group(1)) if m.group(1) else e.url int(m.group(1)) if m.group(1) else e.url
) )
) )
@ -647,7 +647,7 @@ class TelegramClient(TelegramBareClient):
the sent message the sent message
""" """
entity = await self.get_input_entity(entity) entity = await self.get_input_entity(entity)
message, msg_entities = self._parse_message_text(message, parse_mode) message, msg_entities = await self._parse_message_text(message, parse_mode)
request = SendMessageRequest( request = SendMessageRequest(
peer=entity, peer=entity,
@ -705,7 +705,7 @@ class TelegramClient(TelegramBareClient):
Returns: Returns:
the edited message the edited message
""" """
message, msg_entities = self._parse_message_text(message, parse_mode) message, msg_entities = await self._parse_message_text(message, parse_mode)
request = EditMessageRequest( request = EditMessageRequest(
peer=await self.get_input_entity(entity), peer=await self.get_input_entity(entity),
id=self._get_message_id(message_id), id=self._get_message_id(message_id),