LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-07 13:34:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

fixing get_entities_text with cls

Browse Source 
Currently the entity and text are zipped. If a cls parameter is passed, the text would have been filtered but the entities list would not be filtered, therefore the text would not be correspond with the entity.
This commit is contained in:
chrizrobert 2018-06-26 18:37:28 +08:00 committed by GitHub
parent 9971145721
commit fe4ee2c5e1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
telethon/tl/custom/message.py
View File

@ -617,19 +617,21 @@ class Message:
>>> for _, inner_text in m.get_entities_text(MessageEntityCode): >>> for _, inner_text in m.get_entities_text(MessageEntityCode):
>>> print(inner_text) >>> print(inner_text)
""" """
entities_list = self.original_message.entities
if not self.original_message.entities: if not self.original_message.entities:
return [] return []
if cls and self.original_message.entities: if cls and self.original_message.entities:
entities_list = [c for c in self.original_message.entities if isinstance(c, cls)]
texts = get_inner_text( texts = get_inner_text(
self.original_message.message, self.original_message.message,
[c for c in self.original_message.entities entities_list
if isinstance(c, cls)]
) )
else: else:
texts = get_inner_text(self.original_message.message, texts = get_inner_text(entities_list,
self.original_message.entities) self.original_message.entities)
return list(zip(self.original_message.entities, texts)) return list(zip(entities_list, texts))
async def click(self, i=None, j=None, *, text=None, filter=None): async def click(self, i=None, j=None, *, text=None, filter=None):
""" """