如何从注册表单中为WordPress中的新用户分配自定义角色

Question

我想根据注册表单在WordPress中分配自定义用户角色。例如，我在注册表上有“老师”和“学生”。我必须放两个复选框，这样如果用户选择“学生”，他们将自动分配学生角色而不是默认角色。

我该怎么做？

Answer 1

您可以在用户注册并在其中添加角色时添加操作挂钩。

add_action( 'user_register', 'myplugin_registration_save', 10, 1 );

function myplugin_registration_save( $user_id ) {
    $user = new WP_User( $user_id );

    $user->remove_role( 'subscriber' ); // Optional, you don't have to remove this role if you want to keep subscriber as well
    $user->add_role( 'student' );
}