def invite(bot: Bot, update: Update):
chat = update.effective_chat
user = update.effective_user
conn = connected(bot, update, chat, user.id, need_admin=False)
if conn:
chatP = dispatcher.bot.getChat(conn)
else:
chatP = update.effective_chat
if chat.type == "private":
return
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, "admin_err_no_perm_invitelink"))
else:
update.effective_message.reply_text(
tld(chat.id, "admin_chat_no_invitelink"))
def blacklist(bot: Bot, update: Update, args: List[str]):
msg = update.effective_message
chat = update.effective_chat
user = update.effective_user
conn = connected(bot, update, chat, user.id, need_admin=False)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if chat.type == "private":
return
else:
chat_id = update.effective_chat.id
chat_name = chat.title
filter_list = tld(chat.id, "blacklist_active_list").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, "blacklist_active_list").format(
chat_name): #We need to translate
msg.reply_text(tld(chat.id, "blacklist_no_list").format(chat_name),
parse_mode=ParseMode.HTML)
return
msg.reply_text(text, parse_mode=ParseMode.HTML)
def locale_button(bot, update):
chat = update.effective_chat
user = update.effective_user
query = update.callback_query
lang_match = re.findall(r"en-US|en-GB|id|ru|tr", query.data)
if lang_match:
if lang_match[0]:
switch_to_locale(chat.id, lang_match[0])
query.answer(text=tld(chat.id, 'language_switch_success_pm').
format(list_locales[lang_match[0]]))
else:
query.answer(text="Error!", show_alert=True)
try:
LANGUAGE = prev_locale(chat.id)
locale = LANGUAGE.locale_name
curr_lang = list_locales[locale]
except Exception:
curr_lang = "English (US)"
text = tld(chat.id, "language_select_language")
text += tld(chat.id, "language_user_language").format(curr_lang)
conn = connected(bot, update, chat, user.id, need_admin=False)
if conn:
try:
chatlng = prev_locale(conn).locale_name
chatlng = list_locales[chatlng]
text += tld(chat.id, "language_chat_language").format(chatlng)
except Exception:
chatlng = "English (US)"
text += tld(chat.id, "language_sel_user_lang")
bot.edit_message_text(
chat_id=query.message.chat_id,
message_id=query.message.message_id,
text=text,
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup([[
InlineKeyboardButton("English (US) 🇺🇸",
callback_data="set_lang_en-US"),
InlineKeyboardButton("English (UK) 🇬🇧",
callback_data="set_lang_en-GB")
]] + [[
InlineKeyboardButton("Indonesian 🇮🇩",
callback_data="set_lang_id"),
InlineKeyboardButton("Russian 🇷🇺",
callback_data="set_lang_ru"),
InlineKeyboardButton("Turkish 🇹🇷",
callback_data="set_lang_tr")
]] + [[
InlineKeyboardButton(f"{tld(chat.id, 'btn_go_back')}",
callback_data="bot_start")
]]))
# query.message.delete()
bot.answer_callback_query(query.id)
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 demote(bot: Bot, update: Update, args: List[str]) -> str:
chat = update.effective_chat
message = update.effective_message
user = update.effective_user
conn = connected(bot, update, chat, user.id)
if conn:
chatD = dispatcher.bot.getChat(conn)
else:
chatD = update.effective_chat
if chat.type == "private":
return
if not chatD.get_member(bot.id).can_promote_members:
update.effective_message.reply_text(tld(chat.id, "admin_err_no_perm"))
return
user_id = extract_user(message, args)
if not user_id:
message.reply_text(tld(chat.id, "common_err_no_user"))
return ""
user_member = chatD.get_member(user_id)
if user_member.status == 'creator':
message.reply_text(tld(chat.id, "admin_err_demote_creator"))
return ""
if not user_member.status == 'administrator':
message.reply_text(tld(chat.id, "admin_err_demote_noadmin"))
return ""
if user_id == bot.id:
message.reply_text(tld(chat.id, "admin_err_self_demote"))
return ""
try:
bot.promoteChatMember(int(chatD.id),
int(user_id),
can_change_info=False,
can_post_messages=False,
can_edit_messages=False,
can_delete_messages=False,
can_invite_users=False,
can_restrict_members=False,
can_pin_messages=False,
can_promote_members=False)
message.reply_text(tld(chat.id, "admin_demote_success").format(
mention_html(user.id, user.first_name),
mention_html(user_member.user.id, user_member.user.first_name),
html.escape(chatD.title)),
parse_mode=ParseMode.HTML)
return f"<b>{html.escape(chatD.title)}:</b>" \
"\n#DEMOTED" \
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)}"
except BadRequest:
message.reply_text(tld(chat.id, "admin_err_cant_demote"))
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 unblacklist(bot: Bot, update: Update):
msg = update.effective_message
chat = update.effective_chat
user = update.effective_user
words = msg.text.split(None, 1)
conn = connected(bot, update, chat, user.id)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
return
else:
chat_name = chat.title
if len(words) > 1:
text = words[1]
to_unblacklist = list(
set(trigger.strip() for trigger in text.split("\n")
if trigger.strip()))
successful = 0
for trigger in to_unblacklist:
success = sql.rm_from_blacklist(chat_id, trigger.lower())
if success:
successful += 1
if len(to_unblacklist) == 1:
if successful:
msg.reply_text(tld(chat.id, "blacklist_del").format(
html.escape(to_unblacklist[0]), chat_name),
parse_mode=ParseMode.HTML)
else:
msg.reply_text(tld(chat.id, "blacklist_err_not_trigger"))
elif successful == len(to_unblacklist):
msg.reply_text(tld(chat.id, "blacklist_multi_del").format(
successful, chat_name),
parse_mode=ParseMode.HTML)
elif not successful:
msg.reply_text(tld(chat.id,
"blacklist_err_multidel_no_trigger").format(
successful,
len(to_unblacklist) - successful),
parse_mode=ParseMode.HTML)
else:
msg.reply_text(tld(
chat.id, "blacklist_err_multidel_some_no_trigger").format(
successful, chat_name,
len(to_unblacklist) - successful),
parse_mode=ParseMode.HTML)
else:
msg.reply_text(tld(chat.id, "blacklist_err_del_no_args"))
def promote(bot: Bot, update: Update, args: List[str]) -> str:
message = update.effective_message
user = update.effective_user
chat = update.effective_chat
conn = connected(bot, update, chat, user.id)
if conn:
chatD = dispatcher.bot.getChat(conn)
else:
chatD = update.effective_chat
if chat.type == "private":
return
if not chatD.get_member(bot.id).can_promote_members:
update.effective_message.reply_text(tld(chat.id, "admin_err_no_perm"))
return
user_id = extract_user(message, args)
if not user_id:
message.reply_text(tld(chat.id, "common_err_no_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, "admin_err_user_admin"))
return
if user_id == bot.id:
message.reply_text(tld(chat.id, "admin_err_self_promote"))
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, "admin_promote_success").format(
mention_html(user.id, user.first_name),
mention_html(user_member.user.id, user_member.user.first_name),
html.escape(chatD.title)),
parse_mode=ParseMode.HTML)
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 save(bot: Bot, update: Update):
chat = update.effective_chat
user = update.effective_user
conn = connected(bot, update, chat, user.id)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
chat_name = tld(chat.id, "note_is_local")
else:
chat_name = chat.title
msg = update.effective_message
note_name, text, data_type, content, buttons = get_note_type(msg)
note_name = note_name.lower()
if data_type is None:
msg.reply_text(tld(chat.id, "save_invalid"))
return
if len(text.strip()) == 0:
text = note_name
if not sql.get_note(chat_id, note_name):
sql.add_note_to_db(chat_id,
note_name,
text,
data_type,
buttons=buttons,
file=content)
msg.reply_text(tld(chat.id,
"save_success").format(note_name, chat_name,
note_name, note_name),
parse_mode=ParseMode.MARKDOWN)
else:
sql.add_note_to_db(chat_id,
note_name,
text,
data_type,
buttons=buttons,
file=content)
msg.reply_text(tld(chat.id,
"save_updated").format(note_name, chat_name,
note_name, note_name),
parse_mode=ParseMode.MARKDOWN)
def add_blacklist(bot: Bot, update: Update):
msg = update.effective_message
chat = update.effective_chat
user = update.effective_user
words = msg.text.split(None, 1)
conn = connected(bot, update, chat, user.id)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
return
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, "blacklist_add").format(
html.escape(to_blacklist[0]), chat_name),
parse_mode=ParseMode.HTML)
else:
msg.reply_text(tld(chat.id,
"blacklist_add").format(len(to_blacklist)),
chat_name,
parse_mode=ParseMode.HTML)
else:
msg.reply_text(tld(chat.id, "blacklist_err_add_no_args"))
def list_handlers(bot: Bot, update: Update):
chat = update.effective_chat
user = update.effective_user
conn = connected(bot, update, chat, user.id, need_admin=False)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
filter_list = tld(chat.id, "cust_filters_list")
else:
chat_id = update.effective_chat.id
if chat.type == "private":
chat_name = tld(chat.id, "cust_filters_local")
else:
chat_name = chat.title
filter_list = tld(chat.id, "cust_filters_list")
all_handlers = sql.get_chat_triggers(chat_id)
if not all_handlers:
update.effective_message.reply_text(
tld(chat.id, "cust_filters_list_empty").format(chat_name))
return
for keyword in all_handlers:
entry = " • `{}`\n".format(escape_markdown(keyword))
if len(entry) + len(filter_list) > telegram.MAX_MESSAGE_LENGTH:
update.effective_message.reply_text(
filter_list.format(chat_name),
parse_mode=telegram.ParseMode.MARKDOWN)
filter_list = entry
else:
filter_list += entry
update.effective_message.reply_text(filter_list.format(chat_name),
parse_mode=telegram.ParseMode.MARKDOWN)
def clear(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
user = update.effective_user
conn = connected(bot, update, chat, user.id)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
chat_name = tld(chat.id, "note_is_local")
else:
chat_name = chat.title
if len(args) >= 1:
notename = args[0].lower()
if sql.rm_note(chat_id, notename):
update.effective_message.reply_text(tld(
chat.id, "clear_success").format(chat_name),
parse_mode=ParseMode.MARKDOWN)
else:
update.effective_message.reply_text(
tld(chat.id, "note_not_existed"))
def stop_filter(bot: Bot, update: Update):
chat = update.effective_chat
user = update.effective_user
args = update.effective_message.text.split(None, 1)
conn = connected(bot, update, chat, user.id)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
chat_name = tld(chat.id, "cust_filters_local")
else:
chat_name = chat.title
if len(args) < 2:
return
chat_filters = sql.get_chat_triggers(chat_id)
if not chat_filters:
update.effective_message.reply_text(
tld(chat.id, "cust_filters_list_empty").format(chat_name))
return
for keyword in chat_filters:
if keyword == args[1].lower():
sql.remove_filter(chat_id, args[1].lower())
update.effective_message.reply_text(
tld(chat.id, "cust_filters_stop_success").format(chat_name),
parse_mode=telegram.ParseMode.MARKDOWN)
raise DispatcherHandlerStop
update.effective_message.reply_text(
tld(chat.id, "cust_filters_err_wrong_filter"))
def list_notes(bot: Bot, update: Update):
chat = update.effective_chat
user = update.effective_user
conn = connected(bot, update, chat, user.id, need_admin=False)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
msg = tld(chat.id, "note_in_chat")
else:
chat_id = update.effective_chat.id
if chat.type == "private":
chat_name = tld(chat.id, "note_is_local")
msg = tld(chat.id, "note_in_local")
else:
chat_name = chat.title
msg = tld(chat.id, "note_in_chat")
note_list = sql.get_all_chat_notes(chat_id)
for note in note_list:
note_name = " • `#{}`\n".format(note.name.lower())
if len(msg) + len(note_name) > MAX_MESSAGE_LENGTH:
update.effective_message.reply_text(msg,
parse_mode=ParseMode.MARKDOWN)
msg = ""
msg += note_name
if not note_list:
update.effective_message.reply_text(tld(
chat.id, "note_none_in_chat").format(chat_name),
parse_mode=ParseMode.MARKDOWN)
elif len(msg) != 0:
msg += tld(chat.id, "note_get")
update.effective_message.reply_text(msg.format(chat_name),
parse_mode=ParseMode.MARKDOWN)
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 unmute(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, "unmute_invalid"))
return ""
try:
member = chatD.get_member(int(user_id))
except BadRequest as excp:
if excp.message == "User not found.":
message.reply_text(tld(chat.id, "bans_err_usr_not_found"))
return ""
else:
raise
if is_user_ban_protected(chat, user_id, member):
message.reply_text(tld(chat.id, "unmute_is_an_admin"))
return ""
if 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:
message.reply_text(
tld(chat.id, "unmute_not_muted").format(chatD.title))
else:
bot.restrict_chat_member(chatD.id,
int(user_id),
can_send_messages=True,
can_send_media_messages=True,
can_send_other_messages=True,
can_add_web_page_previews=True)
keyboard = []
reply = tld(chat.id, "unmute_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#UNMUTE" \
"\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, "unmute_not_in_chat"))
return ""
def get(bot, update, notename, show_none=True, no_format=False):
chat = update.effective_chat
user = update.effective_user
conn = connected(bot, update, chat, user.id, need_admin=False)
if conn:
chat_id = conn
send_id = user.id
else:
chat_id = update.effective_chat.id
send_id = chat_id
note = sql.get_note(chat_id, notename)
message = update.effective_message
if note:
pass
elif notename[0] == "#":
hashnote = sql.get_note(chat_id, notename[1:])
if hashnote:
note = hashnote
elif show_none:
message.reply_text(tld(chat.id, "note_not_existed"))
return
# If we're replying to a message, reply to that message (unless it's an error)
if message.reply_to_message:
reply_id = message.reply_to_message.message_id
else:
reply_id = message.message_id
if note and note.is_reply:
if MESSAGE_DUMP:
try:
bot.forward_message(chat_id=chat_id,
from_chat_id=MESSAGE_DUMP,
message_id=note.value)
except BadRequest as excp:
if excp.message == "Message to forward not found":
message.reply_text(tld(chat.id, "note_lost"))
sql.rm_note(chat_id, notename)
else:
raise
else:
try:
bot.forward_message(chat_id=chat_id,
from_chat_id=chat_id,
message_id=note.value)
except BadRequest as excp:
if excp.message == "Message to forward not found":
message.reply_text(tld(chat.id, "note_msg_del"))
sql.rm_note(chat_id, notename)
else:
raise
else:
if note:
text = note.value
else:
text = None
keyb = []
parseMode = ParseMode.MARKDOWN
buttons = sql.get_buttons(chat_id, notename)
if no_format:
parseMode = None
text += revert_buttons(buttons)
else:
keyb = build_keyboard(buttons)
keyboard = InlineKeyboardMarkup(keyb)
try:
if note and note.msgtype in (sql.Types.BUTTON_TEXT,
sql.Types.TEXT):
try:
bot.send_message(send_id,
text,
reply_to_message_id=reply_id,
parse_mode=parseMode,
disable_web_page_preview=True,
reply_markup=keyboard)
except BadRequest as excp:
if excp.message == "Wrong http url":
failtext = tld(chat.id, "note_url_invalid")
failtext += "\n\n```\n{}```".format(
note.value + revert_buttons(buttons))
message.reply_text(failtext, parse_mode="markdown")
else:
if note:
ENUM_FUNC_MAP[note.msgtype](send_id,
note.file,
caption=text,
reply_to_message_id=reply_id,
parse_mode=parseMode,
disable_web_page_preview=True,
reply_markup=keyboard)
except BadRequest as excp:
if excp.message == "Entity_mention_user_invalid":
message.reply_text(tld(chat.id, "note_mention_invalid"))
elif FILE_MATCHER.match(note.value):
message.reply_text(tld(chat.id, "note_incorrect_import"))
sql.rm_note(chat_id, notename)
else:
message.reply_text(tld(chat.id, "note_cannot_send"))
LOGGER.exception("Could not parse message #%s in chat %s",
notename, str(chat_id))
LOGGER.warning("Message was: %s", str(note.value))
return
def filters(bot: Bot, update: Update):
chat = update.effective_chat
user = update.effective_user
msg = update.effective_message
args = msg.text.split(
None,
1) # use python's maxsplit to separate Cmd, keyword, and reply_text
conn = connected(bot, update, chat, user.id)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
chat_name = tld(chat.id, "cust_filters_local")
else:
chat_name = chat.title
if len(args) < 2:
return
extracted = split_quotes(args[1])
if len(extracted) < 1:
return
# set trigger -> lower, so as to avoid adding duplicate filters with different cases
keyword = extracted[0].lower()
is_sticker = False
is_document = False
is_image = False
is_voice = False
is_audio = False
is_video = False
buttons = []
# determine what the contents of the filter are - text, image, sticker, etc
if len(extracted) >= 2:
offset = len(extracted[1]) - len(
msg.text) # set correct offset relative to command + notename
content, buttons = button_markdown_parser(
extracted[1], entities=msg.parse_entities(), offset=offset)
content = content.strip()
if not content:
msg.reply_text(tld(chat.id, "cust_filters_err_btn_only"))
return
elif msg.reply_to_message and msg.reply_to_message.sticker:
content = msg.reply_to_message.sticker.file_id
is_sticker = True
elif msg.reply_to_message and msg.reply_to_message.document:
content = msg.reply_to_message.document.file_id
is_document = True
elif msg.reply_to_message and msg.reply_to_message.photo:
content = msg.reply_to_message.photo[
-1].file_id # last elem = best quality
is_image = True
elif msg.reply_to_message and msg.reply_to_message.audio:
content = msg.reply_to_message.audio.file_id
is_audio = True
elif msg.reply_to_message and msg.reply_to_message.voice:
content = msg.reply_to_message.voice.file_id
is_voice = True
elif msg.reply_to_message and msg.reply_to_message.video:
content = msg.reply_to_message.video.file_id
is_video = True
else:
msg.reply_text(tld(chat.id, "cust_filters_err_empty"))
return
# Add the filter
# Note: perhaps handlers can be removed somehow using sql.get_chat_filters
for handler in dispatcher.handlers.get(HANDLER_GROUP, []):
if handler.filters == (keyword, chat_id):
dispatcher.remove_handler(handler, HANDLER_GROUP)
sql.add_filter(chat_id, keyword, content, is_sticker, is_document,
is_image, is_audio, is_voice, is_video, buttons)
msg.reply_text(tld(chat.id,
"cust_filters_add_success").format(keyword, chat_name),
parse_mode=telegram.ParseMode.MARKDOWN)
raise DispatcherHandlerStop
