当用户在 PHP 中注册 WordPress 时，如何创建自定义用户角色？

Question

我的问题是“当用户在注册帐户时选中特定复选框时，如何分配自定义用户角色？”

有两种选择： 选项 1：承包商 选项 2：供应商

注册流程： #1 用户注册 -> 检查承包商或供应商 -> 创建帐户并根据他们选中的选项分配特定用户角色。

到目前为止，我有创建用户流的代码，但我不确定需要向附加功能添加什么条件逻辑。

function contractor_role() {  
 
    //add the contractor user role
    add_role(
        'Contractor',
        'Contractor',
        array(
            'read'         => true,
            'delete_posts' => false
        )
    );
 
}
add_action('admin_init', 'contractor_role');

function vendor_role() {  
 
    //add the vendor user role
    add_role(
        'Vendor',
        'Vendor',
        array(
            'read'         => true,
            'delete_posts' => false
        )
    );
 
}
add_action('admin_init', 'vendor_role');