摘要：有时候我们需要对一些网站用户给予一些小的惩罚，关闭一段时间的登陆功能，或是给予小黑屋的体验，那我们可以对wordpres…
有时候我们需要对一些网站用户给予一些小的惩罚，关闭一段时间的登陆功能，或是给予小黑屋的体验，那我们可以对wordpress后台的用户功能进行开发，添加一个“禁止用户登录”的功能，方便我们更好的管理会员。这段代码也十分简单。
将以下代码函数复制当前wordpress主题的functions.php文件内
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
/**
* 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 );
|
来源：http://www.wazhuti.com/2443.html
