Автоматическая смена ролей после оплаты

[Андрей K]

Всем доброго времени суток! Подскажите пожалуйста как решить задачу, ниже код, работает, после оплаты если товар имеет id 151, то удаляется старая роль: customer и записывается новая роль: um_custom_role_6. Вопрос, как правильно добавить еще одно условие, что если товар имеет id 203, то удаляется старая роль: customer и записывается новая роль: um_custom_role_4 ?

`
add_action( 'woocommerce_order_status_completed', 'change_role_on_purchase' );
function change_role_on_purchase( $order_id ) {
$order = wc_get_order( $order_id );
$items = $order->get_items();

$products_to_check = array( '151' );

foreach ( $items as $item ) {
if ( $order->user_id > 0 && in_array( $item['product_id'], $products_to_check ) ) {
$user = new WP_User( $order->user_id );

// Change role
$user->remove_role( 'customer' );
$user->add_role( 'um_custom_role_6' );

// Exit the loop
break;
}
}
}

`

 

[Blacklight Sorane]

Всем доброго времени суток! Подскажите пожалуйста как решить задачу, ниже код, работает, после оплаты если товар имеет id 151, то удаляется старая роль: customer и записывается новая роль: um_custom_role_6. Вопрос, как правильно добавить еще одно условие, что если товар имеет id 203, то удаляется старая роль: customer и записывается новая роль: um_custom_role_4 ?

`
add_action( 'woocommerce_order_status_completed', 'change_role_on_purchase' );
function change_role_on_purchase( $order_id ) {
$order = wc_get_order( $order_id );
$items = $order->get_items();

$products_to_check = array( '151' );

foreach ( $items as $item ) {
if ( $order->user_id > 0 && in_array( $item['product_id'], $products_to_check ) ) {
$user = new WP_User( $order->user_id );

// Change role
$user->remove_role( 'customer' );
$user->add_role( 'um_custom_role_6' );

// Exit the loop
break;
}
}
}

`

В чем проблема добавить такую же функцию?
Которая будет менять роль на um_custom_role_4?