WordPress用代码限制某些用户登录

在一些情况下，比如某些用户损害了网站的利益，有不方便直接删除账号的，你可能就需要禁止他们登录网站，今天咱们就来介绍下wordpress如何添加禁止用户登录。

将下面代码添加到当前主题函数模板 functions.php中即可。

//WordPress 禁止某些用户登录
//在资料页面添加选项
function lxtx_rc_admin_init() {
	// 编辑用户资料
	add_action( 'edit_user_profile', 'lxtx_rc_edit_user_profile' );
	add_action( 'edit_user_profile_update', 'lxtx_rc_edit_user_profile_update' );
}

add_action( 'admin_init', 'lxtx_rc_admin_init' );
//在个人资料页面添加一个复选框
function lxtx_rc_edit_user_profile() {
	if ( !current_user_can( 'edit_users' ) ) {
	return;
}

global $user_id; 
// 用户不能禁止自己
$current_user = wp_get_current_user();
$current_user_id = $current_user->ID;
if ( $current_user_id == $user_id ) {
	return;
}
?>

<h3>权限设置</h3>

<table class="form-table">
	<tr>
		<th scope="row">禁止用户登录</th>
		<td><label for="lxtx_rc_ban"><input name="lxtx_rc_ban" type="checkbox" id="lxtx_rc_ban" <?php if (lxtx_rc_is_user_banned( $user_id )){echo 'checked="checked"';} ?> /> 请谨慎操作，选中则禁止！</label></td>
	</tr>
 </table>
<?php
}

//添加一个函数来将这个选项的值保存到数据库中
function lxtx_rc_edit_user_profile_update() {
	if ( !current_user_can( 'edit_users' ) ) {
		return;
	}

	global $user_id; 
	// 用户不能禁止自己
	$current_user = wp_get_current_user();
	$current_user_id = $current_user->ID;
		if ( $current_user_id == $user_id ) {
		return;
	}
	// 锁定
	if( isset( $_POST['lxtx_rc_ban'] ) && $_POST['lxtx_rc_ban'] = 'on' ) {
		lxtx_rc_ban_user( $user_id );
	} else { // 解锁
		lxtx_rc_unban_user( $user_id );
	}
}

//禁止用户
function lxtx_rc_ban_user( $user_id ) { 
	$old_status = lxtx_rc_is_user_banned( $user_id );
	// 更新状态
	if ( !$old_status ) {
		update_user_option( $user_id, 'lxtx_rc_banned', true, false );
	}
}

//解禁用户
function lxtx_rc_unban_user( $user_id ) { 
	$old_status = lxtx_rc_is_user_banned( $user_id ); 
	// 更新状态
	if ( $old_status ) {
		update_user_option( $user_id, 'lxtx_rc_banned', false, false );
	}
}

//判断用户是否被禁止
function lxtx_rc_is_user_banned( $user_id ) {
	return get_user_option( 'lxtx_rc_banned', $user_id, false );
}

//阻止已禁止的用户登录
function lxtx_rc_authenticate_user( $user ) {
	if ( is_wp_error( $user ) ) {
		return $user;
	} 
	// 如果用户被禁止，则返回错误提示
	$banned = get_user_option( 'lxtx_rc_banned', $user->ID, false );
	if ( $banned ) {
		return new WP_Error( 'lxtx_rc_banned', __( '抱歉，该用户被禁止登录！请联系站长解禁。', 'rc' ) );
	} 
	return $user;
}

//将该函数挂载到 wp_authenticate_user 钩子
add_filter( 'wp_authenticate_user', 'lxtx_rc_authenticate_user', 1 );

添加完成后，即可在后台“编辑用户”页面里看到增加的“禁止用户登录”选项了，选中后，则会禁止该用户登录。