WordPress自动分配用户邮箱、首次登录后限改的实现及代码问题解决

原有代码的核心异常原因 #

• 触发钩子选型错误：admin_head 钩子会在每次访问后台任意页面时触发，每次全量遍历所有用户会造成不必要的性能损耗，且触发时机偏离需求——自动分配邮箱本应在用户注册时完成，而非反复批量覆盖存量用户数据
• 逻辑顺序错误导致空邮箱用户被重置：
  • 拼接生成的{user_id}@example.com格式邮箱永远不可能等于空格，因此第一个判断分支会无条件将用户邮箱替换为自动生成值
  • 代码执行顺序问题：第一次调用wp_update_user写入自动生成邮箱后，后续判断仍使用未刷新的旧用户数据——当用户原本邮箱为空时，代码会立刻触发第二次wp_update_user，将邮箱值重置回用户原本的空值，这就是未登记邮箱的用户始终无法被成功分配邮箱的核心原因
• 缺少状态标记：没有为“系统自动分配邮箱”“是否完成首次登录”打唯一标记，后续无法实现“首次登录后禁止修改邮箱”的需求
• 无修改拦截逻辑：完全没有涉及用户编辑资料时的邮箱字段禁用、提交拦截逻辑，无法限制用户自主修改邮箱

修正后的可运行代码 #

将以下代码添加到当前主题的functions.php或自定义功能插件中即可生效：

/**
 * 1. 用户注册完成时，为符合角色要求的用户自动分配 {user_id}@example.com 格式的邮箱
 */
add_action('user_register', 'auto_assign_user_email');
function auto_assign_user_email($user_id) {
    $user = get_userdata($user_id);
    // 仅给订阅者角色的用户分配系统邮箱，可根据需求调整角色判断
    if (!in_array('subscriber', (array)$user->roles)) return;
    // 生成对应格式的自动邮箱
    $auto_email = $user_id . '@example.com';
    // 写入用户meta，记录系统自动分配的邮箱值
    update_user_meta($user_id, 'auto_generated_email', $auto_email);
    // 更新用户邮箱
    wp_update_user([
        'ID' => $user_id,
        'user_email' => $auto_email
    ]);
}

/**
 * 2. 存量用户兼容处理：仅执行一次，为所有符合要求的存量用户补分配自动邮箱
 */
add_action('admin_init', 'batch_assign_existing_users_email');
function batch_assign_existing_users_email() {
    // 防止重复执行，用站点配置标记批量任务状态
    if (get_option('auto_email_batch_done')) return;
    
    $user_ids = get_users(['fields' => 'ID', 'role' => 'subscriber']);
    foreach ($user_ids as $uid) {
        // 跳过已经有自动邮箱标记的用户
        if (get_user_meta($uid, 'auto_generated_email', true)) continue;
        $user_data = get_userdata($uid);
        // 跳过已经存在自定义邮箱的用户，避免误覆盖
        if (!empty($user_data->user_email)) continue;
        
        $auto_email = $uid . '@example.com';
        update_user_meta($uid, 'auto_generated_email', $auto_email);
        wp_update_user([
            'ID' => $uid,
            'user_email' => $auto_email
        ]);
    }
    // 标记批量任务已完成
    update_option('auto_email_batch_done', true);
}

/**
 * 3. 记录用户首次登录状态
 */
add_action('wp_login', 'mark_user_first_login', 10, 2);
function mark_user_first_login($user_login, $user) {
    // 仅首次登录时写入标记
    if (!get_user_meta($user->ID, 'first_login_done', true)) {
        update_user_meta($user->ID, 'first_login_done', true);
    }
}

/**
 * 4. 首次登录完成后，锁定后台个人资料页的邮箱编辑字段
 */
add_action('show_user_profile', 'lock_auto_email_field');
add_action('edit_user_profile', 'lock_auto_email_field');
function lock_auto_email_field($user) {
    $auto_email = get_user_meta($user->ID, 'auto_generated_email', true);
    $first_login_done = get_user_meta($user->ID, 'first_login_done', true);
    // 仅对已完成首次登录、有系统自动邮箱的用户做锁定
    if (!$auto_email || !$first_login_done) return;
    ?>
    <script type="text/javascript">
    document.addEventListener('DOMContentLoaded', function() {
        const emailField = document.querySelector('#email');
        if (emailField) {
            emailField.readOnly = true;
            emailField.disabled = true;
            emailField.insertAdjacentHTML('afterend', '<p class="description">系统分配的邮箱无法自主修改</p>');
        }
    })
    </script>
    <?php
}

/**
 * 5. 后端拦截：防止绕过前端禁用修改邮箱，双重保障锁定效果
 */
add_action('personal_options_update', 'block_email_update');
add_action('edit_user_profile_update', 'block_email_update');
function block_email_update($user_id) {
    $auto_email = get_user_meta($user_id, 'auto_generated_email', true);
    $first_login_done = get_user_meta($user_id, 'first_login_done', true);
    // 仅对已完成首次登录、有系统自动邮箱的用户做拦截
    if (!$auto_email || !$first_login_done) return;
    // 强制将提交的邮箱值改回系统分配的地址
    if (isset($_POST['email'])) {
        $_POST['email'] = $auto_email;
    }
}

方案实现效果说明 #

• 自动分配逻辑：用户注册完成瞬间就会被分配对应格式的邮箱，不需要等批量遍历执行；存量补分配任务只会运行一次，不会占用后台加载的性能
• 锁定逻辑完全匹配需求：
  • 用户注册后、首次登录前，仍可正常编辑邮箱字段
  • 用户完成首次登录后，系统会自动记录登录状态，此时无论从前端还是后端，都无法对系统分配的邮箱进行修改
  • 前端禁用输入框+后端拦截提交的双重机制，可完全杜绝绕过前端修改的可能
• 权限隔离：仅对订阅者角色、系统自动分配邮箱的用户做锁定，管理员、其他有自定义邮箱的用户可正常编辑资料

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