php - 根据用户角色为 WooCommerce 订单号添加前缀

Question

我正在尝试根据用户角色更改 WooCommerce 订单前缀，目前，此功能完美无缺：

add_filter( 'woocommerce_order_number', 'change_woocommerce_order_number' );

function change_woocommerce_order_number( $order_id ) {
    $prefix = 'WHLS';
    $new_order_id = $prefix . $order_id;
    return $new_order_id;
}

现在，我正在尝试通过以下方式根据用户角色更改订单前缀：Export/Distributor Gold - GLD-### ( new ) Export/Distributor Silver - SLVR-### ( new ) Wholesale - WHLS- ###（当前功能）

我尝试的代码：（此功能代码不为订单号添加任何前缀）。

add_filter( 'woocommerce_order_number', 'change_woocommerce_order_number' );

function change_woocommerce_order_number( $order_id ) {
    if ( current_user_can( 'wholeseller' ))
    $prefix = 'WHLS';
    $new_order_id = $prefix . $order_id;
    return $new_order_id;
}
   
    if ( current_user_can( 'distributor_gold' )) {
    $prefix = 'GLD';
    $new_order_id = $prefix . $order_id;
    return $new_order_id;
    }
    

    elseif ( current_user_can( 'distributor_silver' )) {
    $prefix = 'SLVR';
    $new_order_id = $prefix . $order_id;
    return $new_order_id;
    }

任何人都可以指出我正确的方向吗？

标签： phpwordpresswoocommerceordersuser-roles

Answer 1

用于$order->get_user()获取与订单关联的用户。

然后$user->roles，获取该用户拥有的角色。

所以你得到：

function filter_woocommerce_order_number( $order_number, $order ) {
    // Get user
    $user = $order->get_user();
    
    // Roles
    $roles = (array) $user->roles;
    
    // Compare
    if ( in_array ( 'wholeseller', $roles ) ) {
        $prefix = 'WHLS';
    } elseif ( in_array ( 'distributor_gold', $roles ) ) {
        $prefix = 'GLD';
    } elseif ( in_array ( 'distributor_silver', $roles ) ) {
        $prefix = 'SLVR';
    } elseif ( in_array ( 'administrator', $roles ) ) {
        $prefix = 'ADMIN';
    } else {
        $prefix = '';
    }
    
    return $prefix . $order_number;
}
add_filter( 'woocommerce_order_number', 'filter_woocommerce_order_number', 10, 2 );