def temp_ban(bot: Bot, update: Update, args: List[str]) -> str:
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
message = update.effective_message # type: Optional[Message]
user_id, reason = extract_user_and_text(message, args)
time.sleep(2)
message.delete()
if can_delete(chat, bot.id):
try:
update.effective_message.reply_to_message.delete()
except AttributeError:
pass
if not user_id:
message.reply_text(
tld(chat.id, "You don't seem to be referring to a user."))
return ""
try:
member = chat.get_member(user_id)
except BadRequest as excp:
if excp.message == "User not found":
message.reply_text(tld(chat.id, "I can't seem to find this user"))
return ""
else:
raise
if is_user_ban_protected(chat, user_id, member):
message.reply_text(
tld(
chat.id,
"This user is ban protected, meaning that you cannot ban this user!"
))
return ""
if user_id == bot.id:
message.reply_text(
tld(chat.id, "I'm not gonna BAN myself, are you crazy?"))
return ""
if not reason:
message.reply_text(
tld(chat.id, "You haven't specified a time to ban this user for!"))
return ""
split_reason = reason.split(None, 1)
time_val = split_reason[0].lower()
if len(split_reason) > 1:
reason = split_reason[1]
else:
reason = ""
bantime = extract_time(message, time_val)
if not bantime:
return ""
log = "<b>{}:</b>" \
"\n#TEMP BANNED" \
"\n<b>Admin:</b> {}" \
"\n<b>User:</b> {} (<code>{}</code>)" \
"\n<b>Time:</b> {}".format(html.escape(chat.title),
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name),
member.user.id,
time_val)
if reason:
log += "\n<b>Reason:</b> {}".format(reason)
try:
chat.kick_member(user_id, until_date=bantime)
bot.send_sticker(update.effective_chat.id, BAN_STICKER)
reply = "{} has been temporarily banned for {}!".format(
mention_html(member.user.id, member.user.first_name), time_val)
bot.sendMessage(chat.id, reply, parse_mode=ParseMode.HTML)
return log
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
bot.send_sticker(chat.id, BAN_STICKER) # banhammer marie sticker
message.reply_text(
tld(chat.id,
"Banned! User will be banned for {}.").format(time_val),
quote=False)
return log
else:
LOGGER.warning(update)
LOGGER.exception("ERROR banning user %s in chat %s (%s) due to %s",
user_id, chat.title, chat.id, excp.message)
message.reply_text(tld(chat.id, "Banned!"))
return ""
def temp_nomedia(bot: Bot, update: Update, args: List[str]) -> str:
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
message = update.effective_message # type: Optional[Message]
conn = connected(bot, update, chat, user.id)
if not conn == False:
chatD = dispatcher.bot.getChat(conn)
else:
if chat.type == "private":
exit(1)
else:
chatD = chat
user_id, reason = extract_user_and_text(message, args)
if not user_id:
message.reply_text(tld(chat.id, "You don't seem to be referring to a user."))
return ""
try:
member = chat.get_member(user_id)
except BadRequest as excp:
if excp.message == "User not found":
message.reply_text(tld(chat.id, "I can't seem to find this user"))
return ""
else:
raise
if is_user_admin(chat, user_id, member):
message.reply_text(tld(chat.id, "I really wish I could restrict admins..."))
return ""
if user_id == bot.id:
message.reply_text(tld(chat.id, "I'm not gonna RESTRICT myself, are you crazy?"))
return ""
if not reason:
message.reply_text(tld(chat.id, "You haven't specified a time to restrict this user for!"))
return ""
split_reason = reason.split(None, 1)
time_val = split_reason[0].lower()
if len(split_reason) > 1:
reason = split_reason[1]
else:
reason = ""
mutetime = extract_time(message, time_val)
if not mutetime:
return ""
log = "<b>{}:</b>" \
"\n#TEMP RESTRICTED" \
"\n<b>• Admin:</b> {}" \
"\n<b>• User:</b> {}" \
"\n<b>• ID:</b> <code>{}</code>" \
"\n<b>• Time:</b> {}".format(html.escape(chat.title), mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name), user_id, time_val)
if reason:
log += "\n<b>• Reason:</b> {}".format(reason)
try:
if member.can_send_messages is None or member.can_send_messages:
bot.restrict_chat_member(chat.id, user_id, until_date=mutetime, can_send_messages=True,
can_send_media_messages=False,
can_send_other_messages=False,
can_add_web_page_previews=False)
message.reply_text(tld(chat.id, "Restricted from sending media for {} in {}!").format(time_val, chatD.title))
return log
else:
message.reply_text(tld(chat.id, "This user is already restricted in {}.").format(chatD.title))
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
message.reply_text(tld(chat.id, "Restricted for {} in {}!").format(time_val, chatD.title), quote=False)
return log
else:
LOGGER.warning(update)
LOGGER.exception("ERROR muting user %s in chat %s (%s) due to %s", user_id, chat.title, chat.id,
excp.message)
message.reply_text(tld(chat.id, "Well damn, I can't restrict that user."))
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]
bot = context.bot
if not user: # ignore channels
return ""
# ignore admins
if is_user_admin(chat, user.id):
sql.update_flood(chat.id, None)
return ""
# ignore approved
if is_approved(chat.id, user.id):
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 = tld(update.effective_message, "Get Out!")
tag = "BANNED"
elif getmode == 2:
chat.kick_member(user.id)
chat.unban_member(user.id)
execstrings = tld(update.effective_message, "Kicked!")
tag = "KICKED"
elif getmode == 3:
bot.restrict_chat_member(chat.id, user.id, can_send_messages=False)
execstrings = tld(update.effective_message, "Now you silent!")
tag = "MUTED"
elif getmode == 4:
bantime = extract_time(msg, getvalue)
chat.kick_member(user.id, until_date=bantime)
execstrings = tld(update.effective_message,
"Tᴇᴍᴘ Bᴀɴɴᴇᴅ ᴛɪʟʟ {}").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 = tld(update.effective_message,
"Mᴜᴛᴇᴅ ᴛɪʟʟ {}").format(getvalue)
tag = "TMUTE"
reply = "{} i like babycorn you like flooding we are not same bro so *{}* in {}".format(
mention_html(user.id, user.first_name), execstrings, chat.title)
msg.reply_text(reply, parse_mode=ParseMode.HTML)
return "#MUTED" \
"\n<b>Chat:</b> {}" \
"\n<b>User:</b> {}" \
"\nFlooded the group.".format(html.escape(chat.title),
mention_html(user.id, user.first_name))
except BadRequest:
msg.reply_text(
tld(
chat.id,
"I can't Mute people here, give me permissions first! Until then, I'll disable antiflood."
))
sql.set_flood(chat.id, 0)
return "#INFO" \
"\n<b>Chat:</b> {}" \
"\nDon't have Mute permissions, so automatically disabled antiflood.".format(chat.title)