forked from kate/uniborg
remove debug code; move length limit to constants
This commit is contained in:
parent
225e377130
commit
e246a8d378
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue