wordpress 禁止用户角色登录后台

//防止普通会员后台更新资料
if ( is_admin() && ( !defined( 'DOING_AJAX' ) || !DOING_AJAX ) ) {
    $current_user = wp_get_current_user();
    $user_role = $current_user->roles[0];
    
    //禁止列表
    $roles = [
        get_option('default_role'),
        'contributor',
        'subscriber',
        'author',
        'editor',
    ];

    if(in_array($user_role,$roles)){
        wp_safe_redirect( home_url() );
        exit();
    }
}