Refactor patch

This commit is contained in:
Lonami 2018-06-26 15:56:25 +02:00 committed by GitHub
parent 2119a999d9
commit 2b989cae27
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -617,21 +617,14 @@ 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: ent = self.original_message.entities
entities_list = [c for c in self.original_message.entities if cls and ent:
if isinstance(c, cls)] ent = [c for c in ent if isinstance(c, cls)]
texts = get_inner_text(
self.original_message.message, texts = get_inner_text(self.original_message.message, ent)
entities_list
)
else:
texts = get_inner_text(entities_list,
self.original_message.entities)
return list(zip(entities_list, 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):