如何在WordPress插件Ultimate Member中按邮箱自动分配注册用户角色

核心逻辑 #

借助Ultimate Member提供的um_registration_complete钩子，在用户注册流程完成后，校验其邮箱是否在你的后台标记列表中，根据标记自动调整用户角色。

具体实现步骤 #

1. 准备邮箱标记列表
   直接复用你已有的有效邮箱与标记数据，示例格式如下（如果你的列表存在数据库或其他存储位置，替换成对应的读取逻辑即可）：

   $valid_email_list = array(
       'it_employee@yourcompany.com' => 'IT-department',
       'client_a@example.com' => '',
       // 更多邮箱条目...
   );
   

2. 添加角色分配钩子函数
   将以下代码添加到主题的functions.php文件或自定义功能插件中（禁止修改Ultimate Member核心文件，避免更新失效）：

   add_action('um_registration_complete', 'auto_assign_role_by_email_tag', 10, 2);
   function auto_assign_role_by_email_tag($user_id, $args) {
       // 获取注册用户的邮箱
       $user_email = get_userdata($user_id)->user_email;
       
       // 替换为你实际的邮箱标记列表
       $valid_email_list = array(
           'it_employee@yourcompany.com' => 'IT-department',
           'client_a@example.com' => '',
           // 更多条目...
       );
   
       // 校验邮箱标记并分配角色
       if (isset($valid_email_list[$user_email]) && $valid_email_list[$user_email] === 'IT-department') {
           $user = new WP_User($user_id);
           // 移除默认角色（根据你的实际默认角色调整，比如默认是subscriber）
           $user->remove_role('subscriber');
           // 添加poweruser角色
           $user->add_role('poweruser');
       }
       // 无IT-department标记的用户，保持默认角色分配逻辑
   }
   

注意事项 #

• 确认poweruser角色已在WordPress中存在，若为自定义角色，需提前通过代码注册该角色
• 测试时使用带有IT-department标记的邮箱注册，在后台用户列表中检查角色是否正确分配
• 若你的邮箱列表存储在数据库选项中，可将$valid_email_list替换为get_option('your_email_list_option_key')来读取数据

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