diff --git a/stdplugins/kbass_toggle_preview.py b/stdplugins/kbass_toggle_preview.py index 8d2a454..b60446d 100644 --- a/stdplugins/kbass_toggle_preview.py +++ b/stdplugins/kbass_toggle_preview.py @@ -1,5 +1,5 @@ """ -Reply to a message with .p to toggle the webpage preview of a message +Reply to a message with .p to toggle the webpage preview of a message """ from telethon.errors import MessageNotModifiedError from telethon.tl.functions.messages import EditMessageRequest @@ -7,16 +7,13 @@ from telethon.tl.functions.messages import EditMessageRequest from stdplugins.kbass_core import self_reply_cmd -@self_reply_cmd(borg, r"^\.p(?: ?)([yn])?$") +@self_reply_cmd(borg, r"^\.p$") async def on_edit_preview(event, target): - preview = event.pattern_match.group(1) == 'y' - if not event.pattern_match.group(1): - preview = not bool(target.media) try: await borg(EditMessageRequest( peer=await event.get_input_chat(), id=target.id, - no_webpage=not preview, + no_webpage=bool(target.media), message=target.message, entities=target.entities ))