def flood(update, context):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
conn = connected(context.bot, update, chat, user.id, need_admin=False)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if update.effective_message.chat.type == "private":
send_message(update.effective_message,
"You can do this command in groups, not PM")
return
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
limit = sql.get_flood_limit(chat_id)
if limit == 0:
if conn:
text = "I'm not currently enforcing flood control in *{}*!".format(
chat_name)
else:
text = "I'm not currently enforcing flood control!"
send_message(update.effective_message, text, parse_mode="markdown")
else:
getmode, getvalue = sql.get_flood_setting(chat.id)
if getmode == 1:
settypeflood = 'banned'
elif getmode == 2:
settypeflood = 'kicked'
elif getmode == 3:
settypeflood = 'muted'
elif getmode == 4:
settypeflood = 'temporarily banned for {}'.format(getvalue)
elif getmode == 5:
settypeflood = 'temporarily muted for {}'.format(getvalue)
if conn:
text = "This chat is currently enforcing flood control after *{}* messages. Any users sending more than that amount of messages will be *{}* in *{}*.".format(
limit, settypeflood, chat_name)
# text = "If member is flooding messages, they will got *{}* in *{}*.".format(settypeflood, chat_name)
else:
text = "This chat is currently enforcing flood control after *{}* messages. Any users sending more than that amount of messages will be *{}*.".format(
limit, settypeflood)
# text = "If member is flooding messages, they will got *{}*.".format(settypeflood)
send_message(update.effective_message, text, parse_mode="markdown")
def check_flood(bot: Bot, update: Update) -> str:
user = update.effective_user
chat = update.effective_chat
msg = update.effective_message
if not user: # ignore channels
return ""
# ignore admins
if is_user_admin(chat, user.id):
sql.update_flood(chat.id, None)
return ""
should_ban = sql.update_flood(chat.id, user.id)
if not should_ban:
return ""
try:
getmode, getvalue = sql.get_flood_setting(chat.id)
if getmode == 1:
chat.kick_member(user.id)
execstrings = "Banned"
tag = "BANNED"
elif getmode == 2:
chat.kick_member(user.id)
chat.unban_member(user.id)
execstrings = "Kicked"
tag = "KICKED"
elif getmode == 3:
bot.restrict_chat_member(
chat.id, user.id, can_send_messages=False
)
execstrings = "Muted"
tag = "MUTED"
elif getmode == 4:
bantime = extract_time(msg, getvalue)
chat.kick_member(user.id, until_date=bantime)
execstrings = "Banned for {}".format(getvalue)
tag = "TBAN"
elif getmode == 5:
mutetime = extract_time(msg, getvalue)
bot.restrict_chat_member(
chat.id,
user.id,
until_date=mutetime,
can_send_messages=False
)
execstrings = "Muted for {}".format(getvalue)
tag = "TMUTE"
send_message(
update.effective_message, tld(
chat.id, "flood_mute").format(execstrings)
)
return tld(chat.id, "flood_logger_success").format(
tag,
html.escape(chat.title),
mention_html(user.id, html.escape(user.first_name)),
)
except BadRequest:
msg.reply_text(tld(chat.id, "flood_err_no_perm"))
sql.set_flood(chat.id, 0)
return tld(chat.id, "flood_logger_fail").format(chat.title)
def set_flood_mode(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]
conn = connected(bot, update, chat, user.id, need_admin=True)
if args:
if args[0].lower() == "ban":
settypeflood = "ban"
sql.set_flood_strength(chat.id, 1, "0")
elif args[0].lower() == "kick":
settypeflood = "kick"
sql.set_flood_strength(chat.id, 2, "0")
elif args[0].lower() == "mute":
settypeflood = "mute"
sql.set_flood_strength(chat.id, 3, "0")
elif args[0].lower() == "tban":
if len(args) == 1:
teks = """It looks like you tried to set time value for antiflood but you didn't specified time; Try, `/setfloodmode tban <timevalue>`.
Examples of time value: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks."""
send_message(update.effective_message,
teks, parse_mode="markdown")
return
settypeflood = "tban for {}".format(args[1])
sql.set_flood_strength(chat.id, 4, str(args[1]))
elif args[0].lower() == "tmute":
if len(args) == 1:
teks = (
update.effective_message,
"""It looks like you tried to set time value for antiflood but you didn't specified time; Try, `/setfloodmode tmute <timevalue>`.
Examples of time value: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks.""",
)
send_message(update.effective_message,
teks, parse_mode="markdown")
return
settypeflood = "tmute for {}".format(args[1])
sql.set_flood_strength(chat.id, 5, str(args[1]))
else:
send_message(
update.effective_message, "I only understand ban/kick/mute/tban/tmute!"
)
return
if conn:
msg.reply_text(
"Exceeding consecutive flood limit will result in {} in {}!".format(
settypeflood, chat.title
)
)
else:
msg.reply_text(
"Exceeding consecutive flood limit will result in {}!".format(
settypeflood
)
)
return (
"<b>{}:</b>\n"
"<b>Admin:</b> {}\n"
"Has changed antiflood mode. User will {}.".format(
settypeflood,
html.escape(chat.title),
mention_html(user.id, html.escape(user.first_name)),
)
)
else:
getmode, getvalue = sql.get_flood_setting(chat.id)
if getmode == 1:
settypeflood = "ban"
elif getmode == 2:
settypeflood = "kick"
elif getmode == 3:
settypeflood = "mute"
elif getmode == 4:
settypeflood = "tban for {}".format(getvalue)
elif getmode == 5:
settypeflood = "tmute for {}".format(getvalue)
if conn:
msg.reply_text(
"Sending more messages than flood limit will result in {} in {}.".format(
settypeflood, chat.title
)
)
else:
msg.reply_text(
"Sending more message than flood limit will result in {}.".format(
settypeflood
)
)
return ""
def set_flood_mode(update, context):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
msg = update.effective_message # type: Optional[Message]
args = context.args
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if update.effective_message.chat.type == "private":
send_message(update.effective_message,
"You can do this command in groups, not PM")
return ""
chat = update.effective_chat
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
if args:
if args[0].lower() == 'ban':
settypeflood = 'banned'
sql.set_flood_strength(chat_id, 1, "0")
elif args[0].lower() == 'kick':
settypeflood = 'kicked'
sql.set_flood_strength(chat_id, 2, "0")
elif args[0].lower() == 'mute':
settypeflood = 'muted'
sql.set_flood_strength(chat_id, 3, "0")
elif args[0].lower() == 'tban':
if len(args) == 1:
teks = """It looks like you are trying to set a temporary value for antiflood, but has not determined the time; use `/setfloodmode tban <timevalue>`.
Examples of time values: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 week."""
send_message(update.effective_message,
teks,
parse_mode="markdown")
return
settypeflood = "temp banned for {}".format(args[1])
sql.set_flood_strength(chat_id, 4, str(args[1]))
elif args[0].lower() == 'tmute':
if len(args) == 1:
teks = """It looks like you are trying to set a temporary value for antiflood, but has not determined the time; use `/setfloodmode tban <timevalue>`.
Examples of time values: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 week."""
send_message(update.effective_message,
teks,
parse_mode="markdown")
return
settypeflood = "temp mute for {}".format(args[1])
sql.set_flood_strength(chat_id, 5, str(args[1]))
else:
send_message(update.effective_message,
"I only understand ban/kick/mute/tban/tmute!")
return
if conn:
text = "Flooding messages now will result `{}` in *{}*!".format(
settypeflood, chat_name)
else:
text = "Flooding messages now will result `{}`!".format(
settypeflood)
send_message(update.effective_message, text, parse_mode="markdown")
return "<b>{}:</b>\n" \
"<b>Admin:</b> {}\n" \
"Has changed antiflood mode. User will {}.".format(settypeflood, html.escape(chat.title),
mention_html(user.id, user.first_name))
else:
getmode, getvalue = sql.get_flood_setting(chat.id)
if getmode == 1:
settypeflood = 'banned'
elif getmode == 2:
settypeflood = 'kicked'
elif getmode == 3:
settypeflood = 'muted'
elif getmode == 4:
settypeflood = 'temp banned for {}'.format(getvalue)
elif getmode == 5:
settypeflood = 'temp mute for {}'.format(getvalue)
if conn:
text = "If member is flooding messages, they will got *{}* in *{}*.".format(
settypeflood, chat_name)
else:
text = "If member is flooding messages, they will got *{}*.".format(
settypeflood)
send_message(update.effective_message,
text,
parse_mode=ParseMode.MARKDOWN)
return ""
def check_flood(update, context) -> str:
user = update.effective_user # type: Optional[User]
chat = update.effective_chat # type: Optional[Chat]
msg = update.effective_message # type: Optional[Message]
if not user: # ignore channels
return ""
# ignore admins
if is_user_admin(chat, user.id):
sql.update_flood(chat.id, None)
return ""
should_ban = sql.update_flood(chat.id, user.id)
if not should_ban:
return ""
try:
getmode, getvalue = sql.get_flood_setting(chat.id)
if getmode == 1:
chat.kick_member(user.id)
execstrings = "Get out!"
tag = "BANNED"
elif getmode == 2:
chat.kick_member(user.id)
chat.unban_member(user.id)
execstrings = "Get out!"
tag = "KICKED"
elif getmode == 3:
context.bot.restrict_chat_member(
chat.id,
user.id,
permissions=ChatPermissions(can_send_messages=False))
execstrings = "Now you shutup!"
tag = "MUTED"
elif getmode == 4:
bantime = extract_time(msg, getvalue)
chat.kick_member(user.id, until_date=bantime)
execstrings = "Get out for {}!".format(getvalue)
tag = "TBAN"
elif getmode == 5:
mutetime = extract_time(msg, getvalue)
context.bot.restrict_chat_member(
chat.id,
user.id,
until_date=mutetime,
permissions=ChatPermissions(can_send_messages=False))
execstrings = "Now you shutup for {}!".format(getvalue)
tag = "TMUTE"
send_message(
update.effective_message,
"I like to leave the flooding to natural disasters. But you, you were just a disappointment. {}"
.format(execstrings))
return "<b>{}:</b>" \
"\n#{}" \
"\n<b>User:</b> {}" \
"\nFlooded the group.".format(tag, html.escape(chat.title),
mention_html(user.id, user.first_name))
except BadRequest:
send_message(
update.effective_message,
"I can't kick people here, give me permissions first! Until then, I'll disable antiflood."
)
sql.set_flood(chat.id, 0)
return "<b>{}:</b>" \
"\n#INFO" \
"\n{}".format(chat.title, "Don't have kick permissions, so automatically disabled antiflood.")