def warn(
user: User, chat: Chat, reason: str, message: Message, warner: User = None
) -> str:
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 __import_data__(chat_id, data):
for user_id, count in data.get("warns", {}).items():
for x in range(int(count)):
sql.warn_user(user_id, chat_id)
