forked from kate/uniborg
add .edit command to ninja
it's now a full port from kateborg
This commit is contained in:
parent
4c7923dd44
commit
923f04e688
|
@ -9,6 +9,7 @@ import telethon.utils
|
||||||
|
|
||||||
from uniborg import util
|
from uniborg import util
|
||||||
|
|
||||||
|
|
||||||
async def get_target_message(event):
|
async def get_target_message(event):
|
||||||
if event.is_reply and (await event.reply_message).from_id == borg.uid:
|
if event.is_reply and (await event.reply_message).from_id == borg.uid:
|
||||||
return event.reply_message
|
return event.reply_message
|
||||||
|
@ -29,12 +30,21 @@ async def await_read(chat, message):
|
||||||
|
|
||||||
await fut
|
await fut
|
||||||
|
|
||||||
@borg.on(util.admin_cmd(r"^\.del(ete)?$"))
|
@borg.on(util.admin_cmd(r"^\.(del)(?:ete)?$"))
|
||||||
|
@borg.on(util.admin_cmd(r"^\.(edit)(.*)$"))
|
||||||
async def delete(event):
|
async def delete(event):
|
||||||
await event.delete()
|
await event.delete()
|
||||||
|
command = event.pattern_match.group(1)
|
||||||
|
if command == 'edit':
|
||||||
|
text = event.pattern_match.group(2).strip()
|
||||||
|
if not text:
|
||||||
|
return
|
||||||
target = await get_target_message(event)
|
target = await get_target_message(event)
|
||||||
if target:
|
if target:
|
||||||
chat = await event.input_chat
|
chat = await event.input_chat
|
||||||
await await_read(chat, target)
|
await await_read(chat, target)
|
||||||
await asyncio.sleep(.5)
|
await asyncio.sleep(.5)
|
||||||
|
if command == 'edit':
|
||||||
|
await borg.edit_message(chat, target, text)
|
||||||
|
else:
|
||||||
await borg.delete_messages(chat, target)
|
await borg.delete_messages(chat, target)
|
||||||
|
|
Loading…
Reference in New Issue