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 __list_all_modules():
from os.path import dirname, basename, isfile
import glob
# 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.")
quit(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 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_")
settings_handler = CommandHandler("settings", get_settings)
settings_callback_handler = CallbackQueryHandler(settings_button, pattern=r"stngs_")
M_CONNECT_BTN_HANDLER = CallbackQueryHandler(m_connect_button, pattern=r"main_connect")
# 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(settings_callback_handler)
dispatcher.add_handler(M_CONNECT_BTN_HANDLER)
# dispatcher.add_error_handler(error_callback)
if WEBHOOK:
LOGGER.info("Using webhooks.")
updater.start_webhook(listen="127.0.0.1",
port=PORT,
url_path=TOKEN)
if CERT_PATH:
updater.bot.set_webhook(url=URL + TOKEN,
certificate=open(CERT_PATH, 'rb'))
def del_lockables(bot: Bot, update: Update):
chat = update.effective_chat
message = update.effective_message
for lockable, filter in LOCK_TYPES.items():
if filter(message) and sql.is_locked(chat.id, lockable) and can_delete(
chat, bot.id):
if lockable == "bots":
new_members = update.effective_message.new_chat_members
for new_mem in new_members:
if new_mem.is_bot:
if not is_bot_admin(chat, bot.id):
message.reply_text(
tld(chat.id, "locks_lock_bots_no_admin"))
return
chat.kick_member(new_mem.id)
message.reply_text(tld(chat.id,
"locks_lock_bots_kick"))
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 ban(bot: Bot, update: Update, args: List[str]) -> str:
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
message = update.effective_message # type: Optional[Message]
user_id, reason = extract_user_and_text(message, args)
if not user_id:
message.reply_text(tld(chat.id, "common_err_no_user"))
return ""
try:
member = chat.get_member(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 user_id == bot.id:
message.reply_text(tld(chat.id, "bans_err_usr_is_bot"))
return ""
if is_user_ban_protected(chat, user_id, member):
message.reply_text(tld(chat.id, "bans_err_usr_is_admin"))
return ""
log = tld(chat.id, "bans_logger").format(
html.escape(chat.title), mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name), user_id)
reply = tld(chat.id, "bans_banned_success").format(
mention_html(user.id, user.first_name),
mention_html(member.user.id, member.user.first_name),
html.escape(chat.title))
if reason:
log += tld(chat.id, "bans_logger_reason").format(reason)
reply += tld(chat.id, "bans_logger_reason").format(reason)
try:
chat.kick_member(user_id)
message.reply_text(reply, parse_mode=ParseMode.HTML)
return log
except BadRequest as excp:
if excp.message == "Reply message not found":
# Do not reply
message.reply_text(reply, quote=False, parse_mode=ParseMode.HTML)
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(
tld(chat.id, "bans_err_unknown").format("banning"))
return ""
def fed_chats(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
user = update.effective_user
if chat.type == "private":
send_message(
update.effective_message,
"This command is specific to the group, not to our pm!",
)
return
fed_id = sql.get_fed_id(chat.id)
info = sql.get_fed_info(fed_id)
if not fed_id:
update.effective_message.reply_text(
"This group is not a part of any federation!")
return
if is_user_fed_admin(fed_id, user.id) is False:
update.effective_message.reply_text(
"Only federation admins can do this!")
return
getlist = sql.all_fed_chats(fed_id)
if len(getlist) == 0:
update.effective_message.reply_text(
"No users are fbanned from the federation {}".format(
info["fname"]),
parse_mode=ParseMode.HTML,
)
return
text = "<b>New chat joined the federation {}:</b>\n".format(info["fname"])
for chats in getlist:
try:
chat_name = dispatcher.bot.getChat(chats).title
except Unauthorized:
sql.chat_leave_fed(chats)
LOGGER.info("Chat {} has leave fed {} because I was kicked".format(
chats, info["fname"]))
continue
text += " • {} (<code>{}</code>)\n".format(chat_name, chats)
try:
update.effective_message.reply_text(text, parse_mode=ParseMode.HTML)
except:
cleanr = re.compile("<.*?>")
cleantext = re.sub(cleanr, "", text)
with BytesIO(str.encode(cleantext)) as output:
output.name = "fedchats.txt"
update.effective_message.reply_document(
document=output,
filename="fedchats.txt",
caption=
"Here is a list of all the chats that joined the federation {}."
.format(info["fname"]),
)
def extract_user_and_text(message: Message, args: List[str]) -> (Optional[int], Optional[str]):
prev_message = message.reply_to_message
split_text = message.text.split(None, 1)
if len(split_text) < 2:
return id_from_reply(message) # only option possible
text_to_parse = split_text[1]
text = ""
entities = list(message.parse_entities([MessageEntity.TEXT_MENTION]))
if len(entities) > 0:
ent = entities[0]
else:
ent = None
# if entity offset matches (command end/text start) then all good
if entities and ent and ent.offset == len(message.text) - len(text_to_parse):
ent = entities[0]
user_id = ent.user.id
text = message.text[ent.offset + ent.length:]
elif len(args) >= 1 and args[0][0] == '@':
user = args[0]
user_id = get_user_id(user)
if not user_id:
return "error", "I don't have that user in my db. You'll be able to interact with them if you reply to that person's message instead, or forward one of that user's messages."
else:
user_id = user_id
res = message.text.split(None, 2)
if len(res) >= 3:
text = res[2]
elif len(args) >= 1 and args[0].isdigit():
user_id = int(args[0])
res = message.text.split(None, 2)
if len(res) >= 3:
text = res[2]
elif prev_message:
user_id, text = id_from_reply(message)
else:
return None, None
try:
message.bot.get_chat(user_id)
except BadRequest as excp:
if excp.message in ("User_id_invalid", "Chat not found"):
return "error", "I don't seem to have interacted with this user before - please forward a message from them to give me control! (like a voodoo doll, I need a piece of them to be able to execute certain commands...)"
else:
LOGGER.exception("Exception %s on user %s", excp.message, user_id)
return None, None
return user_id, text
def del_blackliststicker(update, context):
chat = update.effective_chat # type: Optional[Chat]
message = update.effective_message # type: Optional[Message]
user = update.effective_user
to_match = message.sticker
if not to_match:
return
getmode, value = sql.get_blacklist_setting(chat.id)
chat_filters = sql.get_chat_stickers(chat.id)
for trigger in chat_filters:
if to_match.set_name.lower() == trigger.lower():
try:
if getmode == 0:
return
elif getmode == 1:
message.delete()
elif getmode == 2:
message.delete()
warn(update.effective_user, chat, "Using sticker '{}' which in blacklist stickers".format(trigger), message, update.effective_user, conn=False)
return
elif getmode == 3:
message.delete()
bot.restrict_chat_member(chat.id, update.effective_user.id, can_send_messages=False)
bot.sendMessage(chat.id, "{} muted because using '{}' which in blacklist stickers".format(mention_markdown(user.id, user.first_name), trigger), parse_mode="markdown")
return
elif getmode == 4:
message.delete()
res = chat.unban_member(update.effective_user.id)
if res:
bot.sendMessage(chat.id, "{} kicked because using '{}' which in blacklist stickers".format(mention_markdown(user.id, user.first_name), trigger), parse_mode="markdown")
return
elif getmode == 5:
message.delete()
chat.kick_member(user.id)
bot.sendMessage(chat.id, "{} banned because using '{}' which in blacklist stickers".format(mention_markdown(user.id, user.first_name), trigger), parse_mode="markdown")
return
elif getmode == 6:
message.delete()
bantime = extract_time(message, value)
chat.kick_member(user.id, until_date=bantime)
bot.sendMessage(chat.id, "{} banned for {} because using '{}' which in blacklist stickers".format(mention_markdown(user.id, user.first_name), value, trigger), parse_mode="markdown")
return
elif getmode == 7:
message.delete()
mutetime = extract_time(message, value)
bot.restrict_chat_member(chat.id, user.id, until_date=mutetime, can_send_messages=False)
bot.sendMessage(chat.id, "{} muted for {} because using '{}' which in blacklist stickers".format(mention_markdown(user.id, user.first_name), value, trigger), parse_mode="markdown")
return
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("Error while deleting blacklist message.")
break
def get_exception(excp, filt, chat):
if excp.message == "Unsupported url protocol":
return "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"
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 note cannot be sent because it is in the wrong format."
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)
print(query.message.chat.id)
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(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(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
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 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}."
def rest_handler(bot: Bot, update: Update):
msg = update.effective_message
chat = update.effective_chat
for restriction, filter in RESTRICTION_TYPES.items():
if filter(msg) and sql.is_restr_locked(
chat.id, restriction) and can_delete(chat, bot.id):
try:
msg.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("ERROR in restrictions")
break
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 can_delete(chat, context.bot.id):
message_id = msg.reply_to_message.message_id
if args and args[0].isdigit():
if int(args[0]) < int(1):
return
delete_to = message_id + int(args[0])
else:
delete_to = msg.message_id - 1
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":
send_message(update.effective_message, "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":
send_message(update.effective_message, "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.")
a = send_message(update.effective_message, "Purge complete.")
try:
a.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:
send_message(update.effective_message, "Reply to a message to select where to start purging from.")
return ""
def get_help(bot: Bot, update: Update):
chat = update.effective_chat
args = update.effective_message.text.split(None, 1)
# ONLY send help in PM
if chat.type != chat.PRIVATE:
update.effective_message.reply_text(
tld(chat.id, 'help_pm_only'),
reply_markup=InlineKeyboardMarkup([[
InlineKeyboardButton(text=tld(chat.id, 'btn_help'),
url="t.me/{}?start=help".format(
bot.username))
]]))
return
if len(args) >= 2:
mod_name = None
for x in HELPABLE:
if args[1].lower() == HELPABLE[x].lower():
mod_name = tld(chat.id, "modname_" + x).strip()
module = x
break
if mod_name:
help_txt = tld(chat.id, module + "_help")
if not help_txt:
LOGGER.exception(f"Help string for {module} not found!")
text = tld(chat.id, "here_is_help").format(mod_name, help_txt)
send_help(
chat.id, text,
InlineKeyboardMarkup([[
InlineKeyboardButton(text=tld(chat.id, "btn_go_back"),
callback_data="help_back")
]]))
return
update.effective_message.reply_text(tld(
chat.id, "help_not_found").format(args[1]),
parse_mode=ParseMode.HTML)
return
send_help(
chat.id,
tld(chat.id, "send-help").format(dispatcher.bot.first_name,
tld(chat.id, "cmd_multitrigger")))
def post_telegraph(update, context):
args = context.args
short_name = "Created By @MetaButler"
msg = update.effective_message # type: Optional[Message]
telegraph = Telegraph()
r = telegraph.create_account(short_name=short_name)
auth_url = r["auth_url"]
LOGGER.info(auth_url)
title_of_page = " ".join(args)
page_content = msg.reply_to_message.text
page_content = page_content.replace("\n", "<br>")
response = telegraph.create_page(
title_of_page,
html_content=page_content
)
send_message(update.effective_message, "https://telegra.ph/{}".format(response["path"]))
def snipe(bot: Bot, update: Update, args: List[str]):
try:
chat_id = str(args[0])
del args[0]
except TypeError as excp:
update.effective_message.reply_text(
"Please give me a chat to echo to!")
to_send = " ".join(args)
if len(to_send) >= 2:
try:
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?"
)
def help_button(bot: Bot, update: Update):
query = update.callback_query
chat = update.effective_chat
back_match = re.match(r"help_back", query.data)
mod_match = re.match(r"help_module\((.+?)\)", query.data)
try:
if mod_match:
module = mod_match.group(1)
mod_name = tld(chat.id, "modname_" + module).strip()
help_txt = tld(
chat.id, module +
"_help") # tld_help(chat.id, HELPABLE[module].__mod_name__)
if not help_txt:
LOGGER.exception(f"Help string for {module} not found!")
text = tld(chat.id, "here_is_help").format(mod_name, help_txt)
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(
text=tld(chat.id, "btn_go_back"),
callback_data="help_back")
]]),
disable_web_page_preview=True)
elif back_match:
bot.edit_message_text(chat_id=query.message.chat_id,
message_id=query.message.message_id,
text=tld(chat.id, "send-help").format(
dispatcher.bot.first_name,
tld(chat.id, "cmd_multitrigger")),
parse_mode=ParseMode.MARKDOWN,
reply_markup=InlineKeyboardMarkup(
paginate_modules(chat.id, 0, HELPABLE,
"help")),
disable_web_page_preview=True)
# ensure no spinny white circle
bot.answer_callback_query(query.id)
# query.message.delete()
except BadRequest:
pass
def user_join_fed(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
user = update.effective_user
msg = update.effective_message
fed_id = sql.get_fed_id(chat.id)
if is_user_fed_owner(fed_id, user.id):
user_id = extract_user(msg, args)
if user_id:
user = bot.get_chat(user_id)
elif not msg.reply_to_message and not args:
user = msg.from_user
elif not msg.reply_to_message and (
not args or
(len(args) >= 1 and not args[0].startswith("@")
and not args[0].isdigit()
and not msg.parse_entities([MessageEntity.TEXT_MENTION]))):
msg.reply_text(tld(chat.id, "common_err_no_user"))
return
else:
LOGGER.warning('error')
getuser = sql.search_user_in_fed(fed_id, user_id)
fed_id = sql.get_fed_id(chat.id)
info = sql.get_fed_info(fed_id)
get_owner = eval(info['fusers'])['owner']
get_owner = bot.get_chat(get_owner).id
if user_id == get_owner:
update.effective_message.reply_text(
tld(chat.id, "feds_promote_owner"))
return
if getuser:
update.effective_message.reply_text(
tld(chat.id, "feds_promote_owner"))
return
if user_id == bot.id:
update.effective_message.reply_text(
tld(chat.id, "feds_promote_bot"))
return
res = sql.user_join_fed(fed_id, user_id)
if res:
update.effective_message.reply_text(
tld(chat.id, "feds_promote_success"))
else:
update.effective_message.reply_text("")
else:
update.effective_message.reply_text(tld(chat.id, "feds_owner_only"))
def broadcast(bot: Bot, update: Update):
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:
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 del_blacklist_url(bot: Bot, update: Update):
chat = update.effective_chat
message = update.effective_message
parsed_entities = message.parse_entities(types=["url"])
extracted_domains = []
for obj, url in parsed_entities.items():
extract_url = tldextract.extract(url)
extracted_domains.append(extract_url.domain + "." + extract_url.suffix)
for url in sql.get_blacklisted_urls(chat.id):
if url in extracted_domains:
try:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("Error while deleting blacklist message.")
break
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))
send_message(
update.effective_message,
"Broadcast complete. {} group failed to receive message, maybe because it was kicked"
.format(failed))
def user_demote_fed(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
user = update.effective_user
fed_id = sql.get_fed_id(chat.id)
if is_user_fed_owner(fed_id, user.id):
msg = update.effective_message
user_id = extract_user(msg, args)
if user_id:
user = bot.get_chat(user_id)
elif not msg.reply_to_message and not args:
user = msg.from_user
elif not msg.reply_to_message and (
not args or
(len(args) >= 1 and not args[0].startswith("@")
and not args[0].isdigit()
and not msg.parse_entities([MessageEntity.TEXT_MENTION]))):
msg.reply_text(tld(chat.id, "common_err_no_user"))
return
else:
LOGGER.warning('error')
if user_id == bot.id:
update.effective_message.reply_text(tld(chat.id,
"feds_demote_bot"))
return
if sql.search_user_in_fed(fed_id, user_id) == False:
update.effective_message.reply_text(
tld(chat.id, "feds_demote_target_not_admin"))
return
res = sql.user_demote_fed(fed_id, user_id)
if res == True:
update.effective_message.reply_text(
tld(chat.id, "feds_demote_success"))
else:
update.effective_message.reply_text(
tld(chat.id, "feds_demote_failed"))
else:
update.effective_message.reply_text(tld(chat.id, "feds_owner_only"))
return
def setlog(update, context):
message = update.effective_message # type: Optional[Message]
chat = update.effective_chat # type: Optional[Chat]
if chat.type == chat.CHANNEL:
send_message(
update.effective_message,
"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,
"Failed to set the log channel!\nI might not be the admin on the channel."
)
sql.stop_chat_logging(chat.id)
return
else:
LOGGER.exception("ERROR in setting the log channel.")
context.bot.send_message(chat.id, "Successfully set log channel!")
else:
send_message(
update.effective_message,
"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 sban(bot: Bot, update: Update, args: List[str]) -> str:
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
message = update.effective_message # type: Optional[Message]
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 del_blacklist(bot: Bot, update: Update):
chat = update.effective_chat
message = update.effective_message
to_match = extract_text(message)
if not to_match:
return
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:
message.delete()
except BadRequest as excp:
if excp.message == "Message to delete not found":
pass
else:
LOGGER.exception("Error while deleting blacklist message.")
break
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}."
LOGGER.warning(err)
return err
def log_action(bot: Bot, update: Update, *args, **kwargs):
try:
result = func(bot, update, *args, **kwargs)
except Exception:
return
chat = update.effective_chat # type: Optional[Chat]
message = update.effective_message # type: Optional[Message]
if result:
if chat.type == chat.SUPERGROUP and chat.username:
result += tld(chat.id, "log_channel_link").format(
chat.username, message.message_id)
log_chat = sql.get_chat_log_channel(chat.id)
if log_chat:
send_log(bot, log_chat, chat.id, result)
elif result == "":
pass
else:
LOGGER.warning(
"%s was set as loggable, but had no return statement.", func)
return result
def send_message_raw(chat_id, text, *args, **kwargs):
try:
return dispatcher.bot.sendMessage(chat_id, text, *args, **kwargs)
except error.BadRequest as err:
if str(err) == "Reply message not found":
try:
if kwargs.get('reply_to_message_id'):
kwargs['reply_to_message_id'] = None
return dispatcher.bot.sendMessage(chat_id, text, *args,
**kwargs)
except error.BadRequest as err:
LOGGER.exception("ERROR: {}".format(err))
'''elif str(err) == "Have no rights to send a message":
try:
dispatcher.bot.leaveChat(message.chat.id)
dispatcher.bot.sendMessage(DUMP_CHAT, "I am leave chat `{}`\nBecause of: `Muted`".format(message.chat.title))
except error.BadRequest as err:
if str(err) == "Chat not found":
pass'''
else:
LOGGER.exception("ERROR: {}".format(err))
def send_message(message, text, target_id=None, *args, **kwargs):
if not target_id:
try:
return message.reply_text(text, *args, **kwargs)
except error.BadRequest as err:
if str(err) == "Reply message not found":
try:
return message.reply_text(text,
quote=False,
*args,
**kwargs)
except error.BadRequest as err:
LOGGER.exception("ERROR: {}".format(err))
elif str(err) == "Have no rights to send a message":
try:
dispatcher.bot.leaveChat(message.chat.id)
dispatcher.bot.sendMessage(
DUMP_CHAT,
"I am leave chat `{}`\nBecause of: `Muted`".format(
message.chat.title))
except error.BadRequest as err:
if str(err) == "Chat not found":
pass
else:
LOGGER.exception("ERROR: {}".format(err))
else:
try:
dispatcher.bot.send_message(target_id, text, *args, **kwarg)
except error.BadRequest as err:
LOGGER.exception("ERROR: {}".format(err))
def main():
# test_handler = CommandHandler("test", test) #Unused variable
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")
migrate_handler = MessageHandler(Filters.status_update.migrate,
migrate_chats)
# dispatcher.add_handler(test_handler)
dispatcher.add_handler(start_handler)
dispatcher.add_handler(start_callback_handler)
dispatcher.add_handler(help_handler)
dispatcher.add_handler(help_callback_handler)
dispatcher.add_handler(migrate_handler)
# dispatcher.add_error_handler(error_callback)
# add antiflood processor
Dispatcher.process_update = process_update
LOGGER.info("Using long polling.")
# updater.start_polling(timeout=15, read_latency=4, clean=True)
updater.start_polling(poll_interval=0.0,
timeout=10,
clean=True,
bootstrap_retries=-1,
read_latency=3.0)
LOGGER.info("Successfully loaded")
if len(argv) not in (1, 3, 4):
tbot.disconnect()
else:
tbot.run_until_disconnected()
updater.idle()
