摘要：有时候我们需要对一些网站用户给予一些小的惩罚，关闭一段时间的登陆功能，或是给予小黑屋的体验，那我们可以对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 ); |
