Simplify code #2

Closed
Lonami wants to merge 1 commits from Lonami/uniborg:master into master
1 changed files with 2 additions and 8 deletions

View File

@ -2,15 +2,12 @@
# License, v. 2.0. If a copy of the MPL was not distributed with this # License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/. # file, You can obtain one at http://mozilla.org/MPL/2.0/.
import string
from telethon import events from telethon import events
from telethon.utils import add_surrogate from telethon.utils import add_surrogate
from telethon.tl.types import MessageEntityPre from telethon.tl.types import MessageEntityPre
from telethon.tl.tlobject import TLObject from telethon.tl.tlobject import TLObject
import datetime import datetime
PRINTABLE_SET = set(string.printable.encode())
STR_LEN_MAX = 256 STR_LEN_MAX = 256
BYTE_LEN_MAX = 64 BYTE_LEN_MAX = 64
@ -61,7 +58,7 @@ def yaml_format(obj, indent=0):
return result return result
elif isinstance(obj, bytes): elif isinstance(obj, bytes):
# repr() bytes if it's printable, hex like "FF EE BB" otherwise # repr() bytes if it's printable, hex like "FF EE BB" otherwise
if all(c in PRINTABLE_SET for c in obj): if all(0x20 <= c < 0x7f for c in obj):
return repr(obj) return repr(obj)
else: else:
return ('<…>' if len(obj) > BYTE_LEN_MAX else return ('<…>' if len(obj) > BYTE_LEN_MAX else
@ -91,7 +88,4 @@ async def _(event):
return return
msg = await event.message.get_reply_message() msg = await event.message.get_reply_message()
yaml_text = yaml_format(msg) yaml_text = yaml_format(msg)
await event.edit( await event.edit(yaml_text, parse_mode=parse_pre)
yaml_text,
parse_mode=parse_pre
)