def connected(bot, update, chat, user_id, need_admin=True):
user = update.effective_user
msg = update.effective_message
if chat.type == chat.PRIVATE and sql.get_connected_chat(user_id):
conn_id = sql.get_connected_chat(user_id).chat_id
getstatusadmin = bot.get_chat_member(conn_id, msg.from_user.id)
isadmin = getstatusadmin.status in ADMIN_STATUS
ismember = getstatusadmin.status in MEMBER_STAUS
isallow = sql.allow_connect_to_chat(conn_id)
if isadmin or (isallow and ismember) or (user.id in SUDO_USERS) or (user.id in DEV_USERS):
if need_admin is True:
if getstatusadmin.status in ADMIN_STATUS or user_id in SUDO_USERS or user.id in DEV_USERS:
return conn_id
else:
send_message(msg, "You must be an admin in the connected group!")
raise Exception("Not admin!")
else:
return conn_id
else:
send_message(msg, "The group changed the connection rights or you are no longer an admin.\n"
"I've disconnected you.")
disconnect_chat(bot, update)
raise Exception("Not admin!")
else:
return False
def help_connect_chat(bot: Bot, update: Update):
msg = update.effective_message
if msg.chat.type != "private":
send_message(msg, "PM me with that command to get help.")
return
else:
send_message(msg, "All commands", parse_mode="markdown")
def disconnect_chat(bot: Bot, update: Update):
chat = update.effective_chat
msg = update.effective_message
if chat.type == 'private':
disconnection_status = sql.disconnect(msg.from_user.id)
if disconnection_status:
sql.disconnected_chat = send_message(msg, "Disconnected from chat!")
else:
send_message(msg, "You're not connected!")
else:
send_message(msg, "This command is only available in PM.")
def connection_chat(bot: Bot, update: Update):
chat = update.effective_chat
user = update.effective_user
msg = update.effective_message
conn = connected(bot, update, chat, user.id, need_admin=True)
if conn:
chat_name = dispatcher.bot.getChat(conn).title
else:
if msg.chat.type != "private":
return
chat_name = chat.title
if conn:
message = "You are currently connected with {}.\n".format(chat_name)
else:
message = "You are currently not connected in any group.\n"
send_message(msg, message, parse_mode="markdown")
def blackliststicker(bot: Bot, update: Update, args: List[str]):
msg = update.effective_message # type: Optional[Message]
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
conn = connected(bot, update, chat, user.id, need_admin=False)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
if chat.type == "private":
return
else:
chat_id = update.effective_chat.id
chat_name = chat.title
sticker_list = "<b>List black stickers currently in {}:</b>\n".format(
chat_name)
all_stickerlist = sql.get_chat_stickers(chat_id)
if len(args) > 0 and args[0].lower() == 'copy':
for trigger in all_stickerlist:
sticker_list += "<code>{}</code>\n".format(html.escape(trigger))
elif len(args) == 0:
for trigger in all_stickerlist:
sticker_list += " - <code>{}</code>\n".format(html.escape(trigger))
split_text = split_message(sticker_list)
for text in split_text:
if sticker_list == "<b>List black stickers currently in {}:</b>\n".format(
chat_name).format(chat_name):
send_message(update.effective_message,
"There is no blacklist sticker in <b>{}</b>!".format(
chat_name),
parse_mode=ParseMode.HTML)
return
send_message(update.effective_message, text, parse_mode=ParseMode.HTML)
def add_blackliststicker(bot: Bot, update: Update):
msg = update.effective_message # type: Optional[Message]
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
words = msg.text.split(None, 1)
conn = connected(bot, update, chat, user.id)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
return
else:
chat_name = chat.title
if len(words) > 1:
text = words[1].replace('https://t.me/addstickers/', '')
to_blacklist = list(
set(trigger.strip() for trigger in text.split("\n")
if trigger.strip()))
added = 0
for trigger in to_blacklist:
try:
get = bot.getStickerSet(trigger)
sql.add_to_stickers(chat_id, trigger.lower())
added += 1
except BadRequest:
send_message(update.effective_message,
"Sticker `{}` can not be found!".format(trigger),
parse_mode="markdown")
if added == 0:
return
if len(to_blacklist) == 1:
send_message(
update.effective_message,
"Sticker <code>{}</code> added to blacklist stickers in <b>{}</b>!"
.format(html.escape(to_blacklist[0]), chat_name),
parse_mode=ParseMode.HTML)
else:
send_message(
update.effective_message,
"<code>{}</code> stickers added to blacklist sticker in <b>{}</b>!"
.format(added, chat_name),
parse_mode=ParseMode.HTML)
elif msg.reply_to_message:
added = 0
trigger = msg.reply_to_message.sticker.set_name
if trigger == None:
send_message(update.effective_message, "Sticker is invalid!")
return
try:
get = bot.getStickerSet(trigger)
sql.add_to_stickers(chat_id, trigger.lower())
added += 1
except BadRequest:
send_message(update.effective_message,
"Sticker `{}` can not be found!".format(trigger),
parse_mode="markdown")
if added == 0:
return
send_message(
update.effective_message,
"Sticker <code>{}</code> added to blacklist stickers in <b>{}</b>!"
.format(trigger, chat_name),
parse_mode=ParseMode.HTML)
else:
send_message(
update.effective_message,
"Tell me what stickers you want to add to the blacklist stickers.")
def blacklist_mode(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
msg = update.effective_message # type: Optional[Message]
conn = connected(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,
"You can do this command in groups, not 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 = 'turn off'
sql.set_blacklist_strength(chat_id, 0, "0")
elif args[0].lower() == 'del' or args[0].lower() == 'delete':
settypeblacklist = 'left, the message will be deleted'
sql.set_blacklist_strength(chat_id, 1, "0")
elif args[0].lower() == 'warn':
settypeblacklist = 'warned'
sql.set_blacklist_strength(chat_id, 2, "0")
elif args[0].lower() == 'mute':
settypeblacklist = 'muted'
sql.set_blacklist_strength(chat_id, 3, "0")
elif args[0].lower() == 'kick':
settypeblacklist = 'kicked'
sql.set_blacklist_strength(chat_id, 4, "0")
elif args[0].lower() == 'ban':
settypeblacklist = 'banned'
sql.set_blacklist_strength(chat_id, 5, "0")
elif args[0].lower() == 'tban':
if len(args) == 1:
teks = """It looks like you are trying to set a temporary value to blacklist, but has not determined the time; use `/blstickermode tban <timevalue>`.
Examples of time values: 4m = 4 minute, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks."""
send_message(update.effective_message,
teks,
parse_mode="markdown")
return
settypeblacklist = 'temporary banned 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 are trying to set a temporary value to blacklist, but has not determined the time; use `/blstickermode tmute <timevalue>`.
Examples of time values: 4m = 4 minute, 3h = 3 hours, 6d = 6 days, 5w = 5 weeks."""
send_message(update.effective_message,
teks,
parse_mode="markdown")
return
settypeblacklist = tl(update.effective_message,
'temporary muted 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 = "Blacklist sticker mode changed, will `{}` at *{}*!".format(
settypeblacklist, chat_name)
else:
text = "Blacklist sticker mode changed, will `{}`!".format(
settypeblacklist)
send_message(update.effective_message, text, parse_mode="markdown")
return "<b>{}:</b>\n" \
"<b>Admin:</b> {}\n" \
"Changed sticker 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 = "not active"
elif getmode == 1:
settypeblacklist = "hapus"
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 banned for {}".format(getvalue)
elif getmode == 7:
settypeblacklist = "temporarily muted for {}".format(getvalue)
if conn:
text = "Blacklist sticker mode is currently set to *{}* in *{}*.".format(
settypeblacklist, chat_name)
else:
text = "Blacklist sticker mode is currently set to *{}*.".format(
settypeblacklist)
send_message(update.effective_message,
text,
parse_mode=ParseMode.MARKDOWN)
return ""
def unblackliststicker(bot: Bot, update: Update):
msg = update.effective_message # type: Optional[Message]
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
words = msg.text.split(None, 1)
conn = connected(bot, update, chat, user.id)
if conn:
chat_id = conn
chat_name = dispatcher.bot.getChat(conn).title
else:
chat_id = update.effective_chat.id
if chat.type == "private":
return
else:
chat_name = chat.title
if len(words) > 1:
text = words[1].replace('https://t.me/addstickers/', '')
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_stickers(chat_id, trigger.lower())
if success:
successful += 1
if len(to_unblacklist) == 1:
if successful:
send_message(
update.effective_message,
"Sticker <code>{}</code> deleted 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 on the blacklist stickers...!")
elif successful == len(to_unblacklist):
send_message(
update.effective_message,
"Sticker <code>{}</code> deleted from blacklist in <b>{}</b>!".
format(successful, chat_name),
parse_mode=ParseMode.HTML)
elif not successful:
send_message(
update.effective_message,
"None of these stickers exist, so they cannot be removed.".
format(successful,
len(to_unblacklist) - successful),
parse_mode=ParseMode.HTML)
else:
send_message(
update.effective_message,
"Sticker <code>{}</code> deleted from blacklist. {} did not exist, so it's not deleted."
.format(successful,
len(to_unblacklist) - successful),
parse_mode=ParseMode.HTML)
elif msg.reply_to_message:
trigger = msg.reply_to_message.sticker.set_name
if trigger == None:
send_message(update.effective_message, "Sticker is invalid!")
return
success = sql.rm_from_stickers(chat_id, trigger.lower())
if success:
send_message(
update.effective_message,
"Sticker <code>{}</code> deleted from blacklist in <b>{}</b>!".
format(trigger, chat_name),
parse_mode=ParseMode.HTML)
else:
send_message(
update.effective_message,
"{} not found on blacklisted stickers...!".format(trigger))
else:
send_message(
update.effective_message,
"Tell me what stickers you want to add to the blacklist stickers.")
def connect_chat(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
user = update.effective_user
msg = update.effective_message
if chat.type == 'private':
if len(args) >= 1:
try:
connect_chat = int(args[0])
getstatusadmin = bot.get_chat_member(connect_chat, msg.from_user.id)
except ValueError:
try:
connect_chat = str(args[0])
get_chat = bot.getChat(connect_chat)
connect_chat = get_chat.id
getstatusadmin = bot.get_chat_member(connect_chat, msg.from_user.id)
except BadRequest:
send_message(msg, "Invalid Chat ID!")
return
except BadRequest:
send_message(msg, "Invalid Chat ID!")
return
isadmin = getstatusadmin.status in ADMIN_STATUS
ismember = getstatusadmin.status in MEMBER_STAUS
isallow = sql.allow_connect_to_chat(connect_chat)
if isadmin or (isallow and ismember) or (user.id in SUDO_USERS) or (user.id in DEV_USERS):
connection_status = sql.connect(msg.from_user.id, connect_chat)
if connection_status:
conn_chat = dispatcher.bot.getChat(connected(bot, update, chat, user.id, need_admin=False))
chat_name = conn_chat.title
send_message(msg, "Successfully connected to *{chat_name}*."
" Use /connection for see current available commands.",
parse_mode=ParseMode.MARKDOWN)
sql.add_history_conn(user.id, str(conn_chat.id), chat_name)
else:
send_message(msg, "Connection failed!")
else:
send_message(msg, "Connection to this chat is not allowed!")
else:
gethistory = sql.get_history_conn(user.id)
if gethistory:
buttons = [
InlineKeyboardButton(text="❎ Close button", callback_data="connect_close"),
InlineKeyboardButton(text="🧹 Clear history", callback_data="connect_clear")
]
else:
buttons = []
conn = connected(bot, update, chat, user.id, need_admin=False)
if conn:
connectedchat = dispatcher.bot.getChat(conn)
text = "You are connected to *{}* (`{}`)".format(connectedchat.title, conn)
buttons.append(InlineKeyboardButton(text="🔌 Disconnect", callback_data="connect_disconnect"))
else:
text = "Write the chat ID or tag to connect!"
if gethistory:
text += "\n\n*Connection history:*\n"
text += "╒═══「 *Info* 」\n"
text += "│ Sorted: `Newest`\n"
text += "│\n"
buttons = [buttons]
for x in sorted(gethistory.keys(), reverse=True):
htime = time.strftime("%d/%m/%Y", time.localtime(x))
text += "╞═「 *{}* 」\n│ `{}`\n│ `{}`\n".format(gethistory[x]['chat_name'],
gethistory[x]['chat_id'], htime)
text += "│\n"
buttons.append([InlineKeyboardButton(text=gethistory[x]['chat_name'],
callback_data="connect({})".format(gethistory[x]['chat_id']))])
text += "╘══「 Total {} Chats 」".format(
str(len(gethistory)) + " (max)" if len(gethistory) == 5 else str(len(gethistory)))
conn_hist = InlineKeyboardMarkup(buttons)
elif buttons:
conn_hist = InlineKeyboardMarkup([buttons])
else:
conn_hist = None
send_message(msg, text, parse_mode="markdown", reply_markup=conn_hist)
else:
getstatusadmin = bot.get_chat_member(chat.id, msg.from_user.id)
isadmin = getstatusadmin.status in ADMIN_STATUS
ismember = getstatusadmin.status in MEMBER_STAUS
isallow = sql.allow_connect_to_chat(chat.id)
if isadmin or (isallow and ismember) or (user.id in SUDO_USERS) or (user.id in DEV_USERS):
connection_status = sql.connect(msg.from_user.id, chat.id)
if connection_status:
chat_name = dispatcher.bot.getChat(chat.id).title
send_message(msg, "Successfully connected to *{}*.".format(chat_name),
parse_mode=ParseMode.MARKDOWN)
try:
sql.add_history_conn(user.id, str(chat.id), chat_name)
bot.send_message(msg.from_user.id, f"You have connected with *{chat_name}*."
f" Use /connection for see current available commands.",
parse_mode="markdown")
except BadRequest:
pass
except Unauthorized:
pass
else:
send_message(msg, "Connection failed!")
else:
send_message(msg, "Connection to this chat is not allowed!")
def allow_connections(bot: Bot, update: Update, args: List[str]):
chat = update.effective_chat
if chat.type != chat.PRIVATE:
if len(args) >= 1:
var = args[0]
if var == "no":
sql.set_allow_connect_to_chat(chat.id, False)
send_message(update.effective_message, "Connection has been disabled for this chat")
elif var == "yes":
sql.set_allow_connect_to_chat(chat.id, True)
send_message(update.effective_message, "Connection has been enabled for this chat")
else:
send_message(update.effective_message, "Please enter `yes` or `no`!", parse_mode=ParseMode.MARKDOWN)
else:
get_settings = sql.allow_connect_to_chat(chat.id)
if get_settings:
send_message(update.effective_message, "Connections to this group are *Allowed* for members!",
parse_mode=ParseMode.MARKDOWN)
else:
send_message(update.effective_message, "Connection to this group are *Not Allowed* for members!",
parse_mode=ParseMode.MARKDOWN)
else:
send_message(update.effective_message, "This command is for group only. Not in PM!")