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 DRAGONS) or (user.id in DEV_USERS)): if need_admin is True: if (getstatusadmin.status in ("administrator", "creator") or user_id in DRAGONS or user.id in DEV_USERS): return conn_id else: send_message( update.effective_message, "You must be an admin in the connected group!", ) else: return conn_id else: send_message( update.effective_message, "The group changed the connection rights or you are no longer an admin.\nI've disconnected you.", ) 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 DRAGONS): 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( "Successfully connected to *{}*. \nUse `/helpconnect` to check available commands." .format(chat_name), 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(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, "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!" )
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, "Invalid Chat ID!") return except BadRequest: send_message(update.effective_message, "Invalid Chat ID!") 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 DRAGONS): 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, "Successfully connected to *{}*. \nUse /helpconnect to check available commands.".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, "Connection failed!") else: send_message( update.effective_message, "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 currently 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( 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 DRAGONS): 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, "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( update.effective_message.from_user.id, "You are connected to *{}*. \nUse `/helpconnect` to check available commands.".format( chat_name ), parse_mode="markdown", ) except BadRequest: pass except Unauthorized: pass else: send_message(update.effective_message, "Connection failed!") else: send_message( update.effective_message, "Connection to this chat is not allowed!" )