def __list_all_modules():
import glob
from os.path import basename, dirname, isfile
# This generates a list of modules in this folder for the * in __main__ to
# work.
mod_paths = glob.glob(dirname(__file__) + "/*.py")
all_modules = [
basename(f)[:-3] for f in mod_paths
if isfile(f) and f.endswith(".py") and not f.endswith("__init__.py")
]
if LOAD or NO_LOAD:
to_load = LOAD
if to_load:
if not all(
any(mod == module_name for module_name in all_modules)
for mod in to_load):
LOGGER.error("Invalid loadorder names. Quitting.")
sys.exit(1)
else:
to_load = all_modules
if NO_LOAD:
LOGGER.info("Not loading: {}".format(NO_LOAD))
return [item for item in to_load if item not in NO_LOAD]
return to_load
return all_modules
def error_handler(update, context):
"""Log the error and send a telegram message to notify the developer."""
# Log the error before we do anything else, so we can see it even if
# something breaks.
LOGGER.error(msg="Exception while handling an update:",
exc_info=context.error)
# traceback.format_exception returns the usual python message about an exception, but as a
# list of strings rather than a single string, so we have to join them
# together.
tb_list = traceback.format_exception(None, context.error,
context.error.__traceback__)
tb = "".join(tb_list)
# Build the message with some markup and additional information about what
# happened.
message = ("An exception was raised while handling an update\n"
"<pre>update = {}</pre>\n\n"
"<pre>{}</pre>").format(
html.escape(
json.dumps(update.to_dict(),
indent=2,
ensure_ascii=False)),
html.escape(tb),
)
if len(message) >= 4096:
message = message[:4096]
# Finally, send the message
context.bot.send_message(chat_id=OWNER_ID,
text=message,
parse_mode=ParseMode.HTML)
def get_user_id(username):
# ensure valid userid
if len(username) <= 5:
return None
if username.startswith("@"):
username = username[1:]
users = sql.get_userid_by_name(username)
if not users:
return None
elif len(users) == 1:
return users[0].user_id
else:
for user_obj in users:
try:
userdat = dispatcher.bot.get_chat(user_obj.user_id)
if userdat.username == username:
return userdat.id
except BadRequest as excp:
if excp.message == "Chat not found":
pass
else:
LOGGER.exception("Error extracting user ID")
return None
def log_action(update, context, *args, **kwargs):
result = func(update, context, *args, **kwargs)
chat = update.effective_chat
message = update.effective_message
if result:
if chat.type == chat.SUPERGROUP and chat.username:
result += (
"\n<b>Link:</b> "
'<a href="http://telegram.me/{}/{}">click here</a>'.
format(chat.username, message.message_id))
log_chat = sql.get_chat_log_channel(chat.id)
if log_chat:
try:
send_log(context.bot, log_chat, chat.id, result)
except Unauthorized:
sql.stop_chat_logging(chat.id)
elif result == "":
pass
else:
LOGGER.warning(
"%s was set as loggable, but had no return statement.",
func)
return result
def main():
# test_handler = CommandHandler("test", test)
start_handler = CommandHandler("start",
start,
pass_args=True,
run_async=True)
help_handler = CommandHandler("help", get_help)
help_callback_handler = CallbackQueryHandler(help_button,
pattern=r"help_",
run_async=True)
help_staff_handler = CommandHandler(
"staffhelp",
staff_help,
filters=CustomFilters.support_filter,
run_async=True,
)
settings_handler = CommandHandler("settings", get_settings, run_async=True)
settings_callback_handler = CallbackQueryHandler(settings_button,
pattern=r"stngs_",
run_async=True)
migrate_handler = MessageHandler(Filters.status_update.migrate,
migrate_chats)
is_chat_allowed_handler = MessageHandler(Filters.chat_type.groups,
is_chat_allowed)
# dispatcher.add_handler(test_handler)
dispatcher.add_handler(start_handler)
dispatcher.add_handler(help_handler)
dispatcher.add_handler(settings_handler)
dispatcher.add_handler(help_callback_handler)
dispatcher.add_handler(help_staff_handler)
dispatcher.add_handler(settings_callback_handler)
dispatcher.add_handler(migrate_handler)
dispatcher.add_handler(is_chat_allowed_handler)
dispatcher.add_error_handler(error_handler)
if WEBHOOK:
LOGGER.info("Using webhooks.")
updater.start_webhook(listen="0.0.0.0", port=PORT, url_path=TOKEN)
if CERT_PATH:
updater.bot.set_webhook(url=URL + TOKEN,
certificate=open(CERT_PATH, "rb"))
else:
updater.bot.set_webhook(url=URL + TOKEN)
client.run_until_disconnected()
else:
LOGGER.info("Using long polling.")
updater.start_polling(timeout=15, read_latency=5)
if MESSAGE_DUMP:
updater.bot.send_message(chat_id=MESSAGE_DUMP,
text="System Started...")
client.run_until_disconnected()
updater.idle()
def help_button(update, context):
query = update.callback_query
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)
text = ("Here is the help for the *{}* module:\n".format(
HELPABLE[module].__mod_name__) + HELPABLE[module].__help__)
query.message.edit_text(
text=text,
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup([[
InlineKeyboardButton(text="🔙 Back",
callback_data="help_back")
]]),
)
elif prev_match:
curr_page = int(prev_match.group(1))
query.message.edit_text(
HELP_STRINGS,
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup(
paginate_modules(curr_page - 1, HELPABLE, "help")),
)
elif next_match:
next_page = int(next_match.group(1))
query.message.edit_text(
HELP_STRINGS,
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup(
paginate_modules(next_page + 1, HELPABLE, "help")),
)
elif back_match:
query.message.edit_text(
text=HELP_STRINGS,
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup(
paginate_modules(0, HELPABLE, "help")),
)
# ensure no spinny white circle
# query.message.delete()
context.bot.answer_callback_query(query.id)
except Exception 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:
query.message.edit_text(excp.message)
LOGGER.exception("Exception in help buttons. %s", str(query.data))
def help_button(update, context):
query = update.callback_query
user = update.effective_user
mod_match = re.match(r"help_module\((.+?)\)", query.data)
staff_match = re.match(r"help_staff", query.data)
back_match = re.match(r"help_back", query.data)
try:
if mod_match:
module = mod_match.group(1)
text = ("Here is the help for the *{}* module:\n".format(
HELPABLE[module].__mod_name__) + HELPABLE[module].__help__)
query.message.edit_text(
text=text,
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup([[
InlineKeyboardButton(text="⬅ï¸Go Back⬅ï¸",
callback_data="help_back")
]]),
)
elif staff_match:
query.message.edit_text(
text=STAFF_HELP_STRINGS,
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup([[
InlineKeyboardButton(text="â¬…ï¸ Back",
callback_data="help_back")
]]),
)
elif back_match:
keyb = paginate_modules(0, HELPABLE, "help")
# Add aditional button if staff user detected
if (user.id in DEV_USERS or user.id in SUDO_USERS
or user.id in SUPPORT_USERS):
keyb += [[
InlineKeyboardButton(text="Staff",
callback_data="help_staff")
]]
query.message.edit_text(
text=HELP_STRINGS,
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup(keyb),
)
# ensure no spinny white circle
context.bot.answer_callback_query(query.id)
except Exception 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:
query.message.edit_text(excp.message)
LOGGER.exception("Exception in help buttons. %s", str(query.data))
def command_func(update, context, *args, **kwargs):
try:
context.bot.send_chat_action(chat_id=update.effective_chat.id,
action=ChatAction.TYPING)
return func(update, context, *args, **kwargs)
except error.BadRequest as err:
if str(err) == "Have no rights to send a message":
LOGGER.warning("Bot muted in {} {}".format(
update.effective_message.chat.title,
update.effective_message.chat.id))
def get_exception(excp, filt, chat):
if excp.message == "Unsupported url protocol":
return "You seem to be trying to use the URL protocol which is not supported. Telegram does not support key for multiple protocols, such as tg: //. Please try again!"
elif excp.message == "Reply message not found":
return "noreply"
else:
LOGGER.warning("Message %s could not be parsed", str(filt.reply))
LOGGER.exception("Could not parse filter %s in chat %s",
str(filt.keyword), str(chat.id))
return "This data could not be sent because it is incorrectly formatted."
def main():
# test_handler = CommandHandler("test", test)
start_handler = CommandHandler("start", start, pass_args=True)
help_handler = CommandHandler("help", get_help)
help_callback_handler = CallbackQueryHandler(help_button, pattern=r"help_")
start_callback_handler = CallbackQueryHandler(
send_start, pattern=r"bot_start")
dispatcher.add_handler(start_callback_handler)
startstop_callback_handler = CallbackQueryHandler(
start_stop, pattern=r"close_menu")
dispatcher.add_handler(startstop_callback_handler)
settings_handler = CommandHandler("settings", get_settings)
settings_callback_handler = CallbackQueryHandler(
settings_button, pattern=r"stngs_")
migrate_handler = MessageHandler(
Filters.status_update.migrate, migrate_chats)
is_chat_allowed_handler = MessageHandler(Filters.group, is_chat_allowed)
# dispatcher.add_handler(test_handler)
dispatcher.add_handler(start_handler)
dispatcher.add_handler(help_handler)
dispatcher.add_handler(settings_handler)
dispatcher.add_handler(help_callback_handler)
# dispatcher.add_handler(help_staff_handler)
dispatcher.add_handler(settings_callback_handler)
dispatcher.add_handler(migrate_handler)
dispatcher.add_handler(is_chat_allowed_handler)
dispatcher.add_error_handler(error_handler)
if WEBHOOK:
LOGGER.info("Using webhooks.")
updater.start_webhook(listen="0.0.0.0", port=PORT, url_path=TOKEN)
if CERT_PATH:
updater.bot.set_webhook(
url=URL + TOKEN,
certificate=open(
CERT_PATH,
"rb"))
else:
updater.bot.set_webhook(url=URL + TOKEN)
client.run_until_disconnected()
else:
LOGGER.info("Using long polling.")
updater.start_polling(timeout=15, read_latency=4)
updater.bot.send_message(
chat_id=MESSAGE_DUMP,
text="Who do you think you are, Forrest Gump?")
client.run_until_disconnected()
updater.idle()
def send(msg, bot, update):
if len(str(msg)) > 2000:
with io.BytesIO(str.encode(msg)) as out_file:
out_file.name = "output.txt"
bot.send_document(chat_id=update.effective_chat.id,
document=out_file)
else:
LOGGER.info(f"OUT: '{msg}'")
bot.send_message(chat_id=update.effective_chat.id,
text=f"`{msg}`",
parse_mode=ParseMode.MARKDOWN)
def check_cas(user_id):
cas_url = "https://api.cas.chat/check?user_id={}".format(user_id)
try:
r = get(cas_url, timeout=3)
data = r.json()
except BaseException:
LOGGER.info(f"CAS check failed for {user_id}")
return False
if data and data["ok"]:
return "https://cas.chat/query?u={}".format(user_id)
else:
return False
def tld_list(chat_id, t):
LANGUAGE = prev_locale(chat_id)
if LANGUAGE:
LOCALE = LANGUAGE.locale_name
if LOCALE in ('en-US') and t in strings['en-US']:
return strings['en-US'][t]
if t in strings['en-US']:
return strings['en-US'][t]
LOGGER.warning(f"#NOSTR No string found for {t}.")
return f"No string found for {t}.\nReport it in @HarukaAyaGroup."
def sed(update, context):
sed_result = separate_sed(update.effective_message.text)
if sed_result and update.effective_message.reply_to_message:
if update.effective_message.reply_to_message.text:
to_fix = update.effective_message.reply_to_message.text
elif update.effective_message.reply_to_message.caption:
to_fix = update.effective_message.reply_to_message.caption
else:
return
repl, repl_with, flags = sed_result
if not repl:
update.effective_message.reply_to_message.reply_text(
"You're trying to replace... "
"nothing with something?")
return
try:
# Protects bot from retarded geys -_-
if infinite_checker(repl):
return update.effective_message.reply_text("Nice try -_-")
if "i" in flags and "g" in flags:
text = re.sub(repl, repl_with, to_fix, flags=re.I).strip()
elif "i" in flags:
text = re.sub(repl, repl_with, to_fix, count=1,
flags=re.I).strip()
elif "g" in flags:
text = re.sub(repl, repl_with, to_fix).strip()
else:
text = re.sub(repl, repl_with, to_fix, count=1).strip()
except sre_constants.error:
LOGGER.warning(update.effective_message.text)
LOGGER.exception("SRE constant error")
update.effective_message.reply_text(
"Do you even sed? Apparently not.")
return
# empty string errors -_-
if len(text) >= telegram.MAX_MESSAGE_LENGTH:
update.effective_message.reply_text(
"The result of the sed command was too long for \
telegram!")
elif text:
update.effective_message.reply_to_message.reply_text(text)
def migrate_chats(update, context):
msg = update.effective_message # type: Optional[Message]
if msg.migrate_to_chat_id:
old_chat = update.effective_chat.id
new_chat = msg.migrate_to_chat_id
elif msg.migrate_from_chat_id:
old_chat = msg.migrate_from_chat_id
new_chat = update.effective_chat.id
else:
return
LOGGER.info("Migrating from %s, to %s", str(old_chat), str(new_chat))
for mod in MIGRATEABLE:
mod.__migrate__(old_chat, new_chat)
LOGGER.info("Successfully migrated!")
raise DispatcherHandlerStop
def sban(update, context):
bot = context.bot
args = context.args
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
message = update.effective_message # type: Optional[Message]
update.effective_message.delete()
user_id, reason = extract_user_and_text(message, args)
if not user_id:
return ""
try:
member = chat.get_member(user_id)
except BadRequest as excp:
if excp.message == "User not found":
return ""
else:
raise
if is_user_ban_protected(chat, user_id, member):
return ""
if user_id == bot.id:
return ""
log = tld(chat.id, "bans_sban_logger").format(
html.escape(chat.title), mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name), user_id)
if reason:
log += tld(chat.id, "bans_logger_reason").format(reason)
try:
chat.kick_member(user_id)
return log
except BadRequest as excp:
if excp.message == "Reply message not found":
return log
else:
LOGGER.warning(update)
LOGGER.exception("ERROR banning user %s in chat %s (%s) due to %s",
user_id, chat.title, chat.id, excp.message)
return ""
def error_handler(update, context):
"""Log the error and send a telegram message to notify the developer."""
# Log the error before we do anything else, so we can see it even if something breaks.
LOGGER.error(msg="Exception while handling an update:",
exc_info=context.error)
if isinstance(context.error, SQLAlchemyError) or isinstance(
context.error, DBAPIError):
return
# traceback.format_exception returns the usual python message about an exception, but as a
# list of strings rather than a single string, so we have to join them together.
else:
tb_list = traceback.format_exception(None, context.error,
context.error.__traceback__)
tb_string = "".join(tb_list)
# Build the message with some markup and additional information about what happened.
# You might need to add some logic to deal with messages longer than the 4096 character limit.
message = (
f"An exception was raised while handling an update\n"
f"update = {(json.dumps(update.to_dict(), indent=2, ensure_ascii=False))}"
"\n\n"
f"context.chat_data = {(str(context.chat_data))}\n\n"
f"context.user_data = {(str(context.user_data))}\n\n"
f"{(tb_string)}")
key = (requests.post("https://nekobin.com/api/documents",
json={
"content": message
}).json().get("result").get("key"))
markup = InlineKeyboardMarkup([[
InlineKeyboardButton("Nekobin Url",
url=f"https://nekobin.com/{key}"),
InlineKeyboardButton("Nekobin Raw",
url=f"https://nekobin.com/raw/{key}"),
]])
# Finally, send the message
context.bot.send_message(
chat_id=MESSAGE_DUMP,
text="an error has been found here !!!",
reply_markup=markup,
)
def setlog(update, context):
message = update.effective_message
chat = update.effective_chat
if chat.type == chat.CHANNEL:
message.reply_text(
"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:
context.bot.send_message(
message.forward_from_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":
context.bot.send_message(
chat.id, "Successfully set log channel!")
else:
LOGGER.exception("ERROR in setting the log channel.")
context.bot.send_message(chat.id, "Successfully set log channel!")
else:
message.reply_text(
"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"
)
def broadcast(update, context):
to_send = update.effective_message.text.split(None, 1)
if len(to_send) >= 2:
chats = sql.get_all_chats() or []
failed = 0
for chat in chats:
try:
context.bot.sendMessage(int(chat.chat_id), to_send[1])
sleep(0.1)
except TelegramError:
failed += 1
LOGGER.warning(
"Couldn't send broadcast to %s, group name %s",
str(chat.chat_id),
str(chat.chat_name),
)
update.effective_message.reply_text(
"Broadcast complete. {} groups failed to receive the message, probably "
"due to being kicked.".format(failed))
def tld(chat_id, t, show_none=True):
LANGUAGE = prev_locale(chat_id)
if LANGUAGE:
LOCALE = LANGUAGE.locale_name
if LOCALE in ('en-US') and t in strings['en-US']:
result = decode(
encode(strings['en-US'][t], 'latin-1', 'backslashreplace'),
'unicode-escape')
return result
if t in strings['en-US']:
result = decode(
encode(strings['en-US'][t], 'latin-1', 'backslashreplace'),
'unicode-escape')
return result
err = f"No string found for {t}.\nReport it in @HarukaAyaGroup."
LOGGER.warning(err)
return err
def snipe(update, context):
args = context.args
try:
chat_id = str(args[0])
del args[0]
except (TypeError, IndexError):
update.effective_message.reply_text(
"Please give me a chat to echo to!")
to_send = " ".join(args)
if len(to_send) >= 2:
try:
context.bot.sendMessage(int(chat_id), str(to_send))
except TelegramError:
LOGGER.warning("Couldn't send to group %s", str(chat_id))
update.effective_message.reply_text(
"Couldn't send the message. Perhaps I'm not part of that group?"
)
else:
update.effective_message.reply_text(
"Where should i send??\nGive me the chat id!")
def send_log(bot: Bot, log_chat_id: str, orig_chat_id: str, result: str):
try:
bot.send_message(log_chat_id, result, parse_mode=ParseMode.HTML)
except BadRequest as excp:
if excp.message == "Chat not found":
bot.send_message(
orig_chat_id,
"This log channel has been deleted - unsetting.")
sql.stop_chat_logging(orig_chat_id)
else:
LOGGER.warning(excp.message)
LOGGER.warning(result)
LOGGER.exception("Could not parse")
bot.send_message(
log_chat_id,
result +
"\n\nFormatting has been disabled due to an unexpected error.",
)
def settings_button(update, context):
query = update.callback_query
user = update.effective_user
mod_match = re.match(r"stngs_module\((.+?),(.+?)\)", query.data)
prev_match = re.match(r"stngs_prev\((.+?),(.+?)\)", query.data)
next_match = re.match(r"stngs_next\((.+?),(.+?)\)", query.data)
back_match = re.match(r"stngs_back\((.+?)\)", query.data)
try:
if mod_match:
chat_id = mod_match.group(1)
module = mod_match.group(2)
chat = context.bot.get_chat(chat_id)
text = "*{}* has the following settings for the *{}* module:\n\n".format(
escape_markdown(chat.title), CHAT_SETTINGS[module].__mod_name__
) + CHAT_SETTINGS[module].__chat_settings__(chat_id, user.id)
query.message.reply_text(
text=text,
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup([[
InlineKeyboardButton(
text="Back",
callback_data="stngs_back({})".format(chat_id),
)
]]),
)
elif prev_match:
chat_id = prev_match.group(1)
curr_page = int(prev_match.group(2))
chat = context.bot.get_chat(chat_id)
query.message.reply_text(
"Hi there! There are quite a few settings for {} - go ahead and pick what "
"you're interested in.".format(chat.title),
reply_markup=InlineKeyboardMarkup(
paginate_modules(curr_page - 1,
CHAT_SETTINGS,
"stngs",
chat=chat_id)),
)
elif next_match:
chat_id = next_match.group(1)
next_page = int(next_match.group(2))
chat = context.bot.get_chat(chat_id)
query.message.reply_text(
"Hi there! There are quite a few settings for {} - go ahead and pick what "
"you're interested in.".format(chat.title),
reply_markup=InlineKeyboardMarkup(
paginate_modules(next_page + 1,
CHAT_SETTINGS,
"stngs",
chat=chat_id)),
)
elif back_match:
chat_id = back_match.group(1)
chat = context.bot.get_chat(chat_id)
query.message.reply_text(
text=
"Hi there! There are quite a few settings for {} - go ahead and pick what "
"you're interested in.".format(escape_markdown(chat.title)),
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup(
paginate_modules(0, CHAT_SETTINGS, "stngs", chat=chat_id)),
)
# ensure no spinny white circle
query.message.delete()
context.bot.answer_callback_query(query.id)
except Exception 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:
query.message.edit_text(excp.message)
LOGGER.exception("Exception in settings buttons. %s",
str(query.data))
dispatcher.add_handler(settings_callback_handler)
dispatcher.add_handler(migrate_handler)
dispatcher.add_handler(is_chat_allowed_handler)
dispatcher.add_error_handler(error_handler)
if WEBHOOK:
LOGGER.info("Using webhooks.")
updater.start_webhook(listen="0.0.0.0", port=PORT, url_path=TOKEN)
if CERT_PATH:
updater.bot.set_webhook(url=URL + TOKEN,
certificate=open(CERT_PATH, "rb"))
else:
updater.bot.set_webhook(url=URL + TOKEN)
client.run_until_disconnected()
else:
LOGGER.info("Using long polling.")
updater.start_polling(timeout=15, read_latency=4)
updater.bot.send_message(chat_id=MESSAGE_DUMP,
text="Elizabeth Started...")
client.run_until_disconnected()
updater.idle()
if __name__ == "__main__":
LOGGER.info("Successfully loaded modules: " + str(ALL_MODULES))
client.start(bot_token=TOKEN)
main()
def get(bot, update, notename, show_none=True, no_format=False):
chat_id = update.effective_chat.id
chat = update.effective_chat
user = update.effective_user
conn = connected(bot, update, chat, user.id, need_admin=False)
if conn:
chat_id = conn
user.id
else:
chat_id = update.effective_chat.id
note = sql.get_note(chat_id, notename)
message = update.effective_message
if note:
# 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.is_reply:
if MESSAGE_DUMP:
try:
bot.forward_message(
chat_id=update.effective_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(
"This message seems to have been lost - I'll remove it "
"from your notes list.")
sql.rm_note(chat_id, notename)
else:
raise
else:
try:
bot.forward_message(
chat_id=update.effective_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(
"Looks like the original sender of this note has deleted "
"their message - sorry! Get your bot admin to start using a "
"message dump to avoid this. I'll remove this note from "
"your saved notes.")
sql.rm_note(chat_id, notename)
else:
raise
else:
VALID_NOTE_FORMATTERS = [
"first",
"last",
"fullname",
"username",
"id",
"chatname",
"mention",
]
valid_format = escape_invalid_curly_brackets(
note.value, VALID_NOTE_FORMATTERS)
if valid_format:
text = valid_format.format(
first=escape(message.from_user.first_name),
last=escape(message.from_user.last_name
or message.from_user.first_name),
fullname=" ".join([
escape(message.from_user.first_name),
escape(message.from_user.last_name),
] if message.from_user.last_name else
[escape(message.from_user.first_name)]),
username="******" + escape(message.from_user.username)
if message.from_user.username else mention_html(
message.from_user.id, message.from_user.first_name),
mention=mention_html(message.from_user.id,
message.from_user.first_name),
chatname=escape(message.chat.title)
if message.chat.type != "private" else escape(
message.from_user.first_name),
id=message.from_user.id,
)
else:
text = ""
keyb = []
parseMode = ParseMode.HTML
buttons = sql.get_buttons(chat_id, notename)
if no_format:
parseMode = None
text += revert_buttons(buttons)
else:
text = markdown_to_html(text)
keyb = build_keyboard(buttons)
keyboard = InlineKeyboardMarkup(keyb)
try:
if note.msgtype in (sql.Types.BUTTON_TEXT, sql.Types.TEXT):
bot.send_message(
update.effective_chat.id,
text,
reply_to_message_id=reply_id,
parse_mode=parseMode,
disable_web_page_preview=True,
reply_markup=keyboard,
)
else:
ENUM_FUNC_MAP[note.msgtype](
update.effective_chat.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(
"Looks like you tried to mention someone I've never seen before. If you really "
"want to mention them, forward one of their messages to me, and I'll be able "
"to tag them!")
elif FILE_MATCHER.match(note.value):
message.reply_text(
"This note was an incorrectly imported file from another bot - I can't use "
"it. If you really need it, you'll have to save it again. In "
"the meantime, I'll remove it from your notes list.")
sql.rm_note(chat_id, notename)
else:
message.reply_text(
"This note could not be sent, as it is incorrectly formatted."
)
LOGGER.exception("Could not parse message #%s in chat %s",
notename, str(chat_id))
LOGGER.warning("Message was: %s", str(note.value))
return
elif show_none:
message.reply_text("This note doesn't exist")
def del_lockables(update, context):
chat = update.effective_chat
message = update.effective_message
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(u"{}".format(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(u"{}".format(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
def send(update, message, keyboard, backup_message):
chat = update.effective_chat
cleanserv = sql.clean_service(chat.id)
reply = update.message.message_id
# Clean service welcome
if cleanserv:
try:
dispatcher.bot.delete_message(chat.id, update.message.message_id)
except BadRequest:
pass
reply = False
try:
msg = update.effective_message.reply_text(
message,
parse_mode=ParseMode.MARKDOWN,
reply_markup=keyboard,
reply_to_message_id=reply,
)
except BadRequest as excp:
if excp.message == "Button_url_invalid":
msg = update.effective_message.reply_text(
markdown_parser(
backup_message +
"\nNote: the current message has an invalid url "
"in one of its buttons. Please update."),
parse_mode=ParseMode.MARKDOWN,
reply_to_message_id=reply,
)
elif excp.message == "Unsupported url protocol":
msg = update.effective_message.reply_text(
markdown_parser(
backup_message +
"\nNote: the current message has buttons which "
"use url protocols that are unsupported by "
"telegram. Please update."),
parse_mode=ParseMode.MARKDOWN,
reply_to_message_id=reply,
)
elif excp.message == "Wrong url host":
msg = update.effective_message.reply_text(
markdown_parser(
backup_message +
"\nNote: the current message has some bad urls. "
"Please update."),
parse_mode=ParseMode.MARKDOWN,
reply_to_message_id=reply,
)
LOGGER.warning(message)
LOGGER.warning(keyboard)
LOGGER.exception("Could not parse! got invalid url host errors")
elif excp.message == "Have no rights to send a message":
return
else:
msg = update.effective_message.reply_text(
markdown_parser(backup_message +
"\nNote: An error occured when sending the "
"custom message. Please update."),
parse_mode=ParseMode.MARKDOWN,
reply_to_message_id=reply,
)
LOGGER.exception()
return msg
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
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:
message.delete()
elif getmode == 2:
message.delete()
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.kick_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.kick_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":
pass
else:
LOGGER.exception("Error while deleting blacklist message.")
break
def ban(update, context):
chat = update.effective_chat
user = update.effective_user
message = update.effective_message
args = context.args
if user_can_ban(chat, user, context.bot.id) is False:
message.reply_text("You don't have enough rights to ban users!")
return ""
user_id, reason = extract_user_and_text(message, args)
if not user_id:
message.reply_text("Dude atleast refer some user to ban!")
return ""
try:
member = chat.get_member(user_id)
except BadRequest as excp:
if excp.message == "User not found":
message.reply_text("I can't seem to find this user")
return ""
else:
raise
if is_user_ban_protected(chat, user_id, member):
message.reply_text(
"I'm not gonna ban an admin, don't make fun of yourself!")
return ""
if user_id == context.bot.id:
message.reply_text("I'm not gonna BAN myself, are you crazy or wot?")
return ""
log = ("<b>{}:</b>"
"\n#BANNED"
"\n<b>Admin:</b> {}"
"\n<b>User:</b> {} (<code>{}</code>)".format(
html.escape(chat.title),
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name),
member.user.id,
))
if reason:
log += "\n<b>Reason:</b> {}".format(reason)
try:
chat.kick_member(user_id)
# context.bot.send_sticker(chat.id, BAN_STICKER) # banhammer marie
# sticker
context.bot.sendMessage(
chat.id,
"let {} walk the plank.".format(
mention_html(member.user.id, member.user.first_name)),
parse_mode=ParseMode.HTML,
)
return log
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
message.reply_text("Banned!", quote=False)
return log
else:
LOGGER.warning(update)
LOGGER.exception(
"ERROR banning user %s in chat %s (%s) due to %s",
user_id,
chat.title,
chat.id,
excp.message,
)
message.reply_text("Well damn, I can't ban that user.")
return ""
def temp_ban(update, context):
chat = update.effective_chat
user = update.effective_user
message = update.effective_message
args = context.args
if user_can_ban(chat, user, context.bot.id) is False:
message.reply_text(
"You don't have enough rights to temporarily ban someone!")
return ""
user_id, reason = extract_user_and_text(message, args)
if not user_id:
message.reply_text("Dude! atleast refer some user to ban...")
return ""
try:
member = chat.get_member(user_id)
except BadRequest as excp:
if excp.message == "User not found":
message.reply_text("I can't seem to find this user")
return ""
else:
raise
if is_user_ban_protected(chat, user_id, member):
message.reply_text("Wow! let's start banning Admins themselves?...")
return ""
if user_id == context.bot.id:
message.reply_text("I'm not gonna BAN myself, are you crazy or wot?")
return ""
if not reason:
message.reply_text(
"You haven't specified a time to ban this user for!")
return ""
split_reason = reason.split(None, 1)
time_val = split_reason[0].lower()
if len(split_reason) > 1:
reason = split_reason[1]
else:
reason = ""
bantime = extract_time(message, time_val)
if not bantime:
return ""
log = ("<b>{}:</b>"
"\n#TEMP BANNED"
"\n<b>Admin:</b> {}"
"\n<b>User:</b> {} (<code>{}</code>)"
"\n<b>Time:</b> {}".format(
html.escape(chat.title),
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name),
member.user.id,
time_val,
))
if reason:
log += "\n<b>Reason:</b> {}".format(reason)
try:
chat.kick_member(user_id, until_date=bantime)
# context.bot.send_sticker(chat.id, BAN_STICKER) # banhammer marie
# sticker
message.reply_text(
"Banned! User will be banned for {}.".format(time_val))
return log
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
message.reply_text(
"Goodbye.. we'll meet after {}.".format(time_val), quote=False)
return log
else:
LOGGER.warning(update)
LOGGER.exception(
"ERROR banning user %s in chat %s (%s) due to %s",
user_id,
chat.title,
chat.id,
excp.message,
)
message.reply_text("Well damn, I can't ban that user.")
return ""
