def connected(bot, update, chat, user_id, need_admin=True):
user = update.effective_user
msg = update.effective_message
spam = spamfilters(msg.text, msg.from_user.id, update.effective_chat.id)
if spam is True:
return
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(update: Update, _):
msg = update.effective_message
spam = spamfilters(msg.text, msg.from_user.id, update.effective_chat.id)
if spam is True:
return
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
spam = spamfilters(msg.text, msg.from_user.id, chat.id)
if spam is True:
return
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(update: Update, context: CallbackContext):
chat = update.effective_chat
user = update.effective_user
msg = update.effective_message
spam = spamfilters(msg.text, msg.from_user.id, update.effective_chat.id)
if spam is True:
return
conn = connected(context.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 connect_chat(update: Update, context: CallbackContext):
chat = update.effective_chat
user = update.effective_user
msg = update.effective_message
spam = spamfilters(msg.text, msg.from_user.id, chat.id)
if spam is True:
return
if chat.type == 'private':
if len(args) >= 1:
args = context.args
try:
connect_chat = int(args[0])
getstatusadmin = context.bot.get_chat_member(
connect_chat, msg.from_user.id)
except ValueError:
try:
connect_chat = str(args[0])
get_chat = context.bot.getChat(connect_chat)
connect_chat = get_chat.id
getstatusadmin = context.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(context.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(context.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 = context.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)
context.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!")
