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 = db.get_chat_log_channel(chat.id)
if log_chat:
try:
send_log(context.bot, log_chat, chat.id, result)
except Unauthorized:
db.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 get_user_id(username):
# ensure valid userid
if len(username) <= 5:
return None
if username.startswith("@"):
username = username[1:]
users = users_db.get_userid_by_name(username)
if not users:
return None
elif len(users) == 1:
return users[0]["_id"]
else:
for user_obj in users:
try:
userdat = dispatcher.bot.get_chat(user_obj["_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 __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 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 = ("Berikut adalah bantuan untuk modul *{}* :\n".format(
HELPABLE[module].__mod_name__) + HELPABLE[module].__help__)
query.message.edit_text(
text=text,
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup([[
InlineKeyboardButton(text="⟸ KEMBALI",
callback_data="help_back")
]]),
)
elif staff_match:
query.message.edit_text(
text=STAFF_HELP_STRINGS,
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup([[
InlineKeyboardButton(text="⟸ KEMBALI",
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("Pengecualian di tombol bantuan. %s",
str(query.data))
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)
else:
LOGGER.info("Using long polling.")
updater.start_polling(timeout=15, read_latency=5, clean=True)
if MESSAGE_DUMP:
updater.bot.send_message(
chat_id=MESSAGE_DUMP, text="System Started..."
)
updater.idle()
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 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))
except error.Unauthorized:
return
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 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 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:
db.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 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 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 broadcast(update, context):
to_send = update.effective_message.text.split(None, 1)
if len(to_send) >= 2:
chats = users_db.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 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.",
)
db.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.",
)
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)
else:
LOGGER.info("Using long polling.")
updater.start_polling(timeout=15, read_latency=5, clean=True)
if MESSAGE_DUMP:
updater.bot.send_message(
chat_id=MESSAGE_DUMP, text="System Started..."
)
updater.idle()
if __name__ == "__main__":
LOGGER.info("Successfully loaded modules: " + str(ALL_MODULES))
main()
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 temp_mute(update, context):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
message = update.effective_message # type: Optional[Message]
args = context.args
if user_can_ban(chat, user, context.bot.id) == False:
message.reply_text(
"You don't have enough rights to restrict someone from talking!")
return ""
user_id, reason = extract_user_and_text(message, args)
if not user_id:
message.reply_text("You don't seem to be referring to a user.")
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_admin(chat, user_id, member):
message.reply_text("I really wish I could mute admins...")
return ""
if user_id == context.bot.id:
message.reply_text("I'm not gonna MUTE myself, are you crazy?")
return ""
if not reason:
message.reply_text(
"You haven't specified a time to mute this user for!")
return ""
split_reason = reason.split(None, 1)
time_val = split_reason[0].lower()
if len(split_reason) > 1:
reason = split_reason[1]
else:
reason = ""
mutetime = extract_time(message, time_val)
if not mutetime:
return ""
log = ("<b>{}:</b>"
"\n#TEMP MUTED"
"\n<b>Admin:</b> {}"
"\n<b>User:</b> {}"
"\n<b>Time:</b> {}".format(
html.escape(chat.title),
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name),
time_val,
))
if reason:
log += "\n<b>Reason:</b> {}".format(reason)
try:
if member.can_send_messages is None or member.can_send_messages:
context.bot.restrict_chat_member(
chat.id,
user_id,
until_date=mutetime,
permissions=ChatPermissions(can_send_messages=False),
)
message.reply_text("shut up! 🤐 Taped for {}!".format(time_val))
return log
else:
message.reply_text("This user is already muted.")
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
message.reply_text("shut up! 🤐 Taped for {}!".format(time_val),
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("Well damn, I can't mute that user.")
return ""
def purge(update, context):
args = context.args
msg = update.effective_message # type: Optional[Message]
if msg.reply_to_message:
user = update.effective_user # type: Optional[User]
chat = update.effective_chat # type: Optional[Chat]
if user_can_delete(chat, user, context.bot.id) == False:
msg.reply_text("You don't have enough rights to delete message!")
return ""
if can_delete(chat, context.bot.id):
message_id = msg.reply_to_message.message_id
delete_to = msg.message_id - 1
if args and args[0].isdigit():
new_del = message_id + int(args[0])
# No point deleting messages which haven't been written yet.
if new_del < delete_to:
delete_to = new_del
for m_id in range(delete_to, message_id - 1,
-1): # Reverse iteration over message ids
try:
context.bot.deleteMessage(chat.id, m_id)
except BadRequest as err:
if err.message == "Message can't be deleted":
context.bot.send_message(
chat.id,
"Cannot delete all messages. The messages may be too old, I might "
"not have delete rights, or this might not be a supergroup."
)
elif err.message != "Message to delete not found":
LOGGER.exception("Error while purging chat messages.")
try:
msg.delete()
except BadRequest as err:
if err.message == "Message can't be deleted":
context.bot.send_message(
chat.id,
"Cannot delete all messages. The messages may be too old, I might "
"not have delete rights, or this might not be a supergroup."
)
elif err.message != "Message to delete not found":
LOGGER.exception("Error while purging chat messages.")
del_msg = context.bot.send_message(chat.id, "Purge complete.")
time.sleep(2)
try:
del_msg.delete()
except BadRequest:
pass
return "<b>{}:</b>" \
"\n#PURGE" \
"\n<b>Admin:</b> {}" \
"\nPurged <code>{}</code> messages.".format(html.escape(chat.title),
mention_html(user.id, user.first_name),
delete_to - message_id)
else:
msg.reply_text(
"Reply to a message to select where to start purging from.")
return ""
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
ALL_MODULES = sorted(__list_all_modules())
LOGGER.info("Modules to load: %s", str(ALL_MODULES))
__all__ = ALL_MODULES + ["ALL_MODULES"]
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("Anda tidak punya hak untuk melakukan itu!")
return ""
user_id, reason = extract_user_and_text(message, args)
if not user_id:
message.reply_text("Heei, setidak nya sebutkan satu untuk diblokir...")
return ""
try:
member = chat.get_member(user_id)
except BadRequest as excp:
if excp.message == "Pengguna tidak ditemukan":
message.reply_text("Saya tidak dapat menemukan pengguna itu")
return ""
else:
raise
if is_user_ban_protected(chat, user_id, member):
message.reply_text("Weew, Anda yakin memblokir seorang admin?...")
return ""
if user_id == context.bot.id:
message.reply_text(
"Saya tidak akan memblokir dri saya sendiri, Anda sepertinya tidak waras!"
)
return ""
if not reason:
message.reply_text(
"Anda belum menentukan waktu untuk memblokir pengguna ini!")
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>Alasan:</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(
"Diblokir! Pengguna telah diblokir selama {}.".format(time_val))
return log
except BadRequest as excp:
if excp.message == "Pesan tidak ditemukan":
# Do not reply
message.reply_text(
"Selamat jalan.. temukan saya lagi setelah {}.".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("Gagal! saya gagal memblokir pengguna ini.")
return ""
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 = "*{}* memiliki pengaturan berikut untuk *{}* modul:\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="Kembali",
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(
"Halo! kamu Ada beberapa setelan untuk {} - lanjutkan dan pilih apa "
"yang Anda inginkan.".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(
"Halo! kamu Ada beberapa setelan untuk {} - lanjutkan dan pilih apa "
"yang Anda inginkan.".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=
"Halo! kamu Ada beberapa setelan untuk {} - lanjutkan dan pilih apa "
"yang Anda inginkan.".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("Pengecualian di tombol pengaturan. %s",
str(query.data))
# Copyright (C) 2020 UserindoBot Team, <https://github.com/userbotindo/UserIndoBot.git>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""MongoDB database."""
from pymongo import MongoClient, collection
from ferbot import MONGO_URI, LOGGER
LOGGER.info("Connecting to MongoDB")
DB_CLIENT = MongoClient(MONGO_URI)
_DB = DB_CLIENT["UbotIndo"]
def get_collection(name: str) -> collection:
"""Get the collection from database."""
return _DB[name]
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)
)
def reply_filter(update, context):
chat = update.effective_chat
message = update.effective_message
to_match = extract_text(message)
if not to_match:
return
chat_filters = sql.get_chat_triggers(chat.id)
for keyword in chat_filters:
pattern = r"( |^|[^\w])" + re.escape(keyword) + r"( |$|[^\w])"
if re.search(pattern, to_match, flags=re.IGNORECASE):
filt = sql.get_filter(chat.id, keyword)
if filt.reply == "there is should be a new reply":
buttons = sql.get_buttons(chat.id, filt.keyword)
keyb = build_keyboard_parser(context.bot, chat.id, buttons)
keyboard = InlineKeyboardMarkup(keyb)
VALID_WELCOME_FORMATTERS = [
"first",
"last",
"fullname",
"username",
"id",
"chatname",
"mention",
]
if filt.reply_text:
valid_format = escape_invalid_curly_brackets(
filt.reply_text, VALID_WELCOME_FORMATTERS)
if valid_format:
filtext = 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:
filtext = ""
else:
filtext = ""
if filt.file_type in (sql.Types.BUTTON_TEXT, sql.Types.TEXT):
try:
context.bot.send_message(
chat.id,
markdown_to_html(filtext),
reply_to_message_id=message.message_id,
parse_mode=ParseMode.HTML,
reply_markup=keyboard,
)
except BadRequest as excp:
error_catch = get_exception(excp, filt, chat)
if error_catch == "noreply":
try:
context.bot.send_message(
chat.id,
markdown_to_html(filtext),
parse_mode=ParseMode.HTML,
reply_markup=keyboard,
)
except BadRequest as excp:
LOGGER.exception("Error in filters: " +
excp.message)
send_message(
update.effective_message,
get_exception(excp, filt, chat),
)
else:
try:
send_message(
update.effective_message,
get_exception(excp, filt, chat),
)
except BadRequest as excp:
LOGGER.exception("Failed to send message: " +
excp.message)
else:
if (ENUM_FUNC_MAP[filt.file_type] ==
dispatcher.bot.send_sticker):
ENUM_FUNC_MAP[filt.file_type](
chat.id,
filt.file_id,
reply_to_message_id=message.message_id,
reply_markup=keyboard,
)
else:
ENUM_FUNC_MAP[filt.file_type](
chat.id,
filt.file_id,
caption=markdown_to_html(filtext),
reply_to_message_id=message.message_id,
parse_mode=ParseMode.HTML,
reply_markup=keyboard,
)
break
else:
if filt.is_sticker:
message.reply_sticker(filt.reply)
elif filt.is_document:
message.reply_document(filt.reply)
elif filt.is_image:
message.reply_photo(filt.reply)
elif filt.is_audio:
message.reply_audio(filt.reply)
elif filt.is_voice:
message.reply_voice(filt.reply)
elif filt.is_video:
message.reply_video(filt.reply)
elif filt.has_markdown:
buttons = sql.get_buttons(chat.id, filt.keyword)
keyb = build_keyboard_parser(context.bot, chat.id, buttons)
keyboard = InlineKeyboardMarkup(keyb)
try:
send_message(
update.effective_message,
filt.reply,
parse_mode=ParseMode.MARKDOWN,
reply_markup=keyboard,
)
except BadRequest as excp:
if excp.message == "Unsupported url protocol":
try:
send_message(
update.effective_message,
"You seem to be trying to use an unsupported url protocol. "
"Telegram doesn't support buttons for some protocols, such as tg://. Please try "
"again...",
)
except BadRequest as excp:
LOGGER.exception("Error in filters: " +
excp.message)
elif excp.message == "Reply message not found":
try:
context.bot.send_message(
chat.id,
filt.reply,
parse_mode=ParseMode.MARKDOWN,
reply_markup=keyboard,
)
except BadRequest as excp:
LOGGER.exception("Error in filters: " +
excp.message)
else:
try:
send_message(
update.effective_message,
"This message couldn't be sent as it's incorrectly formatted.",
)
except BadRequest as excp:
LOGGER.exception("Error in filters: " +
excp.message)
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),
)
else:
# LEGACY - all new filters will have has_markdown set to
# True.
try:
send_message(update.effective_message, filt.reply)
except BadRequest as excp:
LOGGER.exception("Error in filters: " + excp.message)
break
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 ""
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 ""
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import telegram.ext as tg
from telegram import Update
from ferbot import LOGGER
try:
from ubotindo import CUSTOM_CMD
except BaseException:
CUSTOM_CMD = False
if CUSTOM_CMD:
CMD_STARTERS = CUSTOM_CMD
LOGGER.debug("Bot custom command handler = \"%s\"", CMD_STARTERS)
else:
CMD_STARTERS = "/"
class CustomCommandHandler(tg.CommandHandler):
def __init__(self, command, callback, **kwargs):
if "admin_ok" in kwargs:
del kwargs["admin_ok"]
super().__init__(command, callback, **kwargs)
def check_update(self, update):
if isinstance(update, Update) and update.effective_message:
message = update.effective_message
if message.text and len(message.text) > 1:
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)
else:
LOGGER.info("Using long polling.")
updater.start_polling(timeout=15, read_latency=5, clean=True)
if MESSAGE_DUMP:
updater.bot.send_message(chat_id=MESSAGE_DUMP,
text="System berhasil dijalankan...")
updater.idle()
if __name__ == "__main__":
LOGGER.info("Modul berhasil dimuat: " + str(ALL_MODULES))
main()
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("Anda tidak memiliki izin untuk melakukan itu!")
return ""
user_id, reason = extract_user_and_text(message, args)
if not user_id:
message.reply_text("Anda tidak mengarahkan ke pengguna!")
return ""
try:
member = chat.get_member(user_id)
except BadRequest as excp:
if excp.message == "Pengguna tidak ditemukan":
message.reply_text("Saya tidak dapat menemukan pengguna ini")
return ""
else:
raise
if is_user_ban_protected(chat, user_id, member):
message.reply_text(
"Saya tidak akan melakukan itu ke admin, jangan bertingkah sok lucu!"
)
return ""
if user_id == context.bot.id:
message.reply_text(
"Saya tidak akan melakukan itu ke diri saya, Apakah anda gangguan jiwa?"
)
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>Alasan:</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,
"{} Selamat tinggal, lain kali jangan bertingkah konyol!".format(
mention_html(member.user.id, member.user.first_name)),
parse_mode=ParseMode.HTML,
)
return log
except BadRequest as excp:
if excp.message == "Pesan tidak ditemukan":
# 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(
"Huhhh, saya gagal melakukannya kepada pengguna itu.")
return ""
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
