async def disable_command(message, chat, strings):
cmd = get_arg(message).lower()
if cmd[0] == '/' or cmd[0] == '!':
cmd = cmd[1:]
# Check on commands aliases
for name, keys in COMMANDS_ALIASES.items():
if cmd in keys:
cmd = name
break
if cmd not in DISABLABLE_COMMANDS:
await message.reply(strings["wot_to_disable"])
return
if await db.disabled.find_one({
'chat_id': chat['chat_id'],
'cmds': {
'$in': [cmd]
}
}):
await message.reply(strings['already_disabled'])
return
await db.disabled.update_one({'chat_id': chat['chat_id']},
{"$addToSet": {
'cmds': {
'$each': [cmd]
}
}},
upsert=True)
await message.reply(strings["disabled"].format(
cmd=cmd, chat_name=chat['chat_title']))
async def enable_command(message, chat, strings):
chat_id = chat['chat_id']
cmd = get_arg(message).lower()
if cmd[0] == '/' or cmd[0] == '!':
cmd = cmd[1:]
# Check on commands aliases
for name, keys in COMMANDS_ALIASES.items():
if cmd in keys:
cmd = name
break
if cmd not in DISABLABLE_COMMANDS:
await message.reply(strings["wot_to_enable"])
return
if not await db.disabled.find_one({
'chat_id': chat['chat_id'],
'cmds': {
'$in': [cmd]
}
}):
await message.reply(strings["already_enabled"])
return
await db.disabled.update_one({'chat_id': chat_id},
{'$pull': {
'cmds': cmd
}})
await message.reply(strings["enabled"].format(cmd=cmd,
chat_name=chat['chat_title'])
)
async def enable_command(message, chat, strings):
chat_id = chat["chat_id"]
cmd = get_arg(message).lower()
if cmd[0] in ["/", "!"]:
cmd = cmd[1:]
# Check on commands aliases
for name, keys in COMMANDS_ALIASES.items():
if cmd in keys:
cmd = name
break
if cmd not in DISABLABLE_COMMANDS:
await message.reply(strings["wot_to_enable"])
return
if not await db.disabled.find_one({
"chat_id": chat["chat_id"],
"cmds": {
"$in": [cmd]
}
}):
await message.reply(strings["already_enabled"])
return
await db.disabled.update_one({"chat_id": chat_id},
{"$pull": {
"cmds": cmd
}})
await message.reply(strings["enabled"].format(cmd=cmd,
chat_name=chat["chat_title"])
)