75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
from app import BOT, DB, Config, Message, bot
|
|
|
|
|
|
async def init_task():
|
|
Config.SUDO_CMD_LIST = [
|
|
sudo_cmd["_id"] async for sudo_cmd in DB.SUDO_CMD_LIST.find()
|
|
]
|
|
|
|
|
|
@bot.add_cmd(cmd="addscmd")
|
|
async def add_scmd(bot: BOT, message: Message):
|
|
"""
|
|
CMD: ADDSCMD
|
|
INFO: Add Sudo Commands.
|
|
FLAGS: -all to instantly add all Commands.
|
|
USAGE:
|
|
.addscmd ping | .addscmd -all
|
|
"""
|
|
if "-all" in message.flags:
|
|
cmds = [{"_id": cmd} for cmd in Config.CMD_DICT.keys()]
|
|
Config.SUDO_CMD_LIST = list(Config.CMD_DICT.keys())
|
|
await DB.SUDO_CMD_LIST.drop()
|
|
await DB.SUDO_CMD_LIST.insert_many(cmds)
|
|
await message.reply("All Commands Added to Sudo!")
|
|
await bot.log(text="All Commands Added to Sudo!")
|
|
return
|
|
cmd = message.flt_input
|
|
response = await message.reply(f"Adding <b>{cmd}</b> to sudo....")
|
|
if cmd in Config.SUDO_CMD_LIST:
|
|
await response.edit(f"<b>{cmd}</b> already in Sudo!")
|
|
return
|
|
await DB.SUDO_CMD_LIST.insert_one({"_id": cmd})
|
|
Config.SUDO_CMD_LIST.append(cmd)
|
|
await response.edit(f"<b>{cmd}</b> added to Sudo!")
|
|
await bot.log(f"<b>{cmd}</b> added to Sudo!")
|
|
|
|
|
|
@bot.add_cmd(cmd="delscmd")
|
|
async def del_scmd(bot: BOT, message: Message):
|
|
"""
|
|
CMD: DELSCMD
|
|
INFO: Remove Sudo Commands.
|
|
FLAGS: -all to instantly remove all Commands.
|
|
USAGE:
|
|
.delscmd ping | .delscmd -all
|
|
"""
|
|
if "-all" in message.flags:
|
|
Config.SUDO_CMD_LIST = []
|
|
await DB.SUDO_CMD_LIST.drop()
|
|
await message.reply("All Commands Removed from Sudo!")
|
|
await bot.log(text="All Commands Removed from Sudo!")
|
|
return
|
|
cmd = message.flt_input
|
|
response = await message.reply(f"Removing <b>{cmd}</b> from sudo....")
|
|
if cmd not in Config.SUDO_CMD_LIST:
|
|
await response.edit(f"<b>{cmd}</b> not in Sudo!")
|
|
return
|
|
await DB.SUDO_CMD_LIST.delete_one({"_id": cmd})
|
|
Config.SUDO_CMD_LIST.remove(cmd)
|
|
await response.edit(f"<b>{cmd}</b> removed from Sudo!")
|
|
await bot.log(f"<b>{cmd}</b> removed from Sudo!")
|
|
|
|
|
|
@bot.add_cmd(cmd="vscmd")
|
|
async def view_sudo_cmd(bot: BOT, message: Message):
|
|
cmds = " ".join(Config.SUDO_CMD_LIST)
|
|
if not cmds:
|
|
await message.reply("No Commands in SUDO!")
|
|
return
|
|
await message.reply(
|
|
text=f"List of <b>{len(Config.SUDO_CMD_LIST)}</b> SUDO CMDS:\n\n{cmds}",
|
|
del_in=30,
|
|
block=False,
|
|
)
|