def connect_button(update: Update, context: CallbackContext):
query = update.callback_query
chat = update.effective_chat
user = update.effective_user
connect_match = re.match(r"connect\((.+?)\)", query.data)
disconnect_match = query.data == "connect_disconnect"
clear_match = query.data == "connect_clear"
connect_close = query.data == "connect_close"
if connect_match:
target_chat = connect_match.group(1)
getstatusadmin = context.bot.get_chat_member(target_chat,
query.from_user.id)
isadmin = getstatusadmin.status in ADMIN_STATUS
ismember = getstatusadmin.status in MEMBER_STAUS
isallow = sql.allow_connect_to_chat(target_chat)
if (isadmin or (isallow and ismember) or (user.id in SUDO_USERS)
or (user.id in DEV_USERS)):
connection_status = sql.connect(query.from_user.id, target_chat)
if connection_status:
conn_chat = dispatcher.bot.getChat(
connected(context.bot,
update,
chat,
user.id,
need_admin=False))
chat_name = conn_chat.title
query.message.edit_text(
f"Successfully connected to *{chat_name}*."
f" Use /connection for see current available commands.",
parse_mode=ParseMode.MARKDOWN,
)
sql.add_history_conn(user.id, str(conn_chat.id), chat_name)
else:
query.message.edit_text("Connection failed!")
else:
context.bot.answer_callback_query(
query.id,
"Connection to this chat is not allowed!",
show_alert=True)
elif disconnect_match:
disconnection_status = sql.disconnect(query.from_user.id)
if disconnection_status:
sql.disconnected_chat = query.message.edit_text(
"Disconnected from chat!")
else:
context.bot.answer_callback_query(query.id,
"You're not connected!",
show_alert=True)
elif clear_match:
sql.clear_history_conn(query.from_user.id)
query.message.edit_text("History connected has been cleared!")
elif connect_close:
query.message.edit_text("Closed.\nTo open again, type /connect")
else:
connect_chat(context.bot, update)
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.")