def del_lockables(bot: Bot, update: Update):
chat = update.effective_chat
message = update.effective_message
for lockable, filter in LOCK_TYPES.items():
if filter(message) and sql.is_locked(chat.id, lockable) and can_delete(
chat, bot.id):
if lockable == "bots":
new_members = update.effective_message.new_chat_members
for new_mem in new_members:
if new_mem.is_bot:
if not is_bot_admin(chat, bot.id):
message.reply_text(
"I see a bot, and I've been told to stop them joining... "
"but I'm not admin!")
return
chat.kick_member(new_mem.id)
message.reply_text(
"Only admins are allowed to add bots to this chat!."
)
else:
try:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("ERROR in lockables")
break
def runmute(bot: Bot, update: Update, args: List[str]):
message = update.effective_message
if not args:
message.reply_text("You don't seem to be referring to a chat/user.")
return
user_id, chat_id = extract_user_and_text(message, args)
if not user_id:
message.reply_text("You don't seem to be referring to a user.")
return
elif not chat_id:
message.reply_text("You don't seem to be referring to a chat.")
return
try:
chat = bot.get_chat(chat_id.split()[0])
except BadRequest as excp:
if excp.message == "Chat not found":
message.reply_text(
"Chat not found! Make sure you entered a valid chat ID and I'm part of that chat."
)
return
else:
raise
if chat.type == 'private':
message.reply_text("I'm sorry, but that's a private chat!")
return
if not is_bot_admin(chat, bot.id) or not chat.get_member(
bot.id).can_restrict_members:
message.reply_text(
"I can't unrestrict people there! Make sure I'm admin and can unban users."
)
return
try:
member = chat.get_member(user_id)
except BadRequest as excp:
if excp.message == "User not found":
message.reply_text("I can't seem to find this user there")
return
else:
raise
if is_user_in_chat(chat, user_id):
if member.can_send_messages and member.can_send_media_messages \
and member.can_send_other_messages and member.can_add_web_page_previews:
message.reply_text(
"This user already has the right to speak in that chat.")
return
if user_id == bot.id:
message.reply_text("I'm not gonna UNMUTE myself, I'm an admin there!")
return
try:
bot.restrict_chat_member(chat.id,
int(user_id),
can_send_messages=True,
can_send_media_messages=True,
can_send_other_messages=True,
can_add_web_page_previews=True)
message.reply_text("Yep, this user can talk in that chat!")
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
message.reply_text('Unmuted!', quote=False)
elif excp.message in RUNMUTE_ERRORS:
message.reply_text(excp.message)
else:
LOGGER.warning(update)
LOGGER.exception(
"ERROR unmnuting user %s in chat %s (%s) due to %s", user_id,
chat.title, chat.id, excp.message)
message.reply_text("Well damn, I can't unmute that user.")
def rban(bot: Bot, update: Update, args: List[str]):
message = update.effective_message
if not args:
message.reply_text("You don't seem to be referring to a chat/user.")
return
user_id, chat_id = extract_user_and_text(message, args)
if not user_id:
message.reply_text("You don't seem to be referring to a user.")
return
elif not chat_id:
message.reply_text("You don't seem to be referring to a chat.")
return
try:
chat = bot.get_chat(chat_id.split()[0])
except BadRequest as excp:
if excp.message == "Chat not found":
message.reply_text(
"Chat not found! Make sure you entered a valid chat ID and I'm part of that chat."
)
return
else:
raise
if chat.type == 'private':
message.reply_text("I'm sorry, but that's a private chat!")
return
if not is_bot_admin(chat, bot.id) or not chat.get_member(
bot.id).can_restrict_members:
message.reply_text(
"I can't restrict people there! Make sure I'm admin and can ban users."
)
return
try:
member = chat.get_member(user_id)
except BadRequest as excp:
if excp.message == "User not found":
message.reply_text("I can't seem to find this user")
return
else:
raise
if is_user_ban_protected(chat, user_id, member):
message.reply_text("I really wish I could ban admins...")
return
if user_id == bot.id:
message.reply_text("I'm not gonna BAN myself, are you crazy?")
return
try:
chat.kick_member(user_id)
message.reply_text("Banned from chat!")
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
message.reply_text('Banned!', quote=False)
elif excp.message in RBAN_ERRORS:
message.reply_text(excp.message)
else:
LOGGER.warning(update)
LOGGER.exception("ERROR banning user %s in chat %s (%s) due to %s",
user_id, chat.title, chat.id, excp.message)
message.reply_text("Well damn, I can't ban that user.")
def runban(bot: Bot, update: Update, args: List[str]):
message = update.effective_message
if not args:
message.reply_text("You don't seem to be referring to a chat/user.")
return
user_id, chat_id = extract_user_and_text(message, args)
if not user_id:
message.reply_text("You don't seem to be referring to a user.")
return
elif not chat_id:
message.reply_text("You don't seem to be referring to a chat.")
return
try:
chat = bot.get_chat(chat_id.split()[0])
except BadRequest as excp:
if excp.message == "Chat not found":
message.reply_text(
"Chat not found! Make sure you entered a valid chat ID and I'm part of that chat."
)
return
else:
raise
if chat.type == 'private':
message.reply_text("I'm sorry, but that's a private chat!")
return
if not is_bot_admin(chat, bot.id) or not chat.get_member(
bot.id).can_restrict_members:
message.reply_text(
"I can't unrestrict people there! Make sure I'm admin and can unban users."
)
return
try:
member = chat.get_member(user_id)
except BadRequest as excp:
if excp.message == "User not found":
message.reply_text("I can't seem to find this user there")
return
else:
raise
if is_user_in_chat(chat, user_id):
message.reply_text(
"Why are you trying to remotely unban someone that's already in that chat?"
)
return
if user_id == bot.id:
message.reply_text("I'm not gonna UNBAN myself, I'm an admin there!")
return
try:
chat.unban_member(user_id)
message.reply_text("Yep, this user can join that chat!")
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
message.reply_text('Unbanned!', quote=False)
elif excp.message in RUNBAN_ERRORS:
message.reply_text(excp.message)
else:
LOGGER.warning(update)
LOGGER.exception(
"ERROR unbanning user %s in chat %s (%s) due to %s", user_id,
chat.title, chat.id, excp.message)
message.reply_text("Well damn, I can't unban that user.")