Lsposed_Modules_Updates_Tra.../plugins/subs.py

67 lines
2.9 KiB
Python
Raw Normal View History

2022-03-20 05:02:17 +00:00
from pyrogram import Client, filters
from pyrogram.types import Message
from defs.source import from_keyword_to_module
from defs.subs import gen_subs_msg, gen_back_button, add_to_subs, remove_from_subs
from plugins.start import sub_msg, not_found_msg, already_sub_msg, unsub_msg, not_sub_msg
sub_help_msg = """
👩🏻💼 » /subscribe <code>包名|关键词|作用域|开源地址|模块地址</code> - 订阅更新
<code>/subscribe nil.nadph.qnotified</code>
<code>/subscribe QNotified</code>
<code>/subscribe com.tencent.mobileqq</code>
<code>/subscribe https://github.com/ferredoxin/QNotified</code>
<code>/subscribe https://modules.lsposed.org/module/nil.nadph.qnotified</code>
"""
unsub_help_msg = """
👩🏻💼 » /unsubscribe <code>包名|关键词|作用域|开源地址|模块地址</code> - 取消订阅更新
<code>/unsubscribe nil.nadph.qnotified</code>
<code>/unsubscribe QNotified</code>
<code>/unsubscribe com.tencent.mobileqq</code>
<code>/unsubscribe https://github.com/ferredoxin/QNotified</code>
<code>/unsubscribe https://modules.lsposed.org/module/nil.nadph.qnotified</code>
"""
@Client.on_message(filters.incoming & filters.private &
filters.command(["subscription"]))
async def subscription_command(_: Client, message: Message):
text = gen_subs_msg(message.from_user.id)
await message.reply(text, reply_markup=gen_back_button(), quote=True, )
@Client.on_message(filters.incoming & filters.private &
filters.command(["subscribe"]))
async def sub_command(_: Client, message: Message):
if len(message.command) == 1:
await message.reply(sub_help_msg, reply_markup=gen_back_button(), quote=True)
else:
data = " ".join(message.command[1:])
module = from_keyword_to_module(data)
if module:
success = add_to_subs(message.from_user.id, module)
if success:
await message.reply(sub_msg.format(module.name), quote=True)
else:
await message.reply(already_sub_msg.format(module.name), quote=True)
else:
await message.reply(not_found_msg.format(data), quote=True)
@Client.on_message(filters.incoming & filters.private &
filters.command(["unsubscribe"]))
async def un_sub_command(_: Client, message: Message):
if len(message.command) == 1:
await message.reply(unsub_help_msg, reply_markup=gen_back_button(), quote=True)
else:
data = " ".join(message.command[1:])
module = from_keyword_to_module(data)
if module:
success = remove_from_subs(message.from_user.id, module)
if success:
await message.reply(unsub_msg.format(module.name), quote=True)
else:
await message.reply(not_sub_msg.format(module.name), quote=True)
else:
await message.reply(not_found_msg.format(data), quote=True)