def del_lockables(bot: Bot, update: Update):
chat = update.effective_chat # type: Optional[Chat]
message = update.effective_message # type: Optional[Message]
for lockable, filter in LOCK_TYPES.items():
if filter(message) and sql.is_locked(chat.id, lockable) and can_delete(
chat, 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, bot.id):
message.reply_text(
tld(
chat.id,
"I see a bot, and I've been told to stop them joining... but I'm not admin!"
))
return
chat.kick_member(new_mem.id)
message.reply_text(
tld(
chat.id,
"Only admins are allowed to add bots to this chat! Get outta here."
))
else:
try:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("ERROR in lockables")
break
def build_lock_message(chat, chatP, user, chatname):
locks = sql.get_locks(chat.id)
restr = sql.get_restr(chat.id)
if not (locks or restr):
res = tld(chatP.id,
"There are no current locks in *{}*.".format(chatname))
else:
res = tld(chatP.id, "These are the locks in *{}*:".format(chatname))
if locks:
res += "\n - sticker = `{}`" \
"\n - audio = `{}`" \
"\n - voice = `{}`" \
"\n - document = `{}`" \
"\n - video = `{}`" \
"\n - videonote = `{}`" \
"\n - contact = `{}`" \
"\n - photo = `{}`" \
"\n - gif = `{}`" \
"\n - url = `{}`" \
"\n - bots = `{}`" \
"\n - forward = `{}`" \
"\n - game = `{}`" \
"\n - location = `{}`".format(locks.sticker, locks.audio, locks.voice, locks.document,
locks.video, locks.videonote, locks.contact, locks.photo, locks.gif, locks.url,
locks.bots, locks.forward, locks.game, locks.location)
if restr:
res += "\n - messages = `{}`" \
"\n - media = `{}`" \
"\n - other = `{}`" \
"\n - previews = `{}`" \
"\n - all = `{}`".format(restr.messages, restr.media, restr.other, restr.preview,
all([restr.messages, restr.media, restr.other, restr.preview]))
return res
def antispam(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat # type: Optional[Chat]
if len(args) > 0:
if args[0].lower() in ["on", "yes"]:
sql.enable_antispam(chat.id)
update.effective_message.reply_text(
tld(
chat.id,
"I've enabled antispam security in this group. This will help protect you "
"from spammers, unsavoury characters, and the biggest trolls."
))
elif args[0].lower() in ["off", "no"]:
sql.disable_antispam(chat.id)
update.effective_message.reply_text(
tld(
chat.id,
"I've disabled antispam security in this group. GBans wont affect your users "
"anymore. You'll be less protected from any trolls and spammers "
"though!"))
else:
update.effective_message.reply_text(
tld(
chat.id,
"Give me some arguments to choose a setting! on/off, yes/no!\n\n"
"Your current setting is: {}\n"
"When True, any gbans that happen will also happen in your group. "
"When False, they won't, leaving you at the possible mercy of "
"spammers.").format(sql.does_chat_gban(chat.id)))
def start(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat # type: Optional[Chat]
#query = update.callback_query #Unused variable
if update.effective_chat.type == "private":
if len(args) >= 1:
if args[0].lower() == "help":
send_help(update.effective_chat.id, tld(chat.id, "send-help").format(
dispatcher.bot.first_name, "" if not ALLOW_EXCL else tld(
chat.id, "\nAll commands can either be used with `/` or `!`.\n"
)))
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, update, user=False)
else:
send_settings(match.group(1), update.effective_user.id, update, user=True)
elif args[0][1:].isdigit() and "rules" in IMPORTED:
IMPORTED["rules"].send_rules(update, args[0], from_pm=True)
elif args[0].lower() == "controlpanel":
control_panel(bot, update)
else:
send_start(bot, update)
else:
try:
update.effective_message.reply_text("Need me?")
except:
print("Nut")
def adminlist(bot, update):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
conn = connected(bot, update, chat, user.id, need_admin=False)
if not conn == False:
chatP = dispatcher.bot.getChat(conn)
else:
chatP = update.effective_chat
if chat.type == "private":
exit(1)
administrators = chatP.get_administrators()
text = tld(chat.id, "Admins in") + " *{}*:".format(
chatP.title or tld(chat.id, "this chat"))
for admin in administrators:
user = admin.user
status = admin.status
if status == "creator":
name = user.first_name + (user.last_name or "") + tld(
chat.id, " (Creator)")
else:
name = user.first_name + (user.last_name or "")
text += f"\n• `{name}`"
update.effective_message.reply_text(text, parse_mode=ParseMode.MARKDOWN)
def invite(bot: Bot, update: Update):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
conn = connected(bot, update, chat, user.id, need_admin=False)
if not conn == False:
chatP = dispatcher.bot.getChat(conn)
else:
chatP = update.effective_chat
if chat.type == "private":
exit(1)
if chatP.username:
update.effective_message.reply_text(chatP.username)
elif chatP.type == chatP.SUPERGROUP or chatP.type == chatP.CHANNEL:
bot_member = chatP.get_member(bot.id)
if bot_member.can_invite_users:
invitelink = chatP.invite_link
#print(invitelink)
if not invitelink:
invitelink = bot.exportChatInviteLink(chatP.id)
update.effective_message.reply_text(invitelink)
else:
update.effective_message.reply_text(
tld(
chat.id,
"I don't have access to the invite link, try changing my permissions!"
))
else:
update.effective_message.reply_text(
tld(
chat.id,
"I can only give you invite links for supergroups and channels, sorry!"
))
def get_help(bot: Bot, update: Update):
chat = update.effective_chat # type: Optional[Chat]
args = update.effective_message.text.split(None, 1)
# ONLY send help in PM
if chat.type != chat.PRIVATE:
update.effective_message.reply_text("Contact me in PM to get the list of possible commands.",
reply_markup=InlineKeyboardMarkup(
[[InlineKeyboardButton(text="Help",
url="t.me/{}?start=help".format(
bot.username))]]))
return
elif len(args) >= 2 and any(args[1].lower() == x for x in HELPABLE):
module = args[1].lower()
mod_name = tld(chat.id, HELPABLE[module].__mod_name__)
help_txt = tld_help(chat.id, HELPABLE[module].__mod_name__)
if help_txt == False:
help_txt = HELPABLE[module].__help__
text = tld(chat.id, "Here is the help for the *{}* module:\n{}").format(mod_name, help_txt)
send_help(chat.id, text, InlineKeyboardMarkup([[InlineKeyboardButton(text=tld(chat.id, "Back"), callback_data="help_back")]]))
else:
send_help(chat.id, tld(chat.id, "send-help").format(dispatcher.bot.first_name, "" if not ALLOW_EXCL else tld(
chat.id, "\nAll commands can either be used with `/` or `!`.\n"
)))
def setlog(bot: Bot, update: Update):
message = update.effective_message # type: Optional[Message]
chat = update.effective_chat # type: Optional[Chat]
if chat.type == chat.CHANNEL:
message.reply_text(tld(chat.id, "Now, forward the /setlog to the group you want to tie this channel to!"))
elif message.forward_from_chat:
sql.set_chat_log_channel(chat.id, message.forward_from_chat.id)
try:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("Error deleting message in log channel. Should work anyway though.")
try:
bot.send_message(message.forward_from_chat.id, tld(chat.id,
"This channel has been set as the log channel for {}.").format(
chat.title or chat.first_name))
except Unauthorized as excp:
if excp.message == "Forbidden: bot is not a member of the channel chat":
bot.send_message(chat.id, tld(chat.id, "Successfully set log channel!"))
else:
LOGGER.exception("ERROR in setting the log channel.")
bot.send_message(chat.id, tld(chat.id, "Successfully set log channel!"))
else:
message.reply_text(tld(chat.id, "*The steps to set a log channel are:*\n"
" • add bot to the desired channel\n"
" • send /setlog to the channel\n"
" • forward the /setlog to the group\n"), ParseMode.MARKDOWN)
def promote(bot: Bot, update: Update, args: List[str]) -> str:
message = update.effective_message # type: Optional[Message]
user = update.effective_user # type: Optional[User]
chat = update.effective_chat # type: Optional[Chat]
conn = connected(bot, update, chat, user.id)
if not conn == False:
chatD = dispatcher.bot.getChat(conn)
else:
chatD = update.effective_chat
if chat.type == "private":
exit(1)
if not chatD.get_member(bot.id).can_promote_members:
update.effective_message.reply_text(
"I can't promote/demote people here! "
"Make sure I'm admin and can appoint new admins.")
exit(1)
user_id = extract_user(message, args)
if not user_id:
message.reply_text(
tld(chat.id, "You don't seem to be referring to a user."))
return ""
user_member = chatD.get_member(user_id)
if user_member.status == 'administrator' or user_member.status == 'creator':
message.reply_text(
tld(chat.id,
"How am I meant to promote someone that's already an admin?"))
return ""
if user_id == bot.id:
message.reply_text(
tld(chat.id,
"I can't promote myself! Get an admin to do it for me."))
return ""
# set same perms as bot - bot can't assign higher perms than itself!
bot_member = chatD.get_member(bot.id)
bot.promoteChatMember(
chatD.id,
user_id,
can_change_info=bot_member.can_change_info,
can_post_messages=bot_member.can_post_messages,
can_edit_messages=bot_member.can_edit_messages,
can_delete_messages=bot_member.can_delete_messages,
#can_invite_users=bot_member.can_invite_users,
can_restrict_members=bot_member.can_restrict_members,
can_pin_messages=bot_member.can_pin_messages,
can_promote_members=bot_member.can_promote_members)
message.reply_text(tld(chat.id,
f"Successfully promoted in *{chatD.title}*!"),
parse_mode=ParseMode.MARKDOWN)
return f"<b>{html.escape(chatD.title)}:</b>" \
"\n#PROMOTED" \
f"\n<b>Admin:</b> {mention_html(user.id, user.first_name)}" \
f"\n<b>User:</b> {mention_html(user_member.user.id, user_member.user.first_name)}"
def gdpr(bot: Bot, update: Update):
update.effective_message.reply_text(
tld(update.effective_chat.id, "Deleting identifiable data..."))
for mod in GDPR:
mod.__gdpr__(update.effective_user.id)
update.effective_message.reply_text(tld(update.effective_chat.id,
"send_gdpr"),
parse_mode=ParseMode.MARKDOWN)
def __user_info__(user_id, chat_id):
if user_id == dispatcher.bot.id:
return tld(
chat_id,
"I've seen them in... Wow. Are they stalking me? They're in all the same places I am... oh. It's me."
)
num_chats = sql.get_user_num_chats(user_id)
return tld(
chat_id,
"I've seen them in <code>{}</code> chats in total.").format(num_chats)
def unlock(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]
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, "Unlocked {} for everyone!").format(args[0]))
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, "Unlocked {} for everyone!").format(args[0]))
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,
"What are you trying to unlock...? Try /locktypes for the list of lockables"
))
else:
bot.sendMessage(chat.id,
tld(chat.id, "What are you trying to unlock...?"))
return ""
def check_afk(bot, update, user_id, fst_name):
chat = update.effective_chat # type: Optional[Chat]
if sql.is_afk(user_id):
user = sql.check_afk_status(user_id)
if not user.reason:
res = tld(chat.id, f"{fst_name} is AFK!")
else:
res = tld(
chat.id,
f"{fst_name} is AFK! says its because of:\n{user.reason}")
update.effective_message.reply_text(res)
def lock(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]
if can_delete(chat, bot.id):
if len(args) >= 1:
if args[0] in LOCK_TYPES:
sql.update_lock(chat.id, args[0], locked=True)
message.reply_text(
tld(chat.id,
"Locked {} messages for all non-admins!").format(
args[0]))
return "<b>{}:</b>" \
"\n#LOCK" \
"\n<b>Admin:</b> {}" \
"\nLocked <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=True)
if args[0] == "previews":
members = users_sql.get_chat_members(str(chat.id))
restr_members(bot,
chat.id,
members,
messages=True,
media=True,
other=True)
message.reply_text(
tld(chat.id,
"Locked {} for all non-admins!").format(args[0]))
return "<b>{}:</b>" \
"\n#LOCK" \
"\n<b>Admin:</b> {}" \
"\nLocked <code>{}</code>.".format(html.escape(chat.title),
mention_html(user.id, user.first_name), args[0])
else:
message.reply_text(
tld(
chat.id,
"What are you trying to lock...? Try /locktypes for the list of lockables"
))
else:
message.reply_text(
tld(
chat.id,
"Make sure I'm a group administrator and have permission to delete messages, then try again."
))
return ""
def unsetlog(bot: Bot, update: Update):
message = update.effective_message # type: Optional[Message]
chat = update.effective_chat # type: Optional[Chat]
log_channel = sql.stop_chat_logging(chat.id)
if log_channel:
try:
bot.send_message(log_channel, tld(chat.id, "Channel has been unlinked from {}").format(chat.title))
message.reply_text(tld(chat.id, "Log channel has been un-set."))
except:
print("Nut")
else:
message.reply_text(tld(chat.id, "No log channel has been set yet!"))
def markdown_help(bot: Bot, update: Update):
chat = update.effective_chat # type: Optional[Chat]
update.effective_message.reply_text(tld(chat.id, "MARKDOWN_HELP-K"),
parse_mode=ParseMode.HTML)
update.effective_message.reply_text(
tld(chat.id,
"Try forwarding the following message to me, and you'll see!"))
update.effective_message.reply_text(
tld(
chat.id,
"/save test This is a markdown test. _italics_, *bold*, `code`, "
"[URL](example.com) [button](buttonurl:github.com) "
"[button2](buttonurl://google.com:same)"))
def flood(bot: Bot, update: Update):
chat = update.effective_chat # type: Optional[Chat]
limit = sql.get_flood_limit(chat.id)
if limit == 0:
update.effective_message.reply_text(
tld(chat.id, "I'm not currently enforcing flood control!"))
else:
update.effective_message.reply_text(
tld(
chat.id,
"I'm currently muting users if they send more than {} consecutive messages."
).format(limit))
def __user_info__(user_id, chat_id):
is_gbanned = sql.is_user_gbanned(user_id)
is_gmuted = sql.is_user_gmuted(user_id)
if not user_id in SUDO_USERS:
text = tld(chat_id, "Globally banned: <b>{}</b>")
if is_gbanned:
text = text.format(tld(chat_id, "Yes"))
user = sql.get_gbanned_user(user_id)
if user.reason:
text += tld(chat_id,
"\nReason: {}").format(html.escape(user.reason))
else:
text = text.format(tld(chat_id, "No"))
text += tld(chat_id, "\nGlobally muted: <b>{}</b>")
if is_gmuted:
text = text.format(tld(chat_id, "Yes"))
user = sql.get_gmuted_user(user_id)
if user.reason:
text += tld(chat_id,
"\nReason: {}").format(html.escape(user.reason))
else:
text = text.format(tld(chat_id, "No"))
return text
else:
return ""
def set_flood(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]
if len(args) >= 1:
val = args[0].lower()
if val == "off" or val == "no" or val == "0":
sql.set_flood(chat.id, 0)
message.reply_text(tld(chat.id, "Antiflood has been disabled."))
elif val.isdigit():
amount = int(val)
if amount <= 0:
sql.set_flood(chat.id, 0)
message.reply_text(tld(chat.id,
"Antiflood has been disabled."))
return "<b>{}:</b>" \
"\n#SETFLOOD" \
"\n<b>Admin:</b> {}" \
"\nDisabled antiflood.".format(html.escape(chat.title), mention_html(user.id, user.first_name))
elif amount < 3:
message.reply_text(
tld(
chat.id,
"Antiflood has to be either 0 (disabled), or a number bigger than 3 (enabled)!"
))
return ""
else:
sql.set_flood(chat.id, amount)
message.reply_text(
tld(chat.id,
"Antiflood has been updated and set to {}").format(
amount))
return "<b>{}:</b>" \
"\n#SETFLOOD" \
"\n<b>Admin:</b> {}" \
"\nSet antiflood to <code>{}</code>.".format(html.escape(chat.title),
mention_html(user.id, user.first_name), amount)
else:
message.reply_text(
tld(
chat.id,
"Unrecognized argument - please use a number, 'off', or 'no'."
))
return ""
def help_button(bot: Bot, update: Update):
query = update.callback_query
chat = update.effective_chat # type: Optional[Chat]
mod_match = re.match(r"help_module\((.+?)\)", query.data)
prev_match = re.match(r"help_prev\((.+?)\)", query.data)
next_match = re.match(r"help_next\((.+?)\)", query.data)
back_match = re.match(r"help_back", query.data)
try:
if mod_match:
module = mod_match.group(1)
mod_name = tld(chat.id, HELPABLE[module].__mod_name__)
help_txt = tld_help(chat.id, HELPABLE[module].__mod_name__)
if help_txt == False:
help_txt = HELPABLE[module].__help__
text = tld(chat.id, "Here is the help for the *{}* module:\n{}").format(mod_name, help_txt)
query.message.reply_text(text=text,
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup(
[[InlineKeyboardButton(text=tld(chat.id, "Back"), callback_data="help_back")]]))
elif prev_match:
curr_page = int(prev_match.group(1))
query.message.reply_text(tld(chat.id, "send-help").format(
dispatcher.bot.first_name, "" if not ALLOW_EXCL else tld(chat.id, "\nAll commands can either be used with `/` or `!`.\n")
),
parse_mode=ParseMode.MARKDOWN, reply_markup=InlineKeyboardMarkup(paginate_modules(chat.id, curr_page - 1, HELPABLE, "help")))
elif next_match:
next_page = int(next_match.group(1))
query.message.reply_text(tld(chat.id, "send-help").format(
dispatcher.bot.first_name, "" if not ALLOW_EXCL else tld(chat.id, "\nAll commands can either be used with `/` or `!`.\n")
),
parse_mode=ParseMode.MARKDOWN, reply_markup=InlineKeyboardMarkup(paginate_modules(chat.id, next_page + 1, HELPABLE, "help")))
elif back_match:
query.message.reply_text(text=tld(chat.id, "send-help").format(
dispatcher.bot.first_name, "" if not ALLOW_EXCL else tld(chat.id, "\nAll commands can either be used with `/` or `!`.\n")
),
parse_mode=ParseMode.MARKDOWN, reply_markup=InlineKeyboardMarkup(paginate_modules(chat.id, 0, HELPABLE, "help")))
# ensure no spinny white circle
bot.answer_callback_query(query.id)
query.message.delete()
except BadRequest as excp:
if excp.message == "Message is not modified":
pass
elif excp.message == "Query_id_invalid":
pass
elif excp.message == "Message can't be deleted":
pass
else:
LOGGER.exception("Exception in help buttons. %s", str(query.data))
def locale(bot, update, args):
chat = update.effective_chat
if len(args) > 0:
locale = args[0].lower()
if locale in list_locales:
if locale in ('en', 'ru', 'ua', 'es', 'tr', 'id'):
switch_to_locale(chat.id, locale)
update.message.reply_text(
tld(chat.id, 'Switched to {} successfully!').format(
list_locales[locale]))
else:
update.message.reply_text("{} is not supported yet!".format(
list_locales[locale]))
else:
update.message.reply_text(
"Is that even a valid language code? Use an internationally accepted ISO code!"
)
else:
LANGUAGE = prev_locale(chat.id)
if LANGUAGE:
locale = LANGUAGE.locale_name
native_lang = list_locales[locale]
update.message.reply_text(
"Current locale for this chat is: *{}*".format(native_lang),
parse_mode=ParseMode.MARKDOWN)
else:
update.message.reply_text(
"Current locale for this chat is: *English*",
parse_mode=ParseMode.MARKDOWN)
def runs(bot: Bot, update: Update):
spam = spamfilters(update.effective_message.text,
update.effective_message.from_user.id,
update.effective_chat.id, update.effective_message)
if spam == True:
return
chat = update.effective_chat # type: Optional[Chat]
update.effective_message.reply_text(random.choice(tld(chat.id, "RUNS-K")))
def paginate_modules(chat_id,
page_n: int,
module_dict: Dict,
prefix,
chat=None) -> List:
if not chat:
modules = sorted([
EqInlineKeyboardButton(tld(chat_id, x.__mod_name__),
callback_data="{}_module({})".format(
prefix, x.__mod_name__.lower()))
for x in module_dict.values()
])
else:
modules = sorted([
EqInlineKeyboardButton(tld(chat_id, x.__mod_name__),
callback_data="{}_module({},{})".format(
prefix, chat, x.__mod_name__.lower()))
for x in module_dict.values()
])
pairs = list(zip(modules[::2], modules[1::2]))
if len(modules) % 2 == 1:
pairs.append((modules[-1], ))
max_num_pages = ceil(len(pairs) / 7)
modulo_page = page_n % max_num_pages
# can only have a certain amount of buttons side by side
if len(pairs) > 7:
pairs = pairs[modulo_page * 7:7 * (modulo_page + 1)] + [
(EqInlineKeyboardButton(
"<<", callback_data="{}_prev({})".format(prefix, modulo_page)),
EqInlineKeyboardButton("⬅️ Back", callback_data="bot_start"),
EqInlineKeyboardButton(">>",
callback_data="{}_next({})".format(
prefix, modulo_page)))
]
else:
pairs += [[
EqInlineKeyboardButton("⬅️ Back", callback_data="bot_start")
]]
return pairs
def add_blacklist(bot: Bot, update: Update):
msg = update.effective_message # type: Optional[Message]
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
words = msg.text.split(None, 1)
conn = connected(bot, update, chat, user.id)
if not conn == False:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
exit(1)
else:
chat_name = chat.title
if len(words) > 1:
text = words[1]
to_blacklist = list(
set(trigger.strip() for trigger in text.split("\n")
if trigger.strip()))
for trigger in to_blacklist:
sql.add_to_blacklist(chat_id, trigger.lower())
if len(to_blacklist) == 1:
msg.reply_text(tld(
chat.id,
"Added <code>{}</code> to the blacklist in <b>{}</b>!").format(
html.escape(to_blacklist[0]), chat_name),
parse_mode=ParseMode.HTML)
else:
msg.reply_text(tld(
chat.id,
"Added <code>{}</code> to the blacklist in <b>{}</b>!").format(
len(to_blacklist)),
chat_name,
parse_mode=ParseMode.HTML)
else:
msg.reply_text(
tld(chat.id,
"Tell me what words you would like to add to the blacklist."))
def check_flood(bot: Bot, update: Update) -> 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:
bot.restrict_chat_member(chat.id, user.id, can_send_messages=False)
msg.reply_text(
tld(
chat.id,
"I like to leave the flooding to natural disasters. But you, you were just a "
"disappointment. *Muted*!"))
return "<b>{}:</b>" \
"\n#MUTED" \
"\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 "<b>{}:</b>" \
"\n#INFO" \
"\nDon't have mute permissions, so automatically disabled antiflood.".format(chat.title)
def afk(bot: Bot, update: Update):
chat = update.effective_chat # type: Optional[Chat]
args = update.effective_message.text.split(None, 1)
if len(args) >= 2:
reason = args[1]
else:
reason = ""
sql.set_afk(update.effective_user.id, reason)
fname = update.effective_user.first_name
update.effective_message.reply_text(
tld(chat.id, f"{fname} have taken rest from Virtual World"))
def blacklist(bot: Bot, update: Update, args: List[str]):
msg = update.effective_message # type: Optional[Message]
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
conn = connected(bot, update, chat, user.id, need_admin=False)
if not conn == False:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if chat.type == "private":
exit(1)
else:
chat_id = update.effective_chat.id
chat_name = chat.title
filter_list = tld(
chat.id, "<b>Current blacklisted words in {}:</b>\n").format(chat_name)
all_blacklisted = sql.get_chat_blacklist(chat_id)
if len(args) > 0 and args[0].lower() == 'copy':
for trigger in all_blacklisted:
filter_list += "<code>{}</code>\n".format(html.escape(trigger))
else:
for trigger in all_blacklisted:
filter_list += " • <code>{}</code>\n".format(html.escape(trigger))
split_text = split_message(filter_list)
for text in split_text:
if filter_list == tld(
chat.id, "<b>Current blacklisted words in {}:</b>\n").format(
chat_name): #We need to translate
msg.reply_text(tld(
chat.id,
"There are no blacklisted messages in <b>{}</b>!").format(
chat_name),
parse_mode=ParseMode.HTML)
return
msg.reply_text(text, parse_mode=ParseMode.HTML)
def get_id(bot: Bot, update: Update, args: List[str]):
spam = spamfilters(update.effective_message.text,
update.effective_message.from_user.id,
update.effective_chat.id, update.effective_message)
if spam == True:
return
user_id = extract_user(update.effective_message, args)
chat = update.effective_chat # type: Optional[Chat]
if user_id:
if update.effective_message.reply_to_message and update.effective_message.reply_to_message.forward_from:
user1 = update.effective_message.reply_to_message.from_user
user2 = update.effective_message.reply_to_message.forward_from
update.effective_message.reply_text(tld(
chat.id,
"The original sender, {}, has an ID of `{}`.\nThe forwarder, {}, has an ID of `{}`."
).format(escape_markdown(user2.first_name), user2.id,
escape_markdown(user1.first_name), user1.id),
parse_mode=ParseMode.MARKDOWN)
else:
user = bot.get_chat(user_id)
update.effective_message.reply_text(tld(
chat.id,
"{}'s id is `{}`.").format(escape_markdown(user.first_name),
user.id),
parse_mode=ParseMode.MARKDOWN)
else:
chat = update.effective_chat # type: Optional[Chat]
if chat.type == "private":
update.effective_message.reply_text(tld(
chat.id, "Your id is `{}`.").format(chat.id),
parse_mode=ParseMode.MARKDOWN)
else:
update.effective_message.reply_text(tld(
chat.id, "This group's id is `{}`.").format(chat.id),
parse_mode=ParseMode.MARKDOWN)
def no_longer_afk(bot: Bot, update: Update):
user = update.effective_user # type: Optional[User]
chat = update.effective_chat # type: Optional[Chat]
if not user: # ignore channels
return
res = sql.rm_afk(user.id)
if res:
firstname = update.effective_user.first_name
try:
update.effective_message.reply_text(
tld(chat.id, f"{firstname} is back Online"))
except:
return
def allow_connections(bot: Bot, update: Update, args: List[str]) -> str:
chat = update.effective_chat # type: Optional[Chat]
if chat.type != chat.PRIVATE:
if len(args) >= 1:
var = args[0]
print(var)
if var == "no":
sql.set_allow_connect_to_chat(chat.id, False)
update.effective_message.reply_text(
tld(chat.id,
"Disabled connections to this chat for users"))
elif var == "yes":
sql.set_allow_connect_to_chat(chat.id, True)
update.effective_message.reply_text(
tld(chat.id, "Enabled connections to this chat for users"))
else:
update.effective_message.reply_text(
tld(chat.id, "Please enter on/yes/off/no in group!"))
else:
update.effective_message.reply_text(
tld(chat.id, "Please enter on/yes/off/no in group!"))
else:
update.effective_message.reply_text(
tld(chat.id, "Please enter on/yes/off/no in group!"))
