def set_warn_limit(bot: Bot, update: Update, args: List[str]) -> str:
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
msg = update.effective_message # type: Optional[Message]
if args:
if args[0].isdigit():
if int(args[0]) < 3:
msg.reply_text(tld(chat.id, "The minimum warn limit is 3!"))
else:
sql.set_warn_limit(chat.id, int(args[0]))
msg.reply_text(
tld(chat.id,
"Updated the warn limit to {}").format(args[0]))
return "<b>{}:</b>" \
"\n#SET_WARN_LIMIT" \
"\n<b>Admin:</b> {}" \
"\nSet the warn limit to <code>{}</code>".format(html.escape(chat.title),
mention_html(user.id, user.first_name), args[0])
else:
msg.reply_text(tld(chat.id, "Give me a number as an arg!"))
else:
limit, soft_warn = sql.get_warn_setting(chat.id)
msg.reply_text(
tld(chat.id, "The current warn limit is {}").format(limit))
return ""
def warns(bot: Bot, update: Update, args: List[str]):
message = update.effective_message # type: Optional[Message]
chat = update.effective_chat # type: Optional[Chat]
user_id = extract_user(message, args) or update.effective_user.id
result = sql.get_warns(user_id, chat.id)
if result and result[0] != 0:
num_warns, reasons = result
limit, soft_warn = sql.get_warn_setting(chat.id)
if reasons:
text = tld(
chat.id,
"This user has {}/{} warnings, for the following reasons:"
).format(num_warns, limit)
for reason in reasons:
text += "\n - {}".format(reason)
msgs = split_message(text)
for msg in msgs:
update.effective_message.reply_text(msg)
else:
update.effective_message.reply_text(
tld(
chat.id,
"User has {}/{} warnings, but no reasons for any of them."
).format(num_warns, limit))
else:
update.effective_message.reply_text(
tld(chat.id, "This user hasn't got any warnings!"))
def set_warn_strength(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
msg = update.effective_message # type: Optional[Message]
if args:
if args[0].lower() in ("on", "yes"):
sql.set_warn_strength(chat.id, False)
msg.reply_text(tld(chat.id, "Too many warns will now result in a ban!"))
return tld(chat.id, "<b>{}:</b>\n" \
"<b>Admin:</b> {}\n" \
"Has enabled strong warns. Users will be banned.").format(html.escape(chat.title),
mention_html(user.id, user.first_name))
elif args[0].lower() in ("off", "no"):
sql.set_warn_strength(chat.id, True)
msg.reply_text(tld(chat.id, "Too many warns will now result in a kick! Users will be able to join again after."))
return tld(chat.id, "<b>{}:</b>\n" \
"<b>Admin:</b> {}\n" \
"Has disabled strong warns. Users will only be kicked.").format(html.escape(chat.title),
mention_html(user.id,
user.first_name))
else:
msg.reply_text(tld(chat.id, "I only understand on/yes/no/off!"))
else:
limit, soft_warn = sql.get_warn_setting(chat.id)
if soft_warn:
msg.reply_text(tld(chat.id, "Warns are currently set to *kick* users when they exceed the limits."),
parse_mode=ParseMode.MARKDOWN)
else:
msg.reply_text(tld(chat.id, "Warns are currently set to *ban* users when they exceed the limits."),
parse_mode=ParseMode.MARKDOWN)
return ""
def __chat_settings__(chat_id, user_id):
num_warn_filters = sql.num_warn_chat_filters(chat_id)
limit, soft_warn = sql.get_warn_setting(chat_id)
return tld(chat_id, "This chat has `{}` warn filters. It takes `{}` warns " \
"before the user gets *{}*.").format(num_warn_filters, limit, "kicked" if soft_warn else "banned")
def warn(user: User,
chat: Chat,
reason: str,
message: Message,
warner: User = None) -> str:
if is_user_admin(chat, user.id):
message.reply_text(tld(chat.id, "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 = tld(chat.id, "{} warnings, {} has been kicked!").format(
limit, mention_html(user.id, user.first_name))
else: # ban
chat.kick_member(user.id)
reply = tld(chat.id, "{} 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 = []
log_reason = "<b>{}:</b>" \
"\n#WARN_BAN" \
"\n<b>Admin:</b> {}" \
"\n<b>User:</b> {}" \
"\n<b>Reason:</b> {}".format(html.escape(chat.title),
warner_tag,
mention_html(user.id, user.first_name), reason)
else:
keyboard = InlineKeyboardMarkup([[
InlineKeyboardButton(tld(chat.id, "Remove warn"),
callback_data="rm_warn({})".format(user.id))
]])
reply = tld(chat.id, "{} has {}/{} warnings... watch out!").format(
mention_html(user.id, user.first_name), num_warns, limit)
if reason:
reply += tld(chat.id, "\nReason for last warn:\n{}").format(
html.escape(reason))
log_reason = "<b>{}:</b>" \
"\n#WARN" \
"\n<b>Admin:</b> {}" \
"\n<b>User:</b> {}" \
"\n<b>Reason:</b> {}".format(html.escape(chat.title),
warner_tag,
mention_html(user.id, user.first_name), reason)
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