如何根据radio button选中状态为Ultimate Member分配用户角色

Ultimate Member原生表单编辑器仅提供字段显隐规则，没有自定义字段绑定角色分配的可视化配置，可通过插件官方提供的动作钩子实现需求，步骤如下：

• 第一步：确认字段基础参数
  进入Ultimate Member注册表单编辑页，找到你添加的判断用户是否为Contractor的单选按钮，记录两个关键信息：
  1. 字段的Meta Key（字段唯一标识，比如常见命名为is_contractor）
  2. 选中「是Contractor」选项对应的选项值（比如你设置的选中值为1或yes）
     同时进入Ultimate Member > 角色页面，分别确认Customer、Contractor两个角色的系统ID（编辑角色时浏览器地址栏role=后的字符串即为角色ID，Ultimate Member自定义角色默认带um_前缀，比如um_contractor）
• 第二步：插入功能代码
  将以下代码添加到当前子主题的functions.php文件中（推荐使用子主题避免主题更新丢失代码），或通过代码片段类插件插入，不要修改Ultimate Member核心文件：

add_action( 'um_after_new_user_register', 'um_auto_set_role_by_custom_radio', 10, 2 );
function um_auto_set_role_by_custom_radio( $user_id, $submitted_data ) {
    // 以下参数请替换为你自己站点的实际值
    $radio_meta_key = 'is_contractor'; // 单选字段的Meta Key
    $selected_val = 'yes'; // 选中Contractor时的选项值
    $default_role = 'customer'; // 系统默认分配的Customer角色ID
    $target_role = 'um_contractor'; // 需要自动分配的Contractor角色ID

    // 校验提交的字段值
    if ( isset( $submitted_data[$radio_meta_key] ) && trim($submitted_data[$radio_meta_key]) === $selected_val ) {
        // 移除默认分配的Customer角色
        (new WP_User($user_id))->remove_role($default_role);
        // 分配Contractor角色
        wp_update_user([
            'ID' => $user_id,
            'role' => $target_role
        ]);
        // 同步Ultimate Member角色缓存，避免权限识别异常
        update_user_meta($user_id, 'um_role', $target_role);
    }
}

• 第三步：功能验证
  代码参数修改完成后，分别注册两个测试账号：
  1. 注册时不选中Contractor选项，注册完成后到WordPress用户列表核对，角色应为默认的Customer
  2. 注册时选中Contractor选项，核对角色是否自动切换为Contractor

注意：如果你的站点开启了注册邮件激活流程，该代码依然可以正常生效，无需额外调整激活后的回调逻辑；如果单选字段值为数字类型，将$selected_val对应改为数字值即可（比如1）。

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