def del_lockables(update, context):
chat = update.effective_chat
message = update.effective_message
for lockable, filter in LOCK_TYPES.items():
if lockable == "rtl":
if sql.is_locked(chat.id, lockable) and can_delete(
chat, context.bot.id):
if message.caption:
check = ad.detect_alphabet(u"{}".format(message.caption))
if "ARABIC" in check:
try:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("ERROR in lockables")
break
if message.text:
check = ad.detect_alphabet(u"{}".format(message.text))
if "ARABIC" in check:
try:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("ERROR in lockables")
break
continue
if lockable == "button":
if sql.is_locked(chat.id, lockable) and can_delete(
chat, context.bot.id):
if message.reply_markup and message.reply_markup.inline_keyboard:
try:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("ERROR in lockables")
break
continue
if lockable == "inline":
if sql.is_locked(chat.id, lockable) and can_delete(
chat, context.bot.id):
if message and message.via_bot:
try:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("ERROR in lockables")
break
continue
if (filter(update) and sql.is_locked(chat.id, lockable)
and can_delete(chat, context.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, context.bot.id):
send_message(
update.effective_message,
"I see a bot and I've been told to stop them from joining..."
"but I'm not admin!",
)
return
chat.kick_member(new_mem.id)
send_message(
update.effective_message,
"Only admins are allowed to add bots in this chat! Get outta here.",
)
break
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 lock(update, context) -> str:
args = context.args
chat = update.effective_chat
user = update.effective_user
if (can_delete(chat, context.bot.id)
or update.effective_message.chat.type == "private"):
if len(args) >= 1:
ltype = args[0].lower()
if ltype in LOCK_TYPES:
# Connection check
conn = connected(context.bot,
update,
chat,
user.id,
need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_id = conn
chat_name = chat.title
text = "Locked all {} messages for non-admins in {}!".format(
ltype, chat_name)
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command is meant to use in group not in PM",
)
return ""
chat = update.effective_chat
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
text = "Locked all {} messages for non-admins!".format(
ltype)
sql.update_lock(chat.id, ltype, locked=True)
send_message(update.effective_message,
text,
parse_mode="markdown")
return ("<b>{}:</b>"
"\n#LOCK"
"\n<b>Admin:</b> {}"
"\nLocked <code>{}</code>.".format(
html.escape(chat.title),
mention_html(user.id, user.first_name),
ltype,
))
elif ltype in LOCK_CHAT_RESTRICTION:
# Connection check
conn = connected(context.bot,
update,
chat,
user.id,
need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_id = conn
chat_name = chat.title
text = "Locked {} for all non-admins in {}!".format(
ltype, chat_name)
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command is meant to use in group not in PM",
)
return ""
chat = update.effective_chat
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
text = "Locked {} for all non-admins!".format(ltype)
current_permission = context.bot.getChat(chat_id).permissions
context.bot.set_chat_permissions(
chat_id=chat_id,
permissions=get_permission_list(
eval(str(current_permission)),
LOCK_CHAT_RESTRICTION[ltype.lower()],
),
)
send_message(update.effective_message,
text,
parse_mode="markdown")
return ("<b>{}:</b>"
"\n#Permission_LOCK"
"\n<b>Admin:</b> {}"
"\nLocked <code>{}</code>.".format(
html.escape(chat.title),
mention_html(user.id, user.first_name),
ltype,
))
else:
send_message(
update.effective_message,
"What are you trying to lock...? Try /locktypes for the list of lockables",
)
else:
send_message(update.effective_message,
"What are you trying to lock...?")
else:
send_message(
update.effective_message,
"I am not administrator or haven't got enough rights.",
)
return ""
def dkick(update, context):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
message = update.effective_message # type: Optional[Message]
args = context.args
bot = context.bot
if user_can_ban(chat, user, context.bot.id) is False:
message.reply_text("You don't have enough rights to kick users!")
return ""
if message.reply_to_message:
user = update.effective_user # type: Optional[User]
chat = update.effective_chat # type: Optional[Chat]
if can_delete(chat, bot.id):
update.effective_message.reply_to_message.delete()
else:
message.reply_text(
"You have to reply to a message to delete it and kick the user.")
return ""
user_id, reason = extract_user_and_text(message, args)
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):
message.reply_text("Yeahh... let's start kicking admins?")
return ""
if user_id == context.bot.id:
message.reply_text("Yeahhh I'm not gonna do that")
return ""
if user_id == 777000 or user_id == 1087968824:
message.reply_text(
str(user_id) +
" is an account reserved for telegram, I cannot kick it!")
return ""
res = chat.unban_member(user_id) # unban on current user = kick
if res:
context.bot.sendMessage(
chat.id,
"Admin {} has successfully kicked {} in <b>{}</b>!".format(
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name),
html.escape(chat.title)),
parse_mode=ParseMode.HTML,
)
log = ("<b>{}:</b>"
"\n#KICKED"
"\n<b>Admin:</b> {}"
"\n<b>User:</b> {} (<code>{}</code>)".format(
html.escape(chat.title),
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name),
member.user.id,
))
if reason:
log += "\n<b>Reason:</b> {}".format(reason)
return log
else:
message.reply_text("Get Out!.")
return ""
def dwarn(user: User,
chat: Chat,
reason: str,
message: Message,
warner: User = None) -> str:
if message.reply_to_message:
user = update.effective_user # type: Optional[User]
chat = update.effective_chat # type: Optional[Chat]
if can_delete(chat, bot.id):
update.effective_message.reply_to_message.delete()
else:
message.reply_text(
"You have to reply to a message to delete it and warn the user.")
return ""
if is_user_admin(chat, user.id):
message.reply_text("Damn admins, can't even be warned!")
return ""
if warner:
warner_tag = mention_html(warner.id, warner.first_name)
else:
warner_tag = "Automated warn filter."
limit, soft_warn = sql.get_warn_setting(chat.id)
num_warns, reasons = sql.warn_user(user.id, chat.id, reason)
if num_warns >= limit:
sql.reset_warns(user.id, chat.id)
if soft_warn: # kick
chat.unban_member(user.id)
reply = "That's {} warnings, {} has been kicked!".format(
limit, mention_html(user.id, user.first_name))
else: # ban
chat.kick_member(user.id)
reply = "That's{} warnings, {} has been banned!".format(
limit, mention_html(user.id, user.first_name))
for warn_reason in reasons:
reply += "\n - {}".format(html.escape(warn_reason))
# message.bot.send_sticker(chat.id, BAN_STICKER) # banhammer marie
# sticker
keyboard = None
log_reason = ("<b>{}:</b>"
"\n#WARN_BAN"
"\n<b>Admin:</b> {}"
"\n<b>User:</b> {} (<code>{}</code>)"
"\n<b>Reason:</b> {}"
"\n<b>Counts:</b> <code>{}/{}</code>".format(
html.escape(chat.title),
warner_tag,
mention_html(user.id, user.first_name),
user.id,
reason,
num_warns,
limit,
))
else:
keyboard = InlineKeyboardMarkup([[
InlineKeyboardButton("Remove warn ⚠️",
callback_data="rm_warn({})".format(user.id))
]])
reply = "User {} has {}/{} warnings... watch out!".format(
mention_html(user.id, user.first_name), num_warns, limit)
if reason:
reply += "\nReason for last warn:\n{}".format(html.escape(reason))
log_reason = ("<b>{}:</b>"
"\n#WARN"
"\n<b>Admin:</b> {}"
"\n<b>User:</b> {} (<code>{}</code>)"
"\n<b>Reason:</b> {}"
"\n<b>Counts:</b> <code>{}/{}</code>".format(
html.escape(chat.title),
warner_tag,
mention_html(user.id, user.first_name),
user.id,
reason,
num_warns,
limit,
))
try:
message.reply_text(reply,
reply_markup=keyboard,
parse_mode=ParseMode.HTML)
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
message.reply_text(reply,
reply_markup=keyboard,
parse_mode=ParseMode.HTML,
quote=False)
else:
raise
return log_reason
def dban(update, context):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
message = update.effective_message # type: Optional[Message]
args = context.args
bot = context.bot
if user_can_ban(chat, user, context.bot.id) is False:
message.reply_text("You don't have enough rights to ban users!")
return ""
if message.reply_to_message:
user = update.effective_user # type: Optional[User]
chat = update.effective_chat # type: Optional[Chat]
if can_delete(chat, bot.id):
update.effective_message.reply_to_message.delete()
else:
message.reply_text(
"You have to reply to a message to delete it and ban the user.")
return ""
user_id, reason = extract_user_and_text(message, args)
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'm not gonna ban an admin, don't make fun of yourself!")
return ""
if user_id == context.bot.id:
message.reply_text(
"I'm not gonna ban myself, that's pretty dumb idea!")
return ""
if user_id == 777000 or user_id == 1087968824:
message.reply_text(
str(user_id) +
" is an account reserved for telegram, I cannot ban it!")
return ""
log = ("<b>{}:</b>"
"\n#BANNED"
"\n<b>Admin:</b> {}"
"\n<b>User:</b> {} (<code>{}</code>)".format(
html.escape(chat.title),
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name),
member.user.id,
))
if reason:
log += "\n<b>Reason:</b> {}".format(reason)
try:
chat.kick_member(user_id)
context.bot.sendMessage(
chat.id,
"Admin {} has successfully banned {} in <b>{}</b>!.".format(
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name),
html.escape(chat.title)),
parse_mode=ParseMode.HTML,
)
return log
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
message.reply_text("Banned!", quote=False)
return log
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.")
return ""