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 (
f"This chat has `{num_warn_filters}` warn filters. "
f"It takes `{limit}` warns before the user gets *{'kicked' if soft_warn else 'banned'}*."
)
def set_warn_limit(update: Update, context: CallbackContext) -> str:
args = context.args
chat: Optional[Chat] = update.effective_chat
user: Optional[User] = update.effective_user
msg: Optional[Message] = update.effective_message
if args:
if args[0].isdigit():
if int(args[0]) < 3:
msg.reply_text("The minimum warn limit is 3!")
else:
sql.set_warn_limit(chat.id, int(args[0]))
msg.reply_text("Updated the warn limit to {}".format(args[0]))
return (
f"<b>{html.escape(chat.title)}:</b>\n"
f"#SET_WARN_LIMIT\n"
f"<b>Admin:</b> {mention_html(user.id, user.first_name)}\n"
f"Set the warn limit to <code>{args[0]}</code>")
else:
msg.reply_text("Give me a number as an arg!")
else:
limit, soft_warn = sql.get_warn_setting(chat.id)
msg.reply_text("The current warn limit is {}".format(limit))
return ""
def warns(update: Update, context: CallbackContext):
args = context.args
message: Optional[Message] = update.effective_message
chat: Optional[Chat] = update.effective_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 = f"This user has {num_warns}/{limit} warns, for the following reasons:"
for reason in reasons:
text += f"\n • {reason}"
msgs = split_message(text)
for msg in msgs:
update.effective_message.reply_text(msg)
else:
update.effective_message.reply_text(
f"User has {num_warns}/{limit} warns, but no reasons for any of them."
)
else:
update.effective_message.reply_text("This user doesn't have any warns!")
def rmwarn_cmd(update: Update, context: CallbackContext) -> str:
args = context.args
message: Optional[Message] = update.effective_message
chat: Optional[Chat] = update.effective_chat
user_id = extract_user(message, args)
if user_id:
warns = sql.get_warns(user_id, chat.id)
if warns and warns[0] !=0:
num_warns, reasons = warns
limit, soft_warn = sql.get_warn_setting(chat.id)
keyboard = InlineKeyboardMarkup([[
InlineKeyboardButton(
"Remove warn", callback_data="rm_warn({})".format(user_id))
]])
reply_text = f"This user has {num_warns}/{limit} warns."
try:
message.reply_text(reply_text, reply_markup=keyboard)
except BadRequest as err:
message.reply_text(
reply_text, reply_markup=keyboard, quote=False)
else:
message.reply_text("This user doesn't have any warns.")
else:
message.reply_text("No user has been mentioned.")
return ""
def set_warn_strength(update: Update, context: CallbackContext):
args = context.args
chat: Optional[Chat] = update.effective_chat
user: Optional[User] = update.effective_user
msg: Optional[Message] = update.effective_message
if args:
if args[0].lower() in ("on", "yes"):
sql.set_warn_strength(chat.id, False)
msg.reply_text("Too many warns will now result in a Ban!")
return (
f"<b>{html.escape(chat.title)}:</b>\n"
f"<b>Admin:</b> {mention_html(user.id, user.first_name)}\n"
f"Has enabled strong warns. Users will be seriously punched.(banned)"
)
elif args[0].lower() in ("off", "no"):
sql.set_warn_strength(chat.id, True)
msg.reply_text(
"Too many warns will now result in a normal punch! Users will be able to join again after."
)
return (
f"<b>{html.escape(chat.title)}:</b>\n"
f"<b>Admin:</b> {mention_html(user.id, user.first_name)}\n"
f"Has disabled strong punches. I will use normal punch on users."
)
else:
msg.reply_text("I only understand on/yes/no/off!")
else:
limit, soft_warn = sql.get_warn_setting(chat.id)
if soft_warn:
msg.reply_text(
"Warns are currently set to *punch* users when they exceed the limits.",
parse_mode=ParseMode.MARKDOWN)
else:
msg.reply_text(
"Warns are currently set to *Ban* users when they exceed the limits.",
parse_mode=ParseMode.MARKDOWN)
return ""
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, They are too far to be eliminated!")
return
if user.id in TIGERS:
if warner:
message.reply_text("Tigers cant be warned.")
else:
message.reply_text(
"Tiger triggered an auto warn filter!\n I can't warn tigers but they should avoid abusing this."
)
return
if user.id in WOLVES:
if warner:
message.reply_text("Wolf disasters are warn immune.")
else:
message.reply_text(
"Wolf Disaster triggered an auto warn filter!\nI can't warn wolves but they should avoid abusing this."
)
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: # punch
chat.unban_member(user.id)
reply = (
f"<code>❕</code><b>Punch Event</b>\n"
f"<code> </code><b>• User:</b> {mention_html(user.id, user.first_name)}\n"
f"<code> </code><b>• Count:</b> {limit}")
else: # ban
chat.kick_member(user.id)
reply = (
f"<code>❕</code><b>Ban Event</b>\n"
f"<code> </code><b>• User:</b> {mention_html(user.id, user.first_name)}\n"
f"<code> </code><b>• Count:</b> {limit}")
for warn_reason in reasons:
reply += f"\n - {html.escape(warn_reason)}"
# message.bot.send_sticker(chat.id, BAN_STICKER) # Saitama's sticker
keyboard = None
log_reason = (f"<b>{html.escape(chat.title)}:</b>\n"
f"#WARN_BAN\n"
f"<b>Admin:</b> {warner_tag}\n"
f"<b>User:</b> {mention_html(user.id, user.first_name)}\n"
f"<b>Reason:</b> {reason}\n"
f"<b>Counts:</b> <code>{num_warns}/{limit}</code>")
else:
keyboard = InlineKeyboardMarkup([[
InlineKeyboardButton(
"🔘 Remove warn", callback_data="rm_warn({})".format(user.id))
]])
reply = (
f"<code>❕</code><b>Warn Event</b>\n"
f"<code> </code><b>• User:</b> {mention_html(user.id, user.first_name)}\n"
f"<code> </code><b>• Count:</b> {num_warns}/{limit}")
if reason:
reply += f"\n<code> </code><b>• Reason:</b> {html.escape(reason)}"
log_reason = (f"<b>{html.escape(chat.title)}:</b>\n"
f"#WARN\n"
f"<b>Admin:</b> {warner_tag}\n"
f"<b>User:</b> {mention_html(user.id, user.first_name)}\n"
f"<b>Reason:</b> {reason}\n"
f"<b>Counts:</b> <code>{num_warns}/{limit}</code>")
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