remove debug code; move length limit to constants

This commit is contained in:
udf 2018-11-03 14:18:59 +02:00
parent 225e377130
commit e246a8d378
Signed by: kate
GPG Key ID: E40724BAD73AF77B
1 changed files with 5 additions and 7 deletions

View File

@ -10,10 +10,9 @@ from telethon.tl.types import MessageEntityPre
from telethon.tl.tlobject import TLObject from telethon.tl.tlobject import TLObject
import datetime import datetime
from telethon.tl.patched import Message
from telethon.tl.types import PeerUser
PRINTABLE_SET = set(bytes(string.printable, 'ascii')) PRINTABLE_SET = set(bytes(string.printable, 'ascii'))
STR_LEN_MAX = 256
BYTE_LEN_MAX = 64
def parse_pre(text): def parse_pre(text):
@ -55,15 +54,15 @@ def yaml_format(obj, indent=0):
result.append(' ' * indent) result.append(' ' * indent)
elif isinstance(obj, str): elif isinstance(obj, str):
# truncate long strings and display elipsis # truncate long strings and display elipsis
result.append(repr(obj[:80])) result.append(repr(obj[:STR_LEN_MAX]))
if len(obj) > 80: if len(obj) > STR_LEN_MAX:
result.append('') result.append('')
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(c in PRINTABLE_SET for c in obj):
result.append(repr(obj)) result.append(repr(obj))
else: else:
if len(obj) > 64: if len(obj) > BYTE_LEN_MAX:
result.append('<…>') result.append('<…>')
else: else:
result.append(' '.join(f'{b:02X}' for b in obj)) result.append(' '.join(f'{b:02X}' for b in obj))
@ -93,7 +92,6 @@ 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)
print(len(yaml_text), yaml_text)
await event.edit( await event.edit(
yaml_text, yaml_text,
parse_mode=parse_pre parse_mode=parse_pre