python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix imports

Browse source
This commit is contained in:
Ilya Samartsev 2020-09-03 14:38:14 +03:00
parent f47873e6d7
commit 979e159223
1 changed files with 7 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

20
aiogram/utils/text_decorations.py
View file

@ -56,15 +56,9 @@ class TextDecoration(ABC):
:return: :return:
""" """
text = text.encode('utf-16-le')
entities = [MessageEntity(**e.to_python()) for e in (entities if entities else [])]
for e in entities:
e.offset *= 2
e.length *= 2
result = "".join( result = "".join(
self._unparse_entities( self._unparse_entities(
text, sorted(entities, key=lambda item: item.offset) if entities else [] text.encode('utf-16-le'), sorted(entities, key=lambda item: item.offset) if entities else []
) )
) )
return result return result
@ -81,15 +75,15 @@ class TextDecoration(ABC):
length = length or len(text) length = length or len(text)
for index, entity in enumerate(entities): for index, entity in enumerate(entities):
if entity.offset < offset: if entity.offset * 2 < offset:
continue continue
if entity.offset > offset: if entity.offset * 2> offset:
yield self.quote(text[offset : entity.offset].decode('utf-16-le')) yield self.quote(text[offset : entity.offset * 2].decode('utf-16-le'))
start = entity.offset start = entity.offset * 2
offset = entity.offset + entity.length offset = entity.offset * 2 + entity.length * 2
sub_entities = list( sub_entities = list(
filter(lambda e: e.offset < (offset or 0), entities[index + 1 :]) filter(lambda e: e.offset * 2 < (offset or 0), entities[index + 1 :])
) )
yield self.apply_entity( yield self.apply_entity(
entity, entity,