def allow_connections(bot: Bot, update: Update, args: List[str]): chat = update.effective_chat 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_button(bot: Bot, update: Update): 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 = 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) 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(bot, update, chat, user.id, need_admin=False)) chat_name = conn_chat.title query.message.edit_text( "Successfully connected to *{}*. Use /connection for see current 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: 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: 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(bot, update, [])
def connected(bot, update, chat, user_id, need_admin=True): user = update.effective_user spam = spamfilters(update.effective_message.text, update.effective_message.from_user.id, update.effective_chat.id) if spam == 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, 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, "You must be an admin in the connected group!") raise Exception("Not admin!") 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(bot, update) raise Exception("Not admin!") else: return False