def connected(bot: Bot, update: Update, chat, user_id, need_admin=True):
user = update.effective_user
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, update.effective_message.from_user.id)
isadmin = getstatusadmin.status in ("administrator", "creator")
ismember = getstatusadmin.status in ("member")
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 == True:
if (getstatusadmin.status in ("administrator", "creator")
or user_id in SUDO_USERS or user.id in DEV_USERS):
return conn_id
else:
send_message(
update.effective_message,
"Debe ser un administrador en el grupo conectado!",
)
else:
return conn_id
else:
send_message(
update.effective_message,
"El grupo cambió los derechos de conexión o ya no eres administrador.\nTe desconecté.",
)
disconnect_chat(update, bot)
else:
return False
def connect_button(update, context):
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 ("administrator", "creator")
ismember = getstatusadmin.status in ("member")
isallow = sql.allow_connect_to_chat(target_chat)
if (isadmin) or (isallow and ismember) or (user.id in SUDO_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(
"Conectado exitosamente a *{}*.\nUsa`/helpconnect` para ver los comandos disponibles."
.format(chat_name),
parse_mode=ParseMode.MARKDOWN,
)
sql.add_history_conn(user.id, str(conn_chat.id), chat_name)
else:
query.message.edit_text("La conexión falló!")
else:
context.bot.answer_callback_query(
query.id,
"No se permite la conexión a este chat!",
show_alert=True)
elif disconnect_match:
disconnection_status = sql.disconnect(query.from_user.id)
if disconnection_status:
sql.disconnected_chat = query.message.edit_text(
"Desconectado del chat!")
else:
context.bot.answer_callback_query(query.id,
"No estas conectado!",
show_alert=True)
elif clear_match:
sql.clear_history_conn(query.from_user.id)
query.message.edit_text("Se borró el historial de conexión!")
elif connect_close:
query.message.edit_text(
"Cerrado.\nPara abrir de nuevo, escriba `/connect`")
else:
connect_chat(update, context)
def allow_connections(update, context) -> str:
chat = update.effective_chat
args = context.args
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,
"La conexión se ha inhabilitado para este chat",
)
elif var == "si":
sql.set_allow_connect_to_chat(chat.id, True)
send_message(
update.effective_message,
"Se ha habilitado la conexión para este chat",
)
else:
send_message(
update.effective_message,
"Por favor escribe `si` o `no`!",
parse_mode=ParseMode.MARKDOWN,
)
else:
get_settings = sql.allow_connect_to_chat(chat.id)
if get_settings:
send_message(
update.effective_message,
"La conexion a este grupo esta *permitida* para los miembros!",
parse_mode=ParseMode.MARKDOWN,
)
else:
send_message(
update.effective_message,
"La conexión a este grupo *no está permitida* para los miembros!",
parse_mode=ParseMode.MARKDOWN,
)
else:
send_message(update.effective_message,
"Este comando es solo para grupos. No en privado!")
def connect_chat(update, context):
chat = update.effective_chat
user = update.effective_user
args = context.args
if update.effective_chat.type == "private":
if args and len(args) >= 1:
try:
connect_chat = int(args[0])
getstatusadmin = context.bot.get_chat_member(
connect_chat, update.effective_message.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, update.effective_message.from_user.id)
except BadRequest:
send_message(update.effective_message,
"ID de chat no válido!")
return
except BadRequest:
send_message(update.effective_message, "ID de chat no válido!")
return
isadmin = getstatusadmin.status in ("administrator", "creator")
ismember = getstatusadmin.status in ("member")
isallow = sql.allow_connect_to_chat(connect_chat)
if (isadmin) or (isallow and ismember) or (user.id in SUDO_USERS):
connection_status = sql.connect(
update.effective_message.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(
update.effective_message,
"Conectado exitosamente a *{}*. \nUsa /helpconnect para ver los comandos disponibles."
.format(chat_name),
parse_mode=ParseMode.MARKDOWN,
)
sql.add_history_conn(user.id, str(conn_chat.id), chat_name)
else:
send_message(update.effective_message,
"La conexión falló!")
else:
send_message(update.effective_message,
"No se permite la conexión a este chat!")
else:
gethistory = sql.get_history_conn(user.id)
if gethistory:
buttons = [
InlineKeyboardButton(text="❎ Cerrar",
callback_data="connect_close"),
InlineKeyboardButton(text="🧹 Limpiar historial",
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 = "Actualmente estás conectado a *{}* (`{}`)".format(
connectedchat.title, conn)
buttons.append(
InlineKeyboardButton(text="🔌 Desconectar",
callback_data="connect_disconnect"))
else:
text = "Escriba el ID de chat o la etiqueta para conectarse!"
if gethistory:
text += "\n\n*Historial de conexiones:*\n"
text += "╒═══「 *Info* 」\n"
text += "│ Ordenado: 'Más nuevo'\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(
update.effective_message,
text,
parse_mode="markdown",
reply_markup=conn_hist,
)
else:
getstatusadmin = context.bot.get_chat_member(
chat.id, update.effective_message.from_user.id)
isadmin = getstatusadmin.status in ("administrator", "creator")
ismember = getstatusadmin.status in ("member")
isallow = sql.allow_connect_to_chat(chat.id)
if (isadmin) or (isallow and ismember) or (user.id in SUDO_USERS):
connection_status = sql.connect(
update.effective_message.from_user.id, chat.id)
if connection_status:
chat_name = dispatcher.bot.getChat(chat.id).title
send_message(
update.effective_message,
"Conectado exitosamente a *{}*.".format(chat_name),
parse_mode=ParseMode.MARKDOWN,
)
try:
sql.add_history_conn(user.id, str(chat.id), chat_name)
context.bot.send_message(
update.effective_message.from_user.id,
"Estas conectado a *{}*. \nUsa `/helpconnect` para ver los comandos disponibles."
.format(chat_name),
parse_mode="markdown",
)
except BadRequest:
pass
except Unauthorized:
pass
else:
send_message(update.effective_message, "La conexión falló!")
else:
send_message(update.effective_message,
"No se permite la conexión a este chat!")