基于自定义字段为Dealer角色隐藏WooCommerce产品变体失效排查

问题概述 #

已在WordPress中创建「Dealer」和「Wholesaler」用户角色，希望通过为产品变体添加自定义复选框，控制是否对Dealer角色隐藏该变体。编写代码后变体未被隐藏，需排查修复。

原代码问题分析 #

1. 变体自定义字段ID未带循环索引：在添加变体复选框时，字段ID未包含$loop变量，导致所有变体的表单字段名称重复，保存时无法正确关联到对应变体。
2. 角色添加逻辑重复执行：add_custom_roles函数每次init钩子触发时都会重新添加角色，可能引发角色权限异常。
3. 过滤器优先级可能不足：woocommerce_variation_is_visible过滤器使用默认优先级10，可能被其他逻辑覆盖。

修复后的完整代码 #

// 添加自定义用户角色- Wholesaler 和 Dealer，避免重复添加
function add_custom_roles() {
    if ( ! get_role( 'wholesaler' ) ) {
        add_role(
            'wholesaler',
            __( 'Wholesaler' ),
            get_role( 'customer' )->capabilities
        );
    }

    if ( ! get_role( 'dealer' ) ) {
        add_role(
            'dealer',
            __( 'Dealer' ),
            get_role( 'customer' )->capabilities
        );
    }
}
add_action( 'init', 'add_custom_roles' );

// 为产品变体添加自定义复选框字段
add_action( 'woocommerce_variation_options', 'add_custom_fields_to_variations', 10, 3 );
function add_custom_fields_to_variations( $loop, $variation_data, $variation ) {
    woocommerce_wp_checkbox(
        array(
            'id'            => '_hide_variation_dealer[' . $loop . ']', // 添加$loop索引确保字段唯一
            'wrapper_class' => 'show_if_variation',
            'label'         => __('Hide for dealers', 'woocommerce' ),
            'description'   => '',
            'value'         => get_post_meta( $variation->ID, '_hide_variation_dealer', true ) ? 'yes' : 'no',
        )
    );
}

// 保存产品变体的自定义字段值
add_action( 'woocommerce_save_product_variation', 'save_custom_fields_for_variations', 10, 2 );
function save_custom_fields_for_variations( $variation_id, $i ) {
    $hide_variation_dealer = isset( $_POST['_hide_variation_dealer'][$i] ) ? 'yes' : 'no';
    update_post_meta( $variation_id, '_hide_variation_dealer', $hide_variation_dealer );
}

// 过滤变体可见性，针对Dealer角色隐藏指定变体
add_filter( 'woocommerce_variation_is_visible', 'hide_specific_product_variation', 20, 4 ); // 提高优先级确保生效
function hide_specific_product_variation( $is_visible, $variation_id, $variable_product, $variation ) {
    // 检查用户是否登录且为Dealer角色
    if( is_user_logged_in() ) {
        $user = wp_get_current_user();
        if ( in_array( 'dealer', (array) $user->roles ) ) {
            // 读取变体的隐藏设置，若为yes则返回false隐藏
            $hide_variation_dealer = get_post_meta( $variation_id, '_hide_variation_dealer', true );
            if ( $hide_variation_dealer === 'yes' ) {
                return false;
            }
        }
    }
    return $is_visible;
}

额外注意事项 #

• 保存代码后，需重新编辑并更新目标可变产品，确保变体的复选框设置被正确保存。
• 若仍未生效，可清除WooCommerce缓存和浏览器缓存，避免旧数据影响。
• 测试时需使用Dealer角色的用户账号登录查看效果，确保角色判断逻辑正确。

内容的提问来源于stack exchange，提问作者SavPhill