如何在WooCommerce订单完成后为指定用户角色分配新角色？

我来帮你搞定这个角色转换失效的问题，你的代码里有几个明显的语法和逻辑错误，导致角色没被正确更新：

原代码的问题点 #

• 用户对象获取错误：你用$user = $order->user_id拿到的只是用户ID（整数），不是WP_User实例，调用$user->set_role()会直接报错，根本执行不了
• 条件判断语法错误：后面的else if里少了in_array()函数，语法不合法，这部分代码无法被正确解析
• 未做用户存在校验：如果订单是访客下单（没有关联用户），直接操作会引发错误
• 不推荐直接访问属性：WooCommerce官方推荐用get_user_id()替代直接访问$order->user_id，保证版本兼容性

修复后的完整代码 #

add_action( 'woocommerce_order_status_completed', 'custom_role_on_purchase', 100, 2 );
function custom_role_on_purchase( $order_id, $order ) {
    // 获取用户ID（用官方推荐的getter方法）
    $user_id = $order->get_user_id();
    
    // 校验用户是否存在，跳过访客订单
    if ( ! $user_id ) {
        return;
    }
    
    // 获取WP_User实例
    $user = get_user_by( 'id', $user_id );
    if ( ! $user ) {
        return;
    }
    
    $user_roles = $user->roles;
    
    // 正确的角色判断与转换
    if ( in_array( 'Role A', $user_roles ) ) {
        $user->set_role( 'Role D' );
    } elseif ( in_array( 'Role B', $user_roles ) ) {
        $user->set_role( 'Role E' );
    } elseif ( in_array( 'Role C', $user_roles ) ) {
        $user->set_role( 'Role F' );
    }
}

额外说明 #

1. 我把钩子的参数改成了2个（$order_id, $order），因为WooCommerce的woocommerce_order_status_completed钩子本来就会传递这两个参数，直接用现成的$order对象更高效
2. 增加了用户存在的双重校验，避免无用户时的错误
3. 用get_user_by()获取WP_User实例，确保能调用set_role()方法
4. 修正了elseif的条件判断语法，补上了in_array()函数

把这段代码替换你原来的代码，测试完成订单后，角色应该就能正常转换了。

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