def enforce_gban(bot: Bot, update: Update):
# Not using @restrict handler to avoid spamming - just ignore if cant gban.
try:
restrict_permission = update.effective_chat.get_member(
bot.id
).can_restrict_members
except Unauthorized:
return
if sql.does_chat_gban(update.effective_chat.id) and restrict_permission:
user = update.effective_user
chat = update.effective_chat
msg = update.effective_message
if user and not is_user_admin(chat, user.id):
check_and_ban(update, user.id)
return
if msg.new_chat_members:
new_members = update.effective_message.new_chat_members
for mem in new_members:
check_and_ban(update, mem.id)
if msg.reply_to_message:
user = msg.reply_to_message.from_user
if user and not is_user_admin(chat, user.id):
check_and_ban(update, user.id, should_message=False)
def check_update(self, update):
if isinstance(update, Update) and update.effective_message:
message = update.effective_message
if (message.entities and message.entities[0].type == MessageEntity.BOT_COMMAND
and message.entities[0].offset == 0):
command = message.text[1:message.entities[0].length]
args = message.text.split()[1:]
command = command.split('@')
command.append(message.bot.username)
if not (command[0].lower() in self.command
and command[1].lower() == message.bot.username.lower()):
return None
filter_result = self.filters(update)
if filter_result:
chat = update.effective_chat
user = update.effective_user
# disabled, admincmd, user admin
if sql.is_command_disabled(chat.id, command[0].lower()):
# check if command was disabled
is_disabled = command[0] in ADMIN_CMDS and is_user_admin(chat, user.id)
if not is_disabled and sql.is_disable_del(chat.id):
# disabled and should delete
update.effective_message.delete()
if not is_disabled:
return None
else:
return args, filter_result
return args, filter_result
else:
return False
def rmwarn_handler(bot: Bot, update: Update) -> str:
chat = update.effective_chat
query = update.callback_query
user = update.effective_user
match = re.match(r"rm_warn\((.+?)\)", query.data)
if match:
user_id = match.group(1)
if not is_user_admin(chat, int(user.id)):
query.answer(text=tld(chat.id, 'warns_remove_admin_only'),
show_alert=True)
return ""
res = sql.remove_warn(user_id, chat.id)
if res:
update.effective_message.edit_text(tld(
chat.id, 'warns_remove_success').format(
mention_html(user.id, user.first_name)),
parse_mode=ParseMode.HTML)
user_member = chat.get_member(user_id)
return tld(chat.id, 'warns_remove_log_channel').format(
html.escape(chat.title), mention_html(user.id,
user.first_name),
mention_html(user_member.user.id, user_member.user.first_name),
user_member.user.id)
else:
update.effective_message.edit_text(tld(
chat.id, 'warns_user_has_no_warns').format(
mention_html(user.id, user.first_name)),
parse_mode=ParseMode.HTML)
return ""
def mute(bot: Bot, update: Update, args: List[str]) -> str:
chat = update.effective_chat
user = update.effective_user
message = update.effective_message
conn = connected(bot, update, chat, user.id)
if conn:
chatD = dispatcher.bot.getChat(conn)
else:
if chat.type == "private":
return
else:
chatD = chat
user_id = extract_user(message, args)
if not user_id:
message.reply_text(tld(chat.id, "mute_invalid"))
return ""
if user_id == bot.id:
message.reply_text(tld(chat.id, "mute_not_myself"))
return ""
member = chatD.get_member(int(user_id))
if member:
if user_id in SUDO_USERS:
message.reply_text(tld(chat.id, "mute_not_sudo"))
elif is_user_admin(chatD, user_id, member=member):
message.reply_text(tld(chat.id, "mute_not_m_admin"))
elif member.can_send_messages is None or member.can_send_messages:
bot.restrict_chat_member(chatD.id,
user_id,
can_send_messages=False)
keyboard = []
reply = tld(chat.id, "mute_success").format(
mention_html(member.user.id, member.user.first_name),
chatD.title)
message.reply_text(reply,
reply_markup=keyboard,
parse_mode=ParseMode.HTML)
return "<b>{}:</b>" \
"\n#MUTE" \
"\n<b>Admin:</b> {}" \
"\n<b>User:</b> {}".format(html.escape(chatD.title),
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name))
else:
message.reply_text(
tld(chat.id, "mute_already_mute").format(chatD.title))
else:
message.reply_text(
tld(chat.id, "mute_not_in_chat").format(chatD.title))
return ""
def nomedia(bot: Bot, update: Update, args: List[str]) -> str:
chat = update.effective_chat
user = update.effective_user
message = update.effective_message
conn = connected(bot, update, chat, user.id)
if conn:
chatD = dispatcher.bot.getChat(conn)
else:
if chat.type == "private":
return
else:
chatD = chat
user_id = extract_user(message, args)
if not user_id:
message.reply_text(tld(chat.id, "restrict_invalid"))
return ""
if user_id == bot.id:
message.reply_text(tld(chat.id, "restrict_is_bot"))
return ""
member = chatD.get_member(int(user_id))
if member:
if is_user_admin(chatD, user_id, member=member):
message.reply_text(tld(chat.id, "restrict_is_admin"))
elif member.can_send_messages is None or member.can_send_messages:
bot.restrict_chat_member(chatD.id,
user_id,
can_send_messages=True,
can_send_media_messages=False,
can_send_other_messages=False,
can_add_web_page_previews=False)
keyboard = []
reply = tld(chat.id, "restrict_success").format(
mention_html(member.user.id, member.user.first_name),
chatD.title)
message.reply_text(reply,
reply_markup=keyboard,
parse_mode=ParseMode.HTML)
return "<b>{}:</b>" \
"\n#RESTRICTED" \
"\n<b>• Admin:</b> {}" \
"\n<b>• User:</b> {}" \
"\n<b>• ID:</b> <code>{}</code>".format(html.escape(chatD.title),
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name), user_id)
else:
message.reply_text(tld(chat.id, "restrict_already_restricted"))
else:
message.reply_text(
tld(chat.id, "mute_not_in_chat").format(chatD.title))
return ""
def enforce_gban(update, context):
# Not using @restrict handler to avoid spamming - just ignore if cant gban.
if sql.does_chat_gban(update.effective_chat.id) and update.effective_chat.get_member(context.bot.id).can_restrict_members:
user = update.effective_user # type: Optional[User]
chat = update.effective_chat # type: Optional[Chat]
msg = update.effective_message # type: Optional[Message]
if user and not is_user_admin(chat, user.id):
check_and_ban(update, user.id)
if msg.new_chat_members:
new_members = update.effective_message.new_chat_members
for mem in new_members:
check_and_ban(update, mem.id)
if msg.reply_to_message:
user = msg.reply_to_message.from_user # type: Optional[User]
if user and not is_user_admin(chat, user.id):
check_and_ban(update, user.id, should_message=False)
def kickme(update, context):
user_id = update.effective_message.from_user.id
if is_user_admin(update.effective_chat, user_id):
send_message(update.effective_message, "I wish I could... but you're an admin.")
return
res = update.effective_chat.unban_member(user_id) # unban on current user = kick
if res:
send_message(update.effective_message, "No problem")
else:
send_message(update.effective_message, "Huh? I can't")
def unlock(bot: Bot, update: Update, args: List[str]) -> str:
chat = update.effective_chat
user = update.effective_user
message = update.effective_message
if is_user_admin(chat, message.from_user.id):
if len(args) >= 1:
if args[0] in LOCK_TYPES:
sql.update_lock(chat.id, args[0], locked=False)
message.reply_text(tld(chat.id,
"locks_unlock_success").format(args[0]),
parse_mode=ParseMode.MARKDOWN)
return "<b>{}:</b>" \
"\n#UNLOCK" \
"\n<b>Admin:</b> {}" \
"\nUnlocked <code>{}</code>.".format(html.escape(chat.title),
mention_html(user.id, user.first_name), args[0])
elif args[0] in RESTRICTION_TYPES:
sql.update_restriction(chat.id, args[0], locked=False)
# members = users_sql.get_chat_members(chat.id)
# if args[0] == "messages":
# unrestr_members(bot, chat.id, members, media=False, other=False, previews=False)
# elif args[0] == "media":
# unrestr_members(bot, chat.id, members, other=False, previews=False)
# elif args[0] == "other":
# unrestr_members(bot, chat.id, members, previews=False)
# elif args[0] == "previews":
# unrestr_members(bot, chat.id, members)
# elif args[0] == "all":
# unrestr_members(bot, chat.id, members, True, True, True, True)
message.reply_text(tld(chat.id,
"locks_unlock_success").format(args[0]),
parse_mode=ParseMode.MARKDOWN)
return "<b>{}:</b>" \
"\n#UNLOCK" \
"\n<b>Admin:</b> {}" \
"\nUnlocked <code>{}</code>.".format(html.escape(chat.title),
mention_html(user.id, user.first_name), args[0])
else:
message.reply_text(tld(chat.id, "locks_type_invalid"))
else:
bot.sendMessage(chat.id, tld(chat.id, "locks_unlock_no_type"))
return ""
def start(update, context):
if update.effective_chat.type == "private":
args = context.args
if len(args) >= 1:
if args[0].lower() == "help":
send_help(update.effective_chat.id, HELP_STRINGS)
elif args[0].lower() == "get_notes":
update.effective_message.reply_text("Now you can get notes in group.")
elif args[0].lower().startswith("stngs_"):
match = re.match("stngs_(.*)", args[0].lower())
chat = dispatcher.bot.getChat(match.group(1))
if is_user_admin(chat, update.effective_user.id):
send_settings(match.group(1), update.effective_user.id, False)
else:
send_settings(match.group(1), update.effective_user.id, True)
elif args[0][1:].isdigit() and "rules" in IMPORTED:
IMPORTED["rules"].send_rules(update, args[0], from_pm=True)
elif args[0][:4] == "wiki":
wiki = args[0].split("-")[1].replace('_', ' ')
message = update.effective_message
getlang = langsql.get_lang(message)
if getlang == "id":
wikipedia.set_lang("id")
pagewiki = wikipedia.page(wiki)
judul = pagewiki.title
summary = pagewiki.summary
if len(summary) >= 4096:
summary = summary[:4000]+"..."
message.reply_text("<b>{}</b>\n{}".format(judul, summary), parse_mode=ParseMode.HTML,
reply_markup=InlineKeyboardMarkup(
[[InlineKeyboardButton(text="use Wikipedia", url=pagewiki.url)]]))
elif args[0][:6].lower() == "verify":
chat_id = args[0].split("_")[1]
verify_welcome(update, context, chat_id)
else:
first_name = update.effective_user.first_name
buttons = InlineKeyboardMarkup(
[[InlineKeyboardButton(text="🎉 Add me to your group", url="https://t.me/{}?startgroup=new".format(context.bot.username))],
[InlineKeyboardButton(text="⚙️ Connect Group", callback_data="main_connect")],
[InlineKeyboardButton(text="🔔 Update Channel", url="https://t.me/Farseen_king_of_telegram")]])
[InlineKeyboardButton(text="Help❓", callback_data="help_back")]])
update.effective_message.reply_text(PM_START_TEXT.format(escape_markdown(first_name), escape_markdown(context.bot.first_name)), reply_markup=buttons, parse_mode=ParseMode.MARKDOWN, disable_web_page_preview=True)
def get_settings(update, context):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
msg = update.effective_message # type: Optional[Message]
args = msg.text.split(None, 1)
# ONLY send settings in PM
if chat.type != chat.PRIVATE:
if is_user_admin(chat, user.id):
text = "Click here to get this chat's settings, as well as yours."
msg.reply_text(text,
reply_markup=InlineKeyboardMarkup(
[[InlineKeyboardButton(text="Settings",
url="t.me/{}?start=stngs_{}".format(
context.bot.username, chat.id))]]))
def kickme(bot: Bot, update: Update):
chat = update.effective_chat # type: Optional[Chat]
user_id = update.effective_message.from_user.id
if is_user_admin(update.effective_chat, user_id):
update.effective_message.reply_text(tld(chat.id, "bans_kick_is_admin"))
return
res = update.effective_chat.unban_member(
user_id) # unban on current user = kick
if res:
update.effective_message.reply_text(tld(chat.id,
"bans_kickme_success"))
else:
update.effective_message.reply_text(tld(chat.id, "bans_kickme_failed"))
def banme(bot: Bot, update: Update):
user_id = update.effective_message.from_user.id
chat = update.effective_chat
if is_user_admin(update.effective_chat, user_id):
update.effective_message.reply_text(
tld(chat.id, "bans_err_usr_is_admin"))
return
res = update.effective_chat.kick_member(user_id)
if res:
update.effective_message.reply_text(tld(chat.id,
"bans_kickme_success"))
else:
update.effective_message.reply_text(tld(chat.id, "bans_kickme_failed"))
def check_update(self, update):
chat = update.effective_chat
user = update.effective_user
if super().check_update(update):
# Should be safe since check_update passed.
command = update.effective_message.text_html.split(
None, 1)[0][1:].split('@')[0]
# disabled, admincmd, user admin
if sql.is_command_disabled(chat.id, command):
return command in ADMIN_CMDS and is_user_admin(
chat, user.id)
# not disabled
else:
return True
return False
def muteme(bot: Bot, update: Update, args: List[str]) -> str:
user_id = update.effective_message.from_user.id
chat = update.effective_chat
user = update.effective_user
if is_user_admin(update.effective_chat, user_id):
update.effective_message.reply_text(tld(chat.id, "mute_is_admin"))
return
res = bot.restrict_chat_member(chat.id, user_id, can_send_messages=False)
if res:
update.effective_message.reply_text(tld(chat.id, "muteme_muted"))
log = "<b>{}:</b>" \
"\n#MUTEME" \
"\n<b>User:</b> {}" \
"\n<b>ID:</b> <code>{}</code>".format(html.escape(chat.title),
mention_html(user.id, user.first_name), user_id)
return log
else:
update.effective_message.reply_text(tld(chat.id, "mute_cant_mute"))
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 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.")
def unlock(update, context) -> str:
args = context.args
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
message = update.effective_message # type: Optional[Message]
if is_user_admin(chat, message.from_user.id):
if len(args) >= 1:
ltype = args[0].lower()
if ltype in LOCK_TYPES:
# Connection check
conn = connected(context.bot,
update,
chat,
user.id,
need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_id = conn
chat_name = chat.title
text = "Unlocked {} for everyone!".format(ltype, chat_name)
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
text = "Permission *{}* was unlocked for all non-admins!".format(
ltype)
sql.update_lock(chat.id, ltype, locked=False)
send_message(update.effective_message,
text,
parse_mode="markdown")
return "<b>{}:</b>" \
"\n#UNLOCK" \
"\n<b>Admin:</b> {}" \
"\nUnlocked <code>{}</code>.".format(html.escape(chat.title), mention_html(user.id, user.first_name), ltype)
elif ltype in UNLOCK_CHAT_RESTRICTION:
# Connection check
conn = connected(context.bot,
update,
chat,
user.id,
need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_id = conn
chat_name = chat.title
text = "Permission *{}* was unlocked for all non-admins at *{}*!".format(
ltype, chat_name)
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
text = "Permission *{}* was unlocked for all non-admins!".format(
ltype)
current_permission = context.bot.getChat(chat_id).permissions
context.bot.set_chat_permissions(
chat_id=chat_id,
permissions=get_permission_list(
eval(str(current_permission)),
UNLOCK_CHAT_RESTRICTION[ltype.lower()]))
send_message(update.effective_message,
text,
parse_mode="markdown")
return "<b>{}:</b>" \
"\n#UNLOCK" \
"\n<b>Admin:</b> {}" \
"\nUnlocked <code>{}</code>.".format(html.escape(chat.title), mention_html(user.id, user.first_name), ltype)
else:
send_message(
update.effective_message,
"What are you trying to unlock...? Try /locktypes for the list of lockables"
)
else:
send_message(update.effective_message,
"What do you want to unlock...?")
return ""
def temp_nomedia(bot: Bot, update: Update, args: List[str]) -> str:
chat = update.effective_chat
user = update.effective_user
message = update.effective_message
conn = connected(bot, update, chat, user.id)
if conn:
chatD = dispatcher.bot.getChat(conn)
else:
if chat.type == "private":
return
else:
chatD = chat
user_id, reason = extract_user_and_text(message, args)
if not user_id:
message.reply_text(tld(chat.id, "mute_not_refer"))
return ""
try:
member = chat.get_member(user_id)
except BadRequest as excp:
if excp.message == "User not found":
message.reply_text(tld(chat.id, "mute_not_existed"))
return ""
else:
raise
if is_user_admin(chat, user_id, member):
message.reply_text(tld(chat.id, "restrict_is_admin"))
return ""
if user_id == bot.id:
message.reply_text(tld(chat.id, "restrict_is_bot"))
return ""
if not reason:
message.reply_text(tld(chat.id, "nomedia_need_time"))
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 += tld(chat.id, "bans_logger_reason").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, "nomedia_success").format(time_val, chatD.title))
return log
else:
message.reply_text(
tld(chat.id,
"restrict_already_restricted").format(chatD.title))
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
message.reply_text(tld(chat.id, "nomedia_success").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, "restrict_cant_restricted"))
return ""
def mute(update, context):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
message = update.effective_message # type: Optional[Message]
args = context.args
user_id, reason = extract_user_and_text(message, args)
if not user_id or user_id == "error":
send_message(
update.effective_message,
"You'll need to either give me a username to mute, or reply to someone to be muted."
)
return ""
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
text = "Muted on *{}*!".format(chat_name)
else:
if update.effective_message.chat.type == "private":
update.effective_send_message(
update.effective_message,
"You can do this command in groups, not PM")
return ""
chat = update.effective_chat
member = chat.get_member(int(user_id))
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
text = "Admin {} Muted user {}\n<b>Reason:</b> {}".format(
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name), reason
or "No reason given")
if user_id == context.bot.id:
send_message(update.effective_message, "I'm not muting myself!")
return ""
check = context.bot.getChatMember(chat.id, user.id)
if check['can_restrict_members'] == False:
send_message(update.effective_message,
"You have no right to restrict someone.")
return ""
if member:
if is_user_admin(chat, user_id, member=member):
send_message(update.effective_message,
"I can't stop an admin from talking!")
elif member.can_send_messages is None or member.can_send_messages:
context.bot.restrict_chat_member(
chat.id,
user_id,
permissions=ChatPermissions(can_send_messages=False))
send_message(update.effective_message,
text,
parse_mode=ParseMode.HTML)
return "<b>{}:</b>" \
"\n#MUTE" \
"\n<b>Admin:</b> {}" \
"\n<b>User:</b> {}".format(html.escape(chat.title),
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name))
else:
send_message(update.effective_message,
"This user is already muted!")
else:
send_message(update.effective_message, "This user isn't in the chat!")
return ""
def temp_mute(update, context):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
message = update.effective_message # type: Optional[Message]
args = context.args
user_id, reason = extract_user_and_text(message, args)
if user_id == "error":
send_message(update.effective_message, reason)
return ""
if not user_id:
send_message(update.effective_message,
"You don't seem to be referring to a user.")
return ""
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":
update.effective_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
try:
member = chat.get_member(user_id)
except BadRequest as excp:
if excp.message == "User not found":
send_message(update.effective_message, "I can't find this user")
return ""
else:
raise
if is_user_admin(chat, user_id, member):
send_message(update.effective_message,
"I really wish I could mute admins...")
return ""
if user_id == context.bot.id:
send_message(update.effective_message,
"I'm not gonna MUTE myself, are you crazy?")
return ""
check = context.bot.getChatMember(chat.id, user.id)
if check['can_restrict_members'] == False:
send_message(update.effective_message,
"You have no right to restrict someone.")
return ""
if not reason:
send_message(update.effective_message,
"You haven't specified a time to mute 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#TMUTE" \
"\n<b>Admin:</b> {}" \
"\n<b>User:</b> {}" \
"\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), time_val)
if reason:
log += "\n<b>Reason:</b> {}".format(reason)
try:
if member.can_send_messages is None or member.can_send_messages:
context.bot.restrict_chat_member(
chat.id,
user_id,
until_date=mutetime,
permissions=ChatPermissions(can_send_messages=False))
if conn:
text = "Muted for *{}* in *{}*!".format(time_val, chat_name)
else:
text = "Admin {} Temporary Muted user {} for {}\n<b>Reason:</b> {}".format(
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name),
time_val, reason or "No reason given")
send_message(update.effective_message,
text,
parse_mode=ParseMode.HTML)
return log
else:
send_message(update.effective_message,
"This user is already muted.")
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
send_message(
update.effective_message,
"Admin {} Temporary Muted user {} for {}\n<b>Reason:</b> {}".
format(mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name),
time_val, reason or "No reason given"),
parse_mode=ParseMode.HTML)
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)
send_message(update.effective_message,
"Well damn, I can't mute that user.")
return ""
def unmute(update, context):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
message = update.effective_message # type: Optional[Message]
args = context.args
user_id = extract_user(message, args)
if not user_id or user_id == "error":
send_message(
update.effective_message,
"You'll need to either give me a username to unmute, or reply to someone to be unmuted."
)
return ""
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
text = "This user already has the right to speak in *{}*.".format(
chat_name)
text2 = "Unmuted on *{}*.".format(chat_name)
else:
if update.effective_message.chat.type == "private":
update.effective_send_message(
update.effective_message,
"You can do this command in groups, not PM")
return ""
chat = update.effective_chat
member = chat.get_member(int(user_id))
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
text = "This user already has the right to speak."
text2 = "{} can talk now".format(
mention_html(member.user.id, member.user.first_name))
check = context.bot.getChatMember(chat.id, user.id)
if check['can_restrict_members'] == False:
send_message(update.effective_message,
"You have no right to restrict someone.")
return ""
if member:
if is_user_admin(chat, user_id, member=member):
send_message(update.effective_message,
"This is an admin, what do you expect me to do?")
return ""
elif member.status != 'kicked' and member.status != 'left':
if member.can_send_messages and member.can_send_media_messages \
and member.can_send_other_messages and member.can_add_web_page_previews:
send_message(update.effective_message,
text,
parse_mode=ParseMode.HTML)
return ""
else:
context.bot.restrict_chat_member(
chat.id,
int(user_id),
permissions=ChatPermissions(
can_send_messages=True,
can_send_media_messages=True,
can_send_other_messages=True,
can_add_web_page_previews=True))
send_message(update.effective_message,
text2,
parse_mode=ParseMode.HTML)
return "<b>{}:</b>" \
"\n#UNMUTE" \
"\n<b>Admin:</b> {}" \
"\n<b>User:</b> {}".format(html.escape(chat.title),
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name))
else:
send_message(
update.effective_message,
"This user isn't even in the chat, unmuting them won't make them talk more than they already do!"
)
return ""
def check_update(self, update):
return super().check_update(update) and not (
sql.is_restr_locked(update.effective_chat.id, 'messages') and
not is_user_admin(update.effective_chat, update.effective_user.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,
conn=False) -> str:
bot = dispatcher.bot
if is_user_admin(chat, user.id):
send_message(update.effective_message, "this is admin of this group")
return ""
if warner:
warner_tag = mention_html(warner.id, warner.first_name)
else:
warner_tag = chat.id, "Automated warn filter."
limit, soft_warn, warn_mode = 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 not soft_warn:
if not warn_mode:
chat.unban_member(user.id)
reply = "{} warnings, {} has been kicked!".format(
limit, mention_html(user.id, user.first_name))
elif warn_mode == 1:
chat.unban_member(user.id)
reply = "{} warnings, {} has been kicked!".format(
limit, mention_html(user.id, user.first_name))
elif warn_mode == 2:
chat.kick_member(user.id)
reply = "{} warnings, {} has been banned!".format(
limit, mention_html(user.id, user.first_name))
elif warn_mode == 3:
message.bot.restrict_chat_member(chat.id,
user.id,
can_send_messages=False)
reply = "{} warnings, {} has been muted!".format(
limit, mention_html(user.id, user.first_name))
else:
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 = 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 = [[
InlineKeyboardButton(text="Remove warn",
callback_data="rm_warn({})".format(user.id))
]]
rules = rules_sql.get_rules(chat.id)
if rules:
keyboard[0].append(
InlineKeyboardButton(text="Rules",
url="t.me/{}?start={}".format(
bot.username, chat.id)))
if num_warns + 1 == limit:
if not warn_mode:
action_mode = "Kicked"
elif warn_mode == 1:
action_mode = "Kicked"
elif warn_mode == 2:
action_mode = "banned"
elif warn_mode == 3:
action_mode = "muted"
reply = "{} has {}/{} warnings... If you are warned again, you will be {}!".format(
mention_html(user.id, user.first_name), num_warns, limit,
action_mode)
else:
reply = "{} 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:
if conn:
send_message_raw(chat.id,
reply,
reply_markup=InlineKeyboardMarkup(keyboard),
parse_mode=ParseMode.HTML)
else:
send_message_raw(chat.id,
reply,
reply_to_message_id=message.message_id,
reply_markup=InlineKeyboardMarkup(keyboard),
parse_mode=ParseMode.HTML)
#send_message(update.effective_message, reply, reply_markup=InlineKeyboardMarkup(keyboard), parse_mode=ParseMode.HTML)
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
if conn:
message.bot.sendMessage(
chat.id,
reply,
reply_markup=InlineKeyboardMarkup(keyboard),
parse_mode=ParseMode.HTML)
else:
try:
message.bot.sendMessage(
chat.id,
reply,
reply_to_message_id=message.message_id,
reply_markup=InlineKeyboardMarkup(keyboard),
parse_mode=ParseMode.HTML,
quote=False)
except BadRequest:
message.bot.sendMessage(
chat.id,
reply,
reply_markup=InlineKeyboardMarkup(keyboard),
parse_mode=ParseMode.HTML,
quote=False)
#send_message(update.effective_message, reply, reply_markup=InlineKeyboardMarkup(keyboard), parse_mode=ParseMode.HTML, quote=False)
else:
raise
return log_reason
