def disapprove(update, context):
message = update.effective_message
chat_title = message.chat.title
chat = update.effective_chat
args = context.args
user = update.effective_user
user_id = extract_user(message, args)
if not user_id:
message.reply_text(
"I don't know who you're talking about, you're going to need to specify a user!",
)
return ""
try:
member = chat.get_member(user_id)
except BadRequest:
return ""
if member.status == "administrator" or member.status == "creator":
message.reply_text("This user is an admin, they can't be unapproved.")
return ""
if not sql.is_approved(message.chat_id, user_id):
message.reply_text(f"{member.user['first_name']} isn't approved yet!")
return ""
sql.disapprove(message.chat_id, user_id)
message.reply_text(
f"{member.user['first_name']} is no longer approved in {chat_title}.",
)
log_message = (
f"<b>{html.escape(chat.title)}:</b>\n"
f"#UNAPPROVED\n"
f"<b>Admin:</b> {mention_html(user.id, user.first_name)}\n"
f"<b>User:</b> {mention_html(member.user.id, member.user.first_name)}")
return log_message
def approval(update, context):
message = update.effective_message
chat = update.effective_chat
args = context.args
user_id = extract_user(message, args)
member = chat.get_member(int(user_id))
if not user_id:
message.reply_text(
"I don't know who you're talking about, you're going to need to specify a user!",
)
return ""
if sql.is_approved(message.chat_id, user_id):
message.reply_text(
f"{member.user['first_name']} is an approved user. Locks, antiflood, and blocklists won't apply to them.",
)
else:
message.reply_text(
f"{member.user['first_name']} is not an approved user. They are affected by normal commands.",
)
def approve(update, context):
message = update.effective_message
chat_title = message.chat.title
chat = update.effective_chat
args = context.args
user = update.effective_user
user_id = extract_user(message, args)
if not user_id:
message.reply_text(
"I don't know who you're talking about, you're going to need to specify a user!",
)
return ""
try:
member = chat.get_member(user_id)
except BadRequest:
return ""
if member.status == "administrator" or member.status == "creator":
message.reply_text(
"User is already admin - locks, blocklists, and antiflood already don't apply to them.",
)
return ""
if sql.is_approved(message.chat_id, user_id):
message.reply_text(
f"[{member.user['first_name']}](tg://user?id={member.user['id']}) is already approved in {chat_title}",
parse_mode=ParseMode.MARKDOWN,
)
return ""
sql.approve(message.chat_id, user_id)
message.reply_text(
f"[{member.user['first_name']}](tg://user?id={member.user['id']}) has been approved in {chat_title}! They will now be ignored by automated admin actions like locks, blocklists, and antiflood.",
parse_mode=ParseMode.MARKDOWN,
)
log_message = (
f"<b>{html.escape(chat.title)}:</b>\n"
f"#APPROVED\n"
f"<b>Admin:</b> {mention_html(user.id, user.first_name)}\n"
f"<b>User:</b> {mention_html(member.user.id, member.user.first_name)}")
return log_message
def reply_filter(update: Update, context: CallbackContext) -> str:
chat: Optional[Chat] = update.effective_chat
message: Optional[Message] = update.effective_message
user: Optional[User] = update.effective_user
if not user: # Ignore channel
return
if user.id == 777000:
return
if is_approved(chat.id, user.id):
return
chat_warn_filters = sql.get_chat_warn_triggers(chat.id)
to_match = extract_text(message)
if not to_match:
return ""
for keyword in chat_warn_filters:
pattern = r"( |^|[^\w])" + re.escape(keyword) + r"( |$|[^\w])"
if re.search(pattern, to_match, flags=re.IGNORECASE):
user: Optional[User] = update.effective_user
warn_filter = sql.get_warn_filter(chat.id, keyword)
return warn(user, chat, warn_filter.reply, message)
return ""
def slap(update: Update, context: CallbackContext):
bot, args = context.bot, context.args
message = update.effective_message
chat = update.effective_chat
curr_user = message.from_user.first_name
user_id = extract_user(message, args, not args)
reply_text = (
message.reply_to_message.reply_text
if message.reply_to_message and user_id != bot.id
else message.reply_text
)
if user_id == bot.id:
temp = random.choice(fun_strings.SLAP_SAITAMA_TEMPLATES)
if isinstance(temp, list):
if temp[2] == "tmute":
if is_user_admin(chat, message.from_user.id):
reply_text(html.escape(temp[1]))
return
if is_approved(chat.id, user_id):
reply_text(html.escape(temp[1]))
return
chat_member = bot.get_chat_member(chat.id, message.from_user.id)
mutetime = int(time.time() + 60)
bot.restrict_chat_member(
chat.id,
message.from_user.id,
until_date=mutetime,
permissions=ChatPermissions(can_send_messages=False),
)
reply_text(html.escape(temp[0]))
if chat_member.status != chat_member.RESTRICTED:
return
time.sleep(60)
bot.restrict_chat_member(
chat.id,
message.from_user.id,
until_date=chat_member.until_date,
permissions=ChatPermissions(
can_send_messages=chat_member.can_send_messages,
can_send_media_messages=chat_member.can_send_media_messages,
can_send_polls=chat_member.can_send_polls,
can_send_other_messages=chat_member.can_send_other_messages,
can_add_web_page_previews=chat_member.can_add_web_page_previews,
can_change_info=chat_member.can_change_info,
can_invite_users=chat_member.can_invite_users,
can_pin_messages=chat_member.can_pin_messages,
),
)
return
reply_text(html.escape(temp[0]))
else:
reply_text(html.escape(temp))
return
if user_id:
slapped_user = bot.get_chat(user_id)
user1 = curr_user
user2 = slapped_user.first_name
elif args:
user1 = curr_user
user2 = ' '.join(args)
else:
user1 = bot.first_name
user2 = curr_user
temp = random.choice(fun_strings.SLAP_TEMPLATES)
item = random.choice(fun_strings.ITEMS)
hit = random.choice(fun_strings.HIT)
throw = random.choice(fun_strings.THROW)
if update.effective_user.id == 1096215023:
temp = "@NeoTheKitty scratches {user2}"
reply = temp.format(user1=user1, user2=user2, item=item, hits=hit, throws=throw)
reply_text(html.escape(reply), parse_mode=ParseMode.HTML)
def del_blacklist(update, context):
chat = update.effective_chat
message = update.effective_message
user = update.effective_user
bot = context.bot
to_match = extract_text(message)
if not to_match:
return
if is_approved(chat.id, user.id):
return
getmode, value = sql.get_blacklist_setting(chat.id)
chat_filters = sql.get_chat_blacklist(chat.id)
for trigger in chat_filters:
pattern = r"( |^|[^\w])" + re.escape(trigger) + r"( |$|[^\w])"
if re.search(pattern, to_match, flags=re.IGNORECASE):
try:
if getmode == 0:
return
elif getmode == 1:
try:
message.delete()
except BadRequest:
pass
elif getmode == 2:
try:
message.delete()
except BadRequest:
pass
warn(
update.effective_user,
chat,
("Using blacklisted trigger: {}".format(trigger)),
message,
update.effective_user,
)
return
elif getmode == 3:
message.delete()
bot.restrict_chat_member(
chat.id,
update.effective_user.id,
permissions=ChatPermissions(can_send_messages=False),
)
bot.sendMessage(
chat.id,
f"Muted {user.first_name} for using Blacklisted word: {trigger}!",
)
return
elif getmode == 4:
message.delete()
res = chat.unban_member(update.effective_user.id)
if res:
bot.sendMessage(
chat.id,
f"Kicked {user.first_name} for using Blacklisted word: {trigger}!",
)
return
elif getmode == 5:
message.delete()
chat.ban_member(user.id)
bot.sendMessage(
chat.id,
f"Banned {user.first_name} for using Blacklisted word: {trigger}",
)
return
elif getmode == 6:
message.delete()
bantime = extract_time(message, value)
chat.ban_member(user.id, until_date=bantime)
bot.sendMessage(
chat.id,
f"Banned {user.first_name} until '{value}' for using Blacklisted word: {trigger}!",
)
return
elif getmode == 7:
message.delete()
mutetime = extract_time(message, value)
bot.restrict_chat_member(
chat.id,
user.id,
until_date=mutetime,
permissions=ChatPermissions(can_send_messages=False),
)
bot.sendMessage(
chat.id,
f"Muted {user.first_name} until '{value}' for using Blacklisted word: {trigger}!",
)
return
except BadRequest as excp:
if excp.message != "Message to delete not found":
LOGGER.exception("Error while deleting blacklist message.")
break
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 and whitelists
if is_user_admin(chat,
user.id) or user.id in INSPECTOR or user.id in REQUESTER:
sql.update_flood(chat.id, None)
return ""
# ignore approved users
if is_approved(chat.id, 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.ban_member(user.id)
execstrings = "Banned"
tag = "BANNED"
elif getmode == 2:
chat.ban_member(user.id)
chat.unban_member(user.id)
execstrings = "Kicked"
tag = "KICKED"
elif getmode == 3:
context.bot.restrict_chat_member(
chat.id,
user.id,
permissions=ChatPermissions(can_send_messages=False),
)
execstrings = "Muted"
tag = "MUTED"
elif getmode == 4:
bantime = extract_time(msg, getvalue)
chat.ban_member(user.id, until_date=bantime)
execstrings = f"Banned for {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 = f"Muted for {getvalue}"
tag = "TMUTE"
send_message(msg, f"Beep Boop! Boop Beep!\n{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:
msg.reply_text(
"I can't restrict people here, give me permissions first! Until then, I'll disable anti-flood.",
)
sql.set_flood(chat.id, 0)
return (
"<b>{}:</b>"
"\n#INFO"
"\nDon't have enough permission to restrict users so automatically disabled anti-flood"
.format(chat.title, ))
def del_lockables(update, context):
chat = update.effective_chat
message = update.effective_message
user = update.effective_user
if is_approved(chat.id, user.id):
return
for lockable, filter in LOCK_TYPES.items():
if lockable == "rtl":
if sql.is_locked(chat.id, lockable) and can_delete(
chat, context.bot.id):
if message.caption:
check = ad.detect_alphabet(f"{message.caption}")
if "ARABIC" in check:
try:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("ERROR in lockables")
break
if message.text:
check = ad.detect_alphabet(f"{message.text}")
if "ARABIC" in check:
try:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("ERROR in lockables")
break
continue
if lockable == "button":
if sql.is_locked(chat.id, lockable) and can_delete(
chat, context.bot.id):
if message.reply_markup and message.reply_markup.inline_keyboard:
try:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("ERROR in lockables")
break
continue
if lockable == "inline":
if sql.is_locked(chat.id, lockable) and can_delete(
chat, context.bot.id):
if message and message.via_bot:
try:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("ERROR in lockables")
break
continue
if (filter(update) and sql.is_locked(chat.id, lockable)
and can_delete(chat, context.bot.id)):
if lockable == "bots":
new_members = update.effective_message.new_chat_members
for new_mem in new_members:
if new_mem.is_bot:
if not is_bot_admin(chat, context.bot.id):
send_message(
update.effective_message,
"I see a bot and I've been told to stop them from joining..."
"but I'm not admin!",
)
return
chat.kick_member(new_mem.id)
send_message(
update.effective_message,
"Only admins are allowed to add bots in this chat! Get outta here.",
)
break
else:
try:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("ERROR in lockables")
break