def connected(bot, update, chat, user_id, need_admin=True):
chat = update.effective_chat # type: Optional[Chat]
if chat.type == chat.PRIVATE and sql.get_connected_chat(user_id):
conn_id = sql.get_connected_chat(user_id).chat_id
if (bot.get_chat_member(
conn_id, user_id).status in ('administrator', 'creator') or
(sql.allow_connect_to_chat(connect_chat) == True)
and bot.get_chat_member(
user_id, update.effective_message.from_user.id).status in
('member')) or (user_id in SUDO_USERS):
if need_admin:
if bot.get_chat_member(
conn_id,
update.effective_message.from_user.id).status in (
'administrator',
'creator') or user_id in SUDO_USERS:
return conn_id
else:
update.effective_message.reply_text(
tld(chat.id, "connection_err_no_admin"))
return
else:
return conn_id
else:
update.effective_message.reply_text(
tld(chat.id, "connection_err_unknown"))
disconnect_chat(bot, update)
return
else:
return False
def connected(bot, update, chat, user_id, need_admin=True):
if chat.type == chat.PRIVATE and sql.get_connected_chat(user_id):
conn_id = sql.get_connected_chat(user_id).chat_id
if (bot.get_chat_member(
conn_id, user_id).status in ('administrator', 'creator') or
(sql.allow_connect_to_chat(connect_chat) == True)
and bot.get_chat_member(
user_id, update.effective_message.from_user.id).status in
('member')) or (user_id in SUDO_USERS):
if need_admin:
if bot.get_chat_member(
conn_id,
update.effective_message.from_user.id).status in (
'administrator',
'creator') or user_id in SUDO_USERS:
return conn_id
else:
update.effective_message.reply_text(
"You need to be a admin in a connected group!")
exit(1)
else:
return conn_id
else:
update.effective_message.reply_text(
"Group changed rights connection or you are not admin anymore.\nI'll disconnect you."
)
disconnect_chat(bot, update)
exit(1)
else:
return False
def connect_chat(bot, update, args):
chat = update.effective_chat # type: Optional[Chat]
user = update.effective_user # type: Optional[User]
if update.effective_chat.type == 'private':
if len(args) >= 1:
try:
connect_chat = int(args[0])
except ValueError:
update.effective_message.reply_text(
tld(chat.id, "common_err_invalid_chatid"))
return
if (bot.get_chat_member(
connect_chat, update.effective_message.from_user.id).status
in ('administrator', 'creator') or
(sql.allow_connect_to_chat(connect_chat) == True)
and bot.get_chat_member(
connect_chat,
update.effective_message.from_user.id).status in
('member')) or (user.id in SUDO_USERS):
connection_status = sql.connect(
update.effective_message.from_user.id, connect_chat)
if connection_status:
chat_name = dispatcher.bot.getChat(
connected(bot, update, chat, user.id,
need_admin=False)).title
update.effective_message.reply_text(
tld(chat.id, "connection_success").format(chat_name),
parse_mode=ParseMode.MARKDOWN)
#Add chat to connection history
history = sql.get_history(user.id)
if history:
#Vars
if history.chat_id1:
history1 = int(history.chat_id1)
if history.chat_id2:
history2 = int(history.chat_id2)
if history.chat_id3:
history3 = int(history.chat_id3)
if history.updated:
number = history.updated
if number == 1 and connect_chat != history2 and connect_chat != history3:
history1 = connect_chat
number = 2
elif number == 2 and connect_chat != history1 and connect_chat != history3:
history2 = connect_chat
number = 3
elif number >= 3 and connect_chat != history2 and connect_chat != history1:
history3 = connect_chat
number = 1
else:
print("Error")
print(history.updated)
print(number)
sql.add_history(user.id, history1, history2, history3,
number)
# print(history.user_id, history.chat_id1, history.chat_id2, history.chat_id3, history.updated)
else:
sql.add_history(user.id, connect_chat, "0", "0", 2)
# Rebuild user's keyboard
keyboard(bot, update)
else:
update.effective_message.reply_text(
tld(chat.id, "connection_fail"))
else:
update.effective_message.reply_text(
tld(chat.id, "connection_err_not_allowed"))
else:
update.effective_message.reply_text(
tld(chat.id, "connection_err_no_chatid"))
history = sql.get_history(user.id)
# print(history.user_id, history.chat_id1, history.chat_id2, history.chat_id3, history.updated)
elif update.effective_chat.type == 'supergroup':
connect_chat = chat.id
if (bot.get_chat_member(connect_chat,
update.effective_message.from_user.id).status
in ('administrator', 'creator') or
(sql.allow_connect_to_chat(connect_chat) == True)
and bot.get_chat_member(
connect_chat, update.effective_message.from_user.id).status
in 'member') or (user.id in SUDO_USERS):
connection_status = sql.connect(
update.effective_message.from_user.id, connect_chat)
if connection_status:
update.effective_message.reply_text(
tld(chat.id, "connection_success").format(chat.id),
parse_mode=ParseMode.MARKDOWN)
else:
update.effective_message.reply_text(
tld(chat.id, "connection_fail"),
parse_mode=ParseMode.MARKDOWN)
else:
update.effective_message.reply_text(
tld(chat.id, "common_err_no_admin"))
else:
update.effective_message.reply_text(tld(chat.id, "common_cmd_pm_only"))
