43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
from pyrogram import filters
|
|
|
|
from app import BOT, Config, Message, bot
|
|
from app.plugins.sudo.users import SUDO_USERS
|
|
|
|
|
|
@BOT.add_cmd(cmd="disable_su", allow_sudo=False)
|
|
async def disable_su(bot: BOT, message: Message):
|
|
u_id = message.from_user.id
|
|
|
|
if u_id in Config.DISABLED_SUPERUSERS:
|
|
return
|
|
|
|
Config.DISABLED_SUPERUSERS.append(u_id)
|
|
|
|
await SUDO_USERS.add_data({"_id": u_id, "disabled": True})
|
|
|
|
await message.reply(
|
|
text="Your <b>SuperUser</b> Access is now <code>Disabled</code>.", del_in=10
|
|
)
|
|
|
|
|
|
@bot.on_message(
|
|
filters=filters.command(commands="enable_su", prefixes=Config.SUDO_TRIGGER)
|
|
& filters.create(
|
|
lambda _, __, m: m.from_user and m.from_user.id in Config.DISABLED_SUPERUSERS
|
|
),
|
|
group=1,
|
|
is_command=True,
|
|
filters_edited=True,
|
|
check_for_reactions=True,
|
|
)
|
|
async def enable_su(bot: BOT, message: Message):
|
|
u_id = message.from_user.id
|
|
|
|
Config.DISABLED_SUPERUSERS.remove(u_id)
|
|
|
|
await SUDO_USERS.add_data({"_id": u_id, "disabled": False})
|
|
|
|
await message.reply(
|
|
text="Your <b>SuperUser</b> Access is now <code>Enabled</code>.", del_in=10
|
|
)
|