diff --git a/data/command.yml b/data/command.yml index 7f35e5c..613f81d 100644 --- a/data/command.yml +++ b/data/command.yml @@ -118,7 +118,12 @@ tg: format: -tg usage: 舔狗语录 -test: - cmd: t - format: -t - usage: 测试 \ No newline at end of file +archive: + cmd: arch + format: -arch + usage: 归档当前对话 + +unarchive: + cmd: unarch + format: -unarch + usage: 撤销归档当前对话 diff --git a/plugins/archive.py b/plugins/archive.py new file mode 100644 index 0000000..d3779e3 --- /dev/null +++ b/plugins/archive.py @@ -0,0 +1,24 @@ +import asyncio +from core import command +from pyrogram import Client +from pyrogram.types import Message + + +@Client.on_message(command("archive")) +async def archive(cli: Client, msg: Message): + if await cli.archive_chats(msg.chat.id): + await msg.edit_text(f"✅ Archive `{msg.chat.title}` successfully!") + else: + await msg.edit_text(f"❌ Failed to archive `{msg.chat.title}`!") + await asyncio.sleep(2) + await msg.delete() + + +@Client.on_message(command("unarchive")) +async def unarchive(cli: Client, msg: Message): + if await cli.unarchive_chats(msg.chat.id): + await msg.edit_text(f"✅ Unarchive `{msg.chat.title}` successfully!") + else: + await msg.edit_text(f"❌ Failed to unarchive `{msg.chat.title}`!") + await asyncio.sleep(2) + await msg.delete()