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)
def warn(user: User,
chat: Chat,
reason: str,
message: Message,
warner: User = None) -> str:
bot = dispatcher.bot
if is_user_admin(chat, user.id):
message.reply_text(tld(chat.id, 'warns_warn_admin_no_warn'))
return ""
if warner:
warner_tag = mention_html(warner.id, warner.first_name)
else:
warner_tag = tld(chat.id, 'warns_warn_admin_autofilter')
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, 'warns_max_warn_kick').format(
limit, mention_html(user.id, user.first_name))
else: # ban
chat.kick_member(user.id)
reply = tld(chat.id, 'warns_max_warn_ban').format(
limit, mention_html(user.id, user.first_name))
for warn_reason in reasons:
reply += "\n - {}".format(html.escape(warn_reason))
keyboard = []
log_reason = tld(chat.id, 'warns_warn_ban_log_channel').format(
html.escape(chat.title), warner_tag,
mention_html(user.id, user.first_name), user.id, reason, num_warns,
limit)
else:
keyboard = [[
InlineKeyboardButton(tld(chat.id, 'warns_btn_remove_warn'),
callback_data="rm_warn({})".format(user.id))
]]
rules = rules_sql.get_rules(chat.id)
if rules:
keyboard[0].append(
InlineKeyboardButton(tld(chat.id, 'warns_btn_rules'),
url="t.me/{}?start={}".format(
bot.username, chat.id)))
reply = tld(chat.id, 'warns_user_warned').format(
mention_html(user.id, user.first_name), num_warns, limit)
if reason:
reply += tld(chat.id, 'warns_latest_warn_reason').format(
html.escape(reason))
log_reason = tld(chat.id, 'warns_warn_log_channel').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=InlineKeyboardMarkup(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=InlineKeyboardMarkup(keyboard),
parse_mode=ParseMode.HTML,
quote=False)
else:
raise
return log_reason
def warn(user: User,
chat: Chat,
reason: str,
message: Message,
warner: User = None) -> str:
if is_user_admin(chat, user.id):
message.reply_text("I'm not going to warn an admin!")
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 = "{} warnings, {} has been kicked!".format(
limit, mention_html(user.id, user.first_name))
else: # ban
chat.kick_member(user.id)
reply = "{} 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))
keyboard = []
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⚠️(admin only)",
callback_data="rm_warn({})".format(user.id))
]])
reply = "{} <b>has been warned!</b>\nTotal warn that this user have {}/{}".format(
mention_html(user.id, user.first_name), num_warns, limit)
if reason:
reply += "\nReason for the warn:\n<code>{}</code>".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 warn(user: User,
chat: Chat,
reason: str,
message: Message,
warner: User = None) -> str:
bot = dispatcher.bot
if is_user_admin(chat, user.id):
message.reply_text("I'm not going to warn an admin!")
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 = "{} warnings, {} has been kicked!".format(
limit, mention_html(user.id, user.first_name))
else: # ban
chat.kick_member(user.id)
reply = "{} 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> {} (<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 = [[
InlineKeyboardButton("тЪая╕П Remove warn",
callback_data="rm_warn({})".format(user.id))
]]
rules = rules_sql.get_rules(chat.id)
if rules:
keyboard[0].append(
InlineKeyboardButton("ЁЯУЭ Rules",
url="t.me/{}?start={}".format(
bot.username, chat.id)))
reply = "{} <b>has been warned!</b>\nThey have {}/{} warnings.".format(
mention_html(user.id, user.first_name), num_warns, limit)
if reason:
reply += "\nThe latest warning was because:\n<code>{}</code>".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=InlineKeyboardMarkup(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=InlineKeyboardMarkup(keyboard),
parse_mode=ParseMode.HTML,
quote=False)
else:
raise
return log_reason
