def invite(update, context):
user = update.effective_user
msg = update.effective_message
chat = update.effective_chat
context.args
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
else:
if msg.chat.type == "private":
msg.reply_text("This command is meant to use in chat not in PM")
return ""
chat = update.effective_chat
if chat.username:
msg.reply_text(chat.username)
elif chat.type == chat.SUPERGROUP or chat.type == chat.CHANNEL:
bot_member = chat.get_member(context.bot.id)
if bot_member.can_invite_users:
invitelink = context.bot.exportChatInviteLink(chat.id)
msg.reply_text(invitelink)
else:
msg.reply_text(
"I don't have access to the invite link, try changing my permissions!"
)
else:
msg.reply_text(
"I can only give you invite links for supergroups and channels, sorry!"
)
def clear(update, context):
args = context.args
chat = update.effective_chat
user = update.effective_user
msg = update.effective_message
conn = connected(context.bot, update, chat, user.id)
note_name, text, data_type, content, buttons = get_note_type(msg)
if not conn == False:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
chat_name = "local notes"
else:
chat_name = chat.title
if len(args) >= 1:
notename = args[0].lower()
if sql.rm_note(chat_id, notename):
update.effective_message.reply_text(
"Successfully deleted '`{note_name}`' from {chat_name}!".
format(note_name=note_name, chat_name=chat_name),
parse_mode=ParseMode.MARKDOWN,
)
else:
update.effective_message.reply_text(
"There is no such notes saved in {chat_name}!".format(
chat_name=chat_name))
def save(update, context):
chat = update.effective_chat
user = update.effective_user
conn = connected(context.bot, update, chat, user.id)
if not conn == False:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
chat_name = "local notes"
else:
chat_name = chat.title
msg = update.effective_message
note_name, text, data_type, content, buttons = get_note_type(msg)
note_name = note_name.lower()
if data_type is None:
msg.reply_text("Bruh! there's no note")
return
sql.add_note_to_db(chat_id,
note_name,
text,
data_type,
buttons=buttons,
file=content)
msg.reply_text(
"Saved '`{note_name}`' in *{chat_name}*.\nGet it with `/get {note_name}`, or `#{note_name}`!"
.format(note_name=note_name, chat_name=chat_name),
parse_mode=ParseMode.MARKDOWN,
)
def list_warn_filters(update, context):
chat = update.effective_chat
user = update.effective_user
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat_id = conn
else:
if chat.type == "private":
return
else:
chat_id = update.effective_chat.id
all_handlers = sql.get_chat_warn_triggers(chat_id)
if not all_handlers:
update.effective_message.reply_text("No warning filters are active here!")
return
filter_list = CURRENT_WARNING_FILTER_STRING
for keyword in all_handlers:
entry = " - {}\n".format(html.escape(keyword))
if len(entry) + len(filter_list) > telegram.MAX_MESSAGE_LENGTH:
update.effective_message.reply_text(filter_list, parse_mode=ParseMode.HTML)
filter_list = entry
else:
filter_list += entry
if not filter_list == CURRENT_WARNING_FILTER_STRING:
update.effective_message.reply_text(filter_list, parse_mode=ParseMode.HTML)
def set_warn_strength(update, context):
chat = update.effective_chat
user = update.effective_user
msg = update.effective_message
args = context.args
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if chat.type == "private":
return
else:
chat_id = update.effective_chat.id
chat_name = chat.title
if args:
if args[0].lower() in ("on", "yes"):
sql.set_warn_strength(chat_id, False)
msg.reply_text("Too many warns will now result in a ban!")
return (
"<b>{}:</b>\n"
"<b>Admin:</b> {}\n"
"Has enabled strong warns. Users will be banned.".format(
chat_name, mention_html(user.id, user.first_name)
)
)
elif args[0].lower() in ("off", "no"):
sql.set_warn_strength(chat_id, True)
msg.reply_text(
"Too many warns will now result in a kick! Users will be able to join again after."
)
return (
"<b>{}:</b>\n"
"<b>Admin:</b> {}\n"
"Has disabled strong warns. Users will only be kicked.".format(
chat_name, mention_html(user.id, user.first_name)
)
)
else:
msg.reply_text("I only understand on/yes/no/off!")
else:
_, soft_warn = sql.get_warn_setting(chat_id)
if soft_warn:
msg.reply_text(
"Warns are currently set to *kick* users when they exceed the limits.",
parse_mode=ParseMode.MARKDOWN,
)
else:
msg.reply_text(
"Warns are currently set to *ban* users when they exceed the limits.",
parse_mode=ParseMode.MARKDOWN,
)
return ""
def warns(update, context):
message = update.effective_message
chat = update.effective_chat
args = context.args
user = update.effective_user
user_id = extract_user(message, args) or update.effective_user.id
conn = connected(context.bot, update, chat, user.id, need_admin=False)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if chat.type == "private":
return
else:
chat_id = update.effective_chat.id
chat_name = chat.title
result = sql.get_warns(user_id, chat_id)
num = 1
if result and result[0] != 0:
num_warns, reasons = result
limit, _ = sql.get_warn_setting(chat_id)
if reasons:
if conn:
text = "This user has {}/{} warnings, in *{}* for the following reasons:".format(
num_warns, limit, chat_name
)
else:
text = (
"This user has {}/{} warnings, for the following reasons:".format(
num_warns,
limit,
)
)
for reason in reasons:
text += "\n {}. {}".format(num, reason)
num += 1
msgs = split_message(text)
for msg in msgs:
update.effective_message.reply_text(msg, parse_mode="markdown")
else:
update.effective_message.reply_text(
"User has {}/{} warnings, but no reasons for any of them.".format(
num_warns, limit
),
parse_mode="markdown",
)
else:
update.effective_message.reply_text("This user hasn't got any warnings!")
def set_warn_limit(update, context) -> str:
chat = update.effective_chat
user = update.effective_user
msg = update.effective_message
args = context.args
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if chat.type == "private":
return
else:
chat_id = update.effective_chat.id
chat_name = chat.title
if args:
if args[0].isdigit():
if int(args[0]) < 3:
msg.reply_text("The minimum warn limit is 3!")
else:
sql.set_warn_limit(chat_id, int(args[0]))
msg.reply_text(
"Updated the warn limit to `{}` in *{}*".format(
escape_markdown(args[0]), chat_name
),
parse_mode="markdown",
)
return (
"<b>{}:</b>"
"\n#SET_WARN_LIMIT"
"\n<b>Admin:</b> {}"
"\nSet the warn limit to <code>{}</code>".format(
html.escape(chat_name),
mention_html(user.id, user.first_name),
args[0],
)
)
else:
msg.reply_text("Give me a number as an arg!")
else:
limit, _ = sql.get_warn_setting(chat_id)
msg.reply_text(
"The current warn in {} limit is {}".format(chat_name, limit)
)
return ""
def add_blacklist(update, context):
msg = update.effective_message
chat = update.effective_chat
user = update.effective_user
words = msg.text.split(None, 1)
conn = connected(context.bot, update, chat, user.id)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
return
else:
chat_name = chat.title
if len(words) > 1:
text = words[1]
to_blacklist = list(set(trigger.strip()
for trigger in text.split("\n") if trigger.strip()))
for trigger in to_blacklist:
sql.add_to_blacklist(chat_id, trigger.lower())
if len(to_blacklist) == 1:
send_message(
update.effective_message,
"Added blacklist <code>{}</code> in chat: <b>{}</b>!".format(
html.escape(to_blacklist[0]), chat_name
),
parse_mode=ParseMode.HTML,
)
else:
send_message(
update.effective_message,
"Added blacklist trigger: <code>{}</code> in <b>{}</b>!".format(
len(to_blacklist),
chat_name),
parse_mode=ParseMode.HTML,
)
else:
send_message(
update.effective_message,
"Tell me which words you would like to add in blacklist.",
)
def enable(update, context):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user
args = context.args
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_name = dispatcher.bot.getChat(conn).title
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command is meant to be used in group not in PM",
)
return ""
chat = update.effective_chat
update.effective_chat.id
chat_name = update.effective_message.chat.title
if len(args) >= 1:
enable_cmd = args[0]
if enable_cmd.startswith(CMD_STARTERS):
enable_cmd = enable_cmd[1:]
if sql.enable_command(chat.id, enable_cmd):
if conn:
text = "Enabled the use of `{}` command in *{}*!".format(
enable_cmd, chat_name
)
else:
text = "Enabled the use of `{}` command!".format(
enable_cmd
)
send_message(
update.effective_message,
text,
parse_mode=ParseMode.MARKDOWN,
)
else:
send_message(
update.effective_message, "Is that even disabled?"
)
else:
send_message(update.effective_message, "What should I enable?")
def commands(update, context):
chat = update.effective_chat
user = update.effective_user
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command is meant to use in group not in PM",
)
return ""
chat = update.effective_chat
text = build_curr_disabled(chat.id)
send_message(update.effective_message, text, parse_mode=ParseMode.MARKDOWN)
def add_warn_filter(update, context):
chat = update.effective_chat
user = update.effective_user
msg = update.effective_message
args = msg.text.split(
None, 1
) # use python's maxsplit to separate Cmd, keyword, and reply_text
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if chat.type == "private":
return
else:
chat_id = update.effective_chat.id
chat_name = chat.title
if len(args) < 2:
return
extracted = split_quotes(args[1])
if len(extracted) >= 2:
# set trigger -> lower, so as to avoid adding duplicate filters with
# different cases
keyword = extracted[0].lower()
content = extracted[1]
else:
return
# Note: perhaps handlers can be removed somehow using sql.get_chat_filters
for handler in dispatcher.handlers.get(WARN_HANDLER_GROUP, []):
if handler.filters == (keyword, chat_id):
dispatcher.remove_handler(handler, WARN_HANDLER_GROUP)
sql.add_warn_filter(chat_id, keyword, content)
update.effective_message.reply_text(
"Warn filter added for `{}` in *{}*!".format(keyword, chat_name),
parse_mode="markdown",
)
raise DispatcherHandlerStop
def list_handlers(update, context):
chat = update.effective_chat
user = update.effective_user
conn = connected(context.bot, update, chat, user.id, need_admin=False)
if not conn == False:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
filter_list = "*Filter in {}:*\n"
else:
chat_id = update.effective_chat.id
if chat.type == "private":
chat_name = "Local filters"
filter_list = "*local filters:*\n"
else:
chat_name = chat.title
filter_list = "*Filters in {}*:\n"
all_handlers = sql.get_chat_triggers(chat_id)
if not all_handlers:
send_message(
update.effective_message,
"No filters saved in {}!".format(chat_name),
)
return
for keyword in all_handlers:
entry = " × `{}`\n".format(escape_markdown(keyword))
if len(entry) + len(filter_list) > telegram.MAX_MESSAGE_LENGTH:
send_message(
update.effective_message,
filter_list.format(chat_name),
parse_mode=telegram.ParseMode.MARKDOWN,
)
filter_list = entry
else:
filter_list += entry
send_message(
update.effective_message,
filter_list.format(chat_name),
parse_mode=telegram.ParseMode.MARKDOWN,
)
def save(update, context):
chat = update.effective_chat
user = update.effective_user
conn = connected(context.bot, update, chat, user.id)
if not conn == False:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
chat_name = "local notes"
else:
chat_name = chat.title
msg = update.effective_message
try:
note_name, text, data_type, content, buttons = get_note_type(msg)
except IndexError:
msg.reply_text(
"This isn't right things T_T\nUsage: /save <notename> <msg/reply to msg>"
)
return
note_name = note_name.lower()
if data_type is None:
msg.reply_text("Bruh! there's no note")
return
if len(text.strip()) == 0:
text = note_name
sql.add_note_to_db(chat_id,
note_name,
text,
data_type,
buttons=buttons,
file=content)
msg.reply_text(
"Saved '`{note_name}`' in *{chat_name}*.\nGet it with `/get {note_name}`, or `#{note_name}`!"
.format(note_name=note_name, chat_name=chat_name),
parse_mode=ParseMode.MARKDOWN,
)
def disable(update, context):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user
args = context.args
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_name = dispatcher.bot.getChat(conn).title
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command meant to be used in group not in PM",
)
return ""
chat = update.effective_chat
chat_name = update.effective_message.chat.title
if len(args) >= 1:
disable_cmd = args[0]
if disable_cmd.startswith(CMD_STARTERS):
disable_cmd = disable_cmd[1:]
if disable_cmd in set(DISABLE_CMDS + DISABLE_OTHER):
disable_db.disable_command(chat.id, disable_cmd)
if conn:
text = "Disabled the use of `{}` command in *{}*!".format(
disable_cmd, chat_name)
else:
text = "Disabled the use of `{}` command!".format(
disable_cmd)
send_message(
update.effective_message,
text,
parse_mode=ParseMode.MARKDOWN,
)
else:
send_message(update.effective_message,
"This command can't be disabled")
else:
send_message(update.effective_message, "What should I disable?")
def remove_warn_filter(update, context):
chat = update.effective_chat
user = update.effective_user
msg = update.effective_message
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat_id = conn
else:
if chat.type == "private":
return
else:
chat_id = update.effective_chat.id
args = msg.text.split(
None, 1
) # use python's maxsplit to separate Cmd, keyword, and reply_text
if len(args) < 2:
return
extracted = split_quotes(args[1])
if len(extracted) < 1:
return
to_remove = extracted[0]
chat_filters = sql.get_chat_warn_triggers(chat_id)
if not chat_filters:
msg.reply_text("No warning filters are active here!")
return
for filt in chat_filters:
if filt == to_remove:
sql.remove_warn_filter(chat_id, to_remove)
msg.reply_text("Yep, I'll stop warning people for that.")
raise DispatcherHandlerStop
msg.reply_text(
"That's not a current warning filter - click: /warnlist for all active warning filters."
)
def stop_filter(update, context):
chat = update.effective_chat
user = update.effective_user
args = update.effective_message.text.split(None, 1)
conn = connected(context.bot, update, chat, user.id)
if not conn == False:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
chat_name = "Local filters"
else:
chat_name = chat.title
if len(args) < 2:
send_message(update.effective_message, "What should i stop?")
return
chat_filters = sql.get_chat_triggers(chat_id)
if not chat_filters:
send_message(update.effective_message, "No filters active here!")
return
for keyword in chat_filters:
if keyword == args[1]:
sql.remove_filter(chat_id, args[1])
send_message(
update.effective_message,
"Okay, I'll stop replying to that filter in *{}*.".format(
chat_name
),
parse_mode=telegram.ParseMode.MARKDOWN,
)
raise DispatcherHandlerStop
send_message(
update.effective_message,
"That's not a filter - Click: /filters to get currently active filters.",
)
def flood(update, context):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
msg = update.effective_message
conn = connected(context.bot, update, chat, user.id, need_admin=False)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command is meant to use in group not in PM",
)
return
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
limit = sql.get_flood_limit(chat_id)
if limit == 0:
if conn:
text = msg.reply_text(
"I'm not enforcing any flood control in {}!".format(chat_name)
)
else:
text = msg.reply_text("I'm not enforcing any flood control here!")
send_message(update.effective_message, text, parse_mode="markdown")
else:
if conn:
text = msg.reply_text(
"I'm currently restricting members after {} consecutive messages in {}.".format(
limit, chat_name
)
)
else:
text = msg.reply_text(
"I'm currently restricting members after {} consecutive messages.".format(
limit
)
)
send_message(update.effective_message, text, parse_mode="markdown")
def blacklist(update, context):
chat = update.effective_chat
user = update.effective_user
args = context.args
conn = connected(context.bot, update, chat, user.id, need_admin=False)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if chat.type == "private":
return
else:
chat_id = update.effective_chat.id
chat_name = chat.title
filter_list = "Current blacklisted words in <b>{}</b>:\n".format(chat_name)
all_blacklisted = sql.get_chat_blacklist(chat_id)
if len(args) > 0 and args[0].lower() == "copy":
for trigger in all_blacklisted:
filter_list += "<code>{}</code>\n".format(html.escape(trigger))
else:
for trigger in all_blacklisted:
filter_list += " - <code>{}</code>\n".format(html.escape(trigger))
# for trigger in all_blacklisted:
# filter_list += " - <code>{}</code>\n".format(html.escape(trigger))
split_text = split_message(filter_list)
for text in split_text:
if filter_list == "Current blacklisted words in <b>{}</b>:\n".format(
chat_name):
send_message(
update.effective_message,
"No blacklisted words in <b>{}</b>!".format(chat_name),
parse_mode=ParseMode.HTML,
)
return
send_message(update.effective_message, text, parse_mode=ParseMode.HTML)
def list_notes(update, context):
chat_id = update.effective_chat.id
chat = update.effective_chat
user = update.effective_user
conn = connected(context.bot, update, chat, user.id, need_admin=False)
if not conn == False:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
msg = "*Notes in {}:*\n"
else:
chat_id = update.effective_chat.id
if chat.type == "private":
chat_name = ""
msg = "*Local Notes:*\n"
else:
chat_name = chat.title
msg = "*Notes saved in {}:*\n"
note_list = sql.get_all_chat_notes(chat_id)
des = "You can get notes by using `/get notename`, or `#notename`.\n"
for note in note_list:
note_name = " × `{}`\n".format(note.name.lower())
if len(msg) + len(note_name) > MAX_MESSAGE_LENGTH:
update.effective_message.reply_text(msg,
parse_mode=ParseMode.MARKDOWN)
msg = ""
msg += note_name
if not note_list:
update.effective_message.reply_text("No notes saved here!")
elif len(msg) != 0:
try:
update.effective_message.reply_text(msg.format(chat_name) + des,
parse_mode=ParseMode.MARKDOWN)
except ValueError:
update.effective_message.reply_text(
"There was a problem in showing notes list, maybe due to some invalid character in note names. Ask in @userbotindo if you're unable to figure it out!"
)
def list_locks(update, context):
chat = update.effective_chat
user = update.effective_user
# Connection check
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_name = chat.title
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command is meant to use in group not in PM",
)
return ""
chat = update.effective_chat
chat_name = update.effective_message.chat.title
res = build_lock_message(chat.id)
if conn:
res = res.replace("Locks in", "*{}*".format(chat_name))
send_message(update.effective_message, res, parse_mode=ParseMode.MARKDOWN)
def import_data(update, context):
msg = update.effective_message
chat = update.effective_chat
user = update.effective_user
# TODO: allow uploading doc with command, not just as reply
# only work with a doc
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_name = dispatcher.bot.getChat(conn).title
else:
if update.effective_message.chat.type == "private":
update.effective_message.reply_text(
"This command can only be runned on group, not PM."
)
return ""
chat = update.effective_chat
chat_name = update.effective_message.chat.title
if msg.reply_to_message and msg.reply_to_message.document:
try:
file_info = context.bot.get_file(msg.reply_to_message.document.file_id)
except BadRequest:
msg.reply_text(
"Try downloading and uploading the file yourself again, This one seem broken!"
)
return
with BytesIO() as file:
file_info.download(out=file)
file.seek(0)
data = json.load(file)
# only import one group
if len(data) > 1 and str(chat.id) not in data:
msg.reply_text(
"There are more than one group in this file and the chat.id is not same! How am i supposed to import it?"
)
return
# Check if backup is this chat
try:
if data.get(str(chat.id)) is None:
if conn:
text = "Backup comes from another chat, I can't return another chat to chat *{}*".format(
chat_name
)
else:
text = "Backup comes from another chat, I can't return another chat to this chat"
return msg.reply_text(text, parse_mode="markdown")
except Exception:
return msg.reply_text("There is problem while importing the data!")
# Check if backup is from self
try:
if str(context.bot.id) != str(data[str(chat.id)]["bot"]):
return msg.reply_text(
"Backup from another bot that is not suggested might cause the problem, documents, photos, videos, audios, records might not work as it should be."
)
except Exception:
pass
# Select data source
if str(chat.id) in data:
data = data[str(chat.id)]["hashes"]
else:
data = data[list(data.keys())[0]]["hashes"]
try:
for mod in DATA_IMPORT:
mod.__import_data__(str(chat.id), data)
except Exception:
msg.reply_text(
"An error occurred while recovering your data. The process failed. If you experience a problem with this, please ask @starryboi"
)
LOGGER.exception(
"Imprt for the chat %s with the name %s failed.",
str(chat.id),
str(chat.title),
)
return
# TODO: some of that link logic
# NOTE: consider default permissions stuff?
if conn:
text = "Backup fully restored on *{}*.".format(chat_name)
else:
text = "Backup fully restored"
msg.reply_text(text, parse_mode="markdown")
def permanent_pin_set(update, context) -> str:
user = update.effective_user
chat = update.effective_chat
args = context.args
bot = context.bot
conn = connected(bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_id = conn
dispatcher.bot.getChat(conn).title
if not args:
get_permapin = sql.get_permapin(chat_id)
text_maker = "Permanent pin is currently set:`{}`".format(
bool(int(get_permapin)))
if get_permapin:
if chat.username:
old_pin = "https://t.me/{}/{}".format(
chat.username, get_permapin)
else:
old_pin = "https://t.me/c/{}/{}".format(
str(chat.id)[4:], get_permapin)
text_maker += "\nTo disable permanent pin: `/permanentpin off`"
text_maker += "\n\n[Permanent pin message is here]({})".format(
old_pin)
update.effective_message.reply_text(text_maker,
parse_mode="markdown")
return ""
prev_message = args[0]
if prev_message == "off":
sql.set_permapin(chat_id, 0)
update.effective_message.reply_text(
"Permanently pin has been disabled!")
return
if "/" in prev_message:
prev_message = prev_message.split("/")[-1]
else:
if update.effective_message.chat.type == "private":
update.effective_message.reply_text(
"You can do this command on groups, not on PM!!!")
return ""
chat = update.effective_chat
chat_id = update.effective_chat.id
update.effective_message.chat.title
if update.effective_message.reply_to_message:
prev_message = update.effective_message.reply_to_message.message_id
elif len(args) >= 1 and args[0] == "off":
sql.set_permapin(chat.id, 0)
update.effective_message.reply_text(
"Permanently pin has been disabled!")
return
else:
get_permapin = sql.get_permapin(chat.id)
text_maker = "Current permanent pin: `{}`".format(
bool(int(get_permapin)))
if get_permapin:
if chat.username:
old_pin = "https://t.me/{}/{}".format(
chat.username, get_permapin)
else:
old_pin = "https://t.me/c/{}/{}".format(
str(chat.id)[4:], get_permapin)
text_maker += "\nTo disable permanent pin: `/permanentpin off`"
text_maker += "\n\n[Permanent pin message is here]({})".format(
old_pin)
update.effective_message.reply_text(text_maker,
parse_mode="markdown")
return ""
is_group = chat.type != "private" and chat.type != "channel"
if prev_message and is_group:
sql.set_permapin(chat.id, prev_message)
update.effective_message.reply_text("Permanent Pin Set successfully!")
return ("<b>{}:</b>"
"\n#PERMANENT_PIN"
"\n<b>Admin:</b> {}".format(
html.escape(chat.title),
mention_html(user.id, user.first_name)))
return ""
def set_flood(update, context) -> str:
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
message = update.effective_message # type: Optional[Message]
args = context.args
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command is meant to use in group not in PM",
)
return ""
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
if len(args) >= 1:
val = args[0].lower()
if val == "off" or val == "no" or val == "0":
sql.set_flood(chat_id, 0)
if conn:
text = message.reply_text(
"Antiflood has been disabled in {}.".format(chat_name)
)
else:
text = message.reply_text("Antiflood has been disabled.")
send_message(update.effective_message, text, parse_mode="markdown")
elif val.isdigit():
amount = int(val)
if amount <= 0:
sql.set_flood(chat_id, 0)
if conn:
text = message.reply_text(
"Antiflood has been disabled in {}.".format(chat_name)
)
else:
text = message.reply_text("Antiflood has been disabled.")
return (
"<b>{}:</b>"
"\n#SETFLOOD"
"\n<b>Admin:</b> {}"
"\nDisable antiflood.".format(
html.escape(chat_name), mention_html(user.id, user.first_name)
)
)
elif amount < 3:
send_message(
update.effective_message,
"Antiflood must be either 0 (disabled) or number greater than 3!",
)
return ""
else:
sql.set_flood(chat_id, amount)
if conn:
text = message.reply_text(
"Anti-flood has been set to {} in chat: {}".format(
amount, chat_name
)
)
else:
text = message.reply_text(
"Successfully updated anti-flood limit to {}!".format(amount)
)
send_message(update.effective_message, text, parse_mode="markdown")
return (
"<b>{}:</b>"
"\n#SETFLOOD"
"\n<b>Admin:</b> {}"
"\nSet antiflood to <code>{}</code>.".format(
html.escape(chat_name),
mention_html(user.id, user.first_name),
amount,
)
)
else:
message.reply_text("Invalid argument please use a number, 'off' or 'no'")
else:
message.reply_text(
(
"Use `/setflood number` to enable anti-flood.\nOr use `/setflood off` to disable antiflood!."
),
parse_mode="markdown",
)
return ""
def filters(update, context):
chat = update.effective_chat
user = update.effective_user
msg = update.effective_message
args = msg.text.split(
None,
1) # use python's maxsplit to separate Cmd, keyword, and reply_text
conn = connected(context.bot, update, chat, user.id)
if not conn == False:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
chat_name = "local filters"
else:
chat_name = chat.title
if not msg.reply_to_message and len(args) < 2:
send_message(
update.effective_message,
"Please provide keyboard keyword for this filter to reply with!",
)
return
if msg.reply_to_message:
if len(args) < 2:
send_message(
update.effective_message,
"Please provide keyword for this filter to reply with!",
)
return
else:
keyword = args[1]
else:
extracted = split_quotes(args[1])
if len(extracted) < 1:
return
# set trigger -> lower, so as to avoid adding duplicate filters with
# different cases
keyword = extracted[0].lower()
# Add the filter
# Note: perhaps handlers can be removed somehow using sql.get_chat_filters
for handler in dispatcher.handlers.get(HANDLER_GROUP, []):
if handler.filters == (keyword, chat_id):
dispatcher.remove_handler(handler, HANDLER_GROUP)
text, file_type, file_id = get_filter_type(msg)
if not msg.reply_to_message and len(extracted) >= 2:
offset = len(extracted[1]) - len(
msg.text) # set correct offset relative to command + notename
text, buttons = button_markdown_parser(extracted[1],
entities=msg.parse_entities(),
offset=offset)
text = text.strip()
if not text:
send_message(
update.effective_message,
"There is no note message - You can't JUST have buttons, you need a message to go with it!",
)
return
elif msg.reply_to_message and len(args) >= 2:
if msg.reply_to_message.text:
text_to_parsing = msg.reply_to_message.text
elif msg.reply_to_message.caption:
text_to_parsing = msg.reply_to_message.caption
else:
text_to_parsing = ""
offset = len(text_to_parsing
) # set correct offset relative to command + notename
text, buttons = button_markdown_parser(text_to_parsing,
entities=msg.parse_entities(),
offset=offset)
text = text.strip()
elif not text and not file_type:
send_message(
update.effective_message,
"Please provide keyword for this filter reply with!",
)
return
elif msg.reply_to_message:
if msg.reply_to_message.text:
text_to_parsing = msg.reply_to_message.text
elif msg.reply_to_message.caption:
text_to_parsing = msg.reply_to_message.caption
else:
text_to_parsing = ""
offset = len(text_to_parsing
) # set correct offset relative to command + notename
text, buttons = button_markdown_parser(text_to_parsing,
entities=msg.parse_entities(),
offset=offset)
text = text.strip()
if (msg.reply_to_message.text
or msg.reply_to_message.caption) and not text:
send_message(
update.effective_message,
"There is no note message - You can't JUST have buttons, you need a message to go with it!",
)
return
else:
send_message(update.effective_message, "Invalid filter!")
return
add = addnew_filter(update, chat_id, keyword, text, file_type, file_id,
buttons)
# This is an old method
# sql.add_filter(chat_id, keyword, content, is_sticker, is_document, is_image, is_audio, is_voice, is_video, buttons)
if add:
send_message(
update.effective_message,
"Saved filter '{}' in *{}*!".format(keyword, chat_name),
parse_mode=telegram.ParseMode.MARKDOWN,
)
raise DispatcherHandlerStop
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:
if (ENUM_FUNC_MAP[note.msgtype] ==
dispatcher.bot.send_sticker):
ENUM_FUNC_MAP[note.msgtype](
chat_id,
note.file,
reply_to_message_id=reply_id,
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,
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 set_flood_mode(update, context):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
msg = update.effective_message # type: Optional[Message]
args = context.args
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command is meant to use in group not in PM",
)
return ""
chat = update.effective_chat
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
if args:
if args[0].lower() == "ban":
settypeflood = "ban"
sql.set_flood_strength(chat_id, 1, "0")
elif args[0].lower() == "kick":
settypeflood = "kick"
sql.set_flood_strength(chat_id, 2, "0")
elif args[0].lower() == "mute":
settypeflood = "mute"
sql.set_flood_strength(chat_id, 3, "0")
elif args[0].lower() == "tban":
if len(args) == 1:
teks = """It looks like you tried to set time value for antiflood but you didn't specified time; Try, `/setfloodmode tban <timevalue>`.
Examples of time value: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks."""
send_message(update.effective_message, teks, parse_mode="markdown")
return
settypeflood = "tban for {}".format(args[1])
sql.set_flood_strength(chat_id, 4, str(args[1]))
elif args[0].lower() == "tmute":
if len(args) == 1:
teks = """It looks like you tried to set time value for antiflood but you didn't specified time; Try, `/setfloodmode tmute <timevalue>`.
Examples of time value: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks."""
send_message(update.effective_message, teks, parse_mode="markdown")
return
settypeflood = "tmute for {}".format(args[1])
sql.set_flood_strength(chat_id, 5, str(args[1]))
else:
send_message(
update.effective_message, "I only understand ban/kick/mute/tban/tmute!"
)
return
if conn:
text = msg.reply_text(
"Exceeding consecutive flood limit will result in {} in {}!".format(
settypeflood, chat_name
)
)
else:
text = msg.reply_text(
"Exceeding consecutive flood limit will result in {}!".format(
settypeflood
)
)
send_message(update.effective_message, text, parse_mode="markdown")
return (
"<b>{}:</b>\n"
"<b>Admin:</b> {}\n"
"Has changed antiflood mode. User will {}.".format(
settypeflood,
html.escape(chat.title),
mention_html(user.id, user.first_name),
)
)
else:
getmode, getvalue = sql.get_flood_setting(chat.id)
if getmode == 1:
settypeflood = "ban"
elif getmode == 2:
settypeflood = "kick"
elif getmode == 3:
settypeflood = "mute"
elif getmode == 4:
settypeflood = "tban for {}".format(getvalue)
elif getmode == 5:
settypeflood = "tmute for {}".format(getvalue)
if conn:
text = msg.reply_text(
"Sending more messages than flood limit will result in {} in {}.".format(
settypeflood, chat_name
)
)
else:
text = msg.reply_text(
"Sending more message than flood limit will result in {}.".format(
settypeflood
)
)
send_message(update.effective_message, text, parse_mode=ParseMode.MARKDOWN)
return ""
def lock(update, context) -> str:
args = context.args
chat = update.effective_chat
user = update.effective_user
if (
can_delete(chat, context.bot.id)
or update.effective_message.chat.type == "private"
):
if len(args) >= 1:
ltype = args[0].lower()
if ltype in LOCK_TYPES:
# Connection check
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_id = conn
chat_name = chat.title
text = "Locked all {} messages for non-admins in {}!".format(
ltype, chat_name
)
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command is meant to use in group not in PM",
)
return ""
chat = update.effective_chat
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
text = "Locked all {} messages for non-admins!".format(ltype)
sql.update_lock(chat.id, ltype, locked=True)
send_message(update.effective_message, text, parse_mode="markdown")
return (
"<b>{}:</b>"
"\n#LOCK"
"\n<b>Admin:</b> {}"
"\nLocked <code>{}</code>.".format(
html.escape(chat.title),
mention_html(user.id, user.first_name),
ltype,
)
)
elif ltype in LOCK_CHAT_RESTRICTION:
# Connection check
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_id = conn
chat_name = chat.title
text = "Locked {} for all non-admins in {}!".format(
ltype, chat_name
)
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command is meant to use in group not in PM",
)
return ""
chat = update.effective_chat
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
text = "Locked {} for all non-admins!".format(ltype)
current_permission = context.bot.getChat(chat_id).permissions
context.bot.set_chat_permissions(
chat_id=chat_id,
permissions=get_permission_list(
eval(str(current_permission)),
LOCK_CHAT_RESTRICTION[ltype.lower()],
),
)
send_message(update.effective_message, text, parse_mode="markdown")
return (
"<b>{}:</b>"
"\n#Permission_LOCK"
"\n<b>Admin:</b> {}"
"\nLocked <code>{}</code>.".format(
html.escape(chat.title),
mention_html(user.id, user.first_name),
ltype,
)
)
else:
send_message(
update.effective_message,
"What are you trying to lock...? Try /locktypes for the list of lockables",
)
else:
send_message(update.effective_message, "What are you trying to lock...?")
else:
send_message(
update.effective_message,
"I am not administrator or haven't got enough rights.",
)
return ""
def blacklist_mode(update, context):
chat = update.effective_chat
user = update.effective_user
msg = update.effective_message
args = context.args
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command can be only used in group not in PM",
)
return ""
chat = update.effective_chat
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
if args:
if (
args[0].lower() == "off"
or args[0].lower() == "nothing"
or args[0].lower() == "no"
):
settypeblacklist = "do nothing"
sql.set_blacklist_strength(chat_id, 0, "0")
elif args[0].lower() == "del" or args[0].lower() == "delete":
settypeblacklist = "will delete blacklisted message"
sql.set_blacklist_strength(chat_id, 1, "0")
elif args[0].lower() == "warn":
settypeblacklist = "warn the sender"
sql.set_blacklist_strength(chat_id, 2, "0")
elif args[0].lower() == "mute":
settypeblacklist = "mute the sender"
sql.set_blacklist_strength(chat_id, 3, "0")
elif args[0].lower() == "kick":
settypeblacklist = "kick the sender"
sql.set_blacklist_strength(chat_id, 4, "0")
elif args[0].lower() == "ban":
settypeblacklist = "ban the sender"
sql.set_blacklist_strength(chat_id, 5, "0")
elif args[0].lower() == "tban":
if len(args) == 1:
teks = """It looks like you tried to set time value for blacklist but you didn't specified time; Try, `/blacklistmode tban <timevalue>`.
Examples of time value: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks."""
send_message(
update.effective_message,
teks,
parse_mode="markdown")
return ""
restime = extract_time(msg, args[1])
if not restime:
teks = """Invalid time value!
Example of time value: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks."""
send_message(
update.effective_message,
teks,
parse_mode="markdown")
return ""
settypeblacklist = "temporarily ban for {}".format(args[1])
sql.set_blacklist_strength(chat_id, 6, str(args[1]))
elif args[0].lower() == "tmute":
if len(args) == 1:
teks = """It looks like you tried to set time value for blacklist but you didn't specified time; try, `/blacklistmode tmute <timevalue>`.
Examples of time value: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks."""
send_message(
update.effective_message,
teks,
parse_mode="markdown")
return ""
restime = extract_time(msg, args[1])
if not restime:
teks = """Invalid time value!
Examples of time value: 4m = 4 minutes, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks."""
send_message(
update.effective_message,
teks,
parse_mode="markdown")
return ""
settypeblacklist = "temporarily mute for {}".format(args[1])
sql.set_blacklist_strength(chat_id, 7, str(args[1]))
else:
send_message(
update.effective_message,
"I only understand: off/del/warn/ban/kick/mute/tban/tmute!",
)
return ""
if conn:
text = "Changed blacklist mode: `{}` in *{}*!".format(
settypeblacklist, chat_name
)
else:
text = "Changed blacklist mode: `{}`!".format(settypeblacklist)
send_message(update.effective_message, text, parse_mode="markdown")
return (
"<b>{}:</b>\n"
"<b>Admin:</b> {}\n"
"Changed the blacklist mode. will {}.".format(
html.escape(chat.title),
mention_html(user.id, user.first_name),
settypeblacklist,
)
)
else:
getmode, getvalue = sql.get_blacklist_setting(chat.id)
if getmode == 0:
settypeblacklist = "do nothing"
elif getmode == 1:
settypeblacklist = "delete"
elif getmode == 2:
settypeblacklist = "warn"
elif getmode == 3:
settypeblacklist = "mute"
elif getmode == 4:
settypeblacklist = "kick"
elif getmode == 5:
settypeblacklist = "ban"
elif getmode == 6:
settypeblacklist = "temporarily ban for {}".format(getvalue)
elif getmode == 7:
settypeblacklist = "temporarily mute for {}".format(getvalue)
if conn:
text = "Current blacklistmode: *{}* in *{}*.".format(
settypeblacklist, chat_name
)
else:
text = "Current blacklistmode: *{}*.".format(settypeblacklist)
send_message(update.effective_message, text,
parse_mode=ParseMode.MARKDOWN)
return ""
def unblacklist(update, context):
msg = update.effective_message
chat = update.effective_chat
user = update.effective_user
words = msg.text.split(None, 1)
conn = connected(context.bot, update, chat, user.id)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
return
else:
chat_name = chat.title
if len(words) > 1:
text = words[1]
to_unblacklist = list(set(trigger.strip()
for trigger in text.split("\n") if trigger.strip()))
successful = 0
for trigger in to_unblacklist:
success = sql.rm_from_blacklist(chat_id, trigger.lower())
if success:
successful += 1
if len(to_unblacklist) == 1:
if successful:
send_message(
update.effective_message,
"Removed <code>{}</code> from blacklist in <b>{}</b>!".format(
html.escape(
to_unblacklist[0]),
chat_name),
parse_mode=ParseMode.HTML,
)
else:
send_message(
update.effective_message,
"This is not a blacklist trigger!")
elif successful == len(to_unblacklist):
send_message(
update.effective_message,
"Removed <code>{}</code> from blacklist in <b>{}</b>!".format(
successful, chat_name
),
parse_mode=ParseMode.HTML,
)
elif not successful:
send_message(
update.effective_message,
"None of these triggers exist so it can't be removed.".format(
successful, len(to_unblacklist) - successful
),
parse_mode=ParseMode.HTML,
)
else:
send_message(
update.effective_message,
"Removed <code>{}</code> from blacklist. {} did not exist, "
"so were not removed.".format(
successful, len(to_unblacklist) - successful
),
parse_mode=ParseMode.HTML,
)
else:
send_message(
update.effective_message,
"Tell me which words you would like to remove from blacklist!",
)
def unlock(update, context) -> str:
args = context.args
chat = update.effective_chat
user = update.effective_user
message = update.effective_message
if is_user_admin(chat, message.from_user.id):
if len(args) >= 1:
ltype = args[0].lower()
if ltype in LOCK_TYPES:
# Connection check
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_id = conn
chat_name = chat.title
text = "Unlocked {} messages for everyone in {}!".format(
ltype, chat_name
)
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command is meant to use in group not in PM",
)
return ""
chat = update.effective_chat
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
text = "Unlocked {} messages for everyone!".format(ltype)
sql.update_lock(chat.id, ltype, locked=False)
send_message(update.effective_message, text, parse_mode="markdown")
return (
"<b>{}:</b>"
"\n#UNLOCK"
"\n<b>Admin:</b> {}"
"\nUnlocked <code>{}</code>.".format(
html.escape(chat.title),
mention_html(user.id, user.first_name),
ltype,
)
)
elif ltype in UNLOCK_CHAT_RESTRICTION:
# Connection check
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_id = conn
chat_name = chat.title
text = "Unlocked {} for everyone in {}!".format(ltype, chat_name)
else:
if update.effective_message.chat.type == "private":
send_message(
update.effective_message,
"This command is meant to use in group not in PM",
)
return ""
chat = update.effective_chat
chat_id = update.effective_chat.id
chat_name = update.effective_message.chat.title
text = "Unlocked {} for everyone!".format(ltype)
current_permission = context.bot.getChat(chat_id).permissions
context.bot.set_chat_permissions(
chat_id=chat_id,
permissions=get_permission_list(
eval(str(current_permission)),
UNLOCK_CHAT_RESTRICTION[ltype.lower()],
),
)
send_message(update.effective_message, text, parse_mode="markdown")
return (
"<b>{}:</b>"
"\n#UNLOCK"
"\n<b>Admin:</b> {}"
"\nUnlocked <code>{}</code>.".format(
html.escape(chat.title),
mention_html(user.id, user.first_name),
ltype,
)
)
else:
send_message(
update.effective_message,
"What are you trying to unlock...? Try /locktypes for the list of lockables.",
)
else:
send_message(update.effective_message, "What are you trying to unlock...?")
return ""
