现在会有很多WordPress插件能帮助实现注册邀请码功能，今天想要分享WordPress主题开发注册邀请码功能教程，可以实现在当前使用的WordPress主题中集成注册邀请码功能。

实现思路

1、需要新建一个数据表来保存邀请码，数据表结构如下：

code – 邀请码

max – 邀请码使用次数

users – 使用这个验证码的所有用户

status – 验证码是否可用

2、后台需要两个页面：邀请码列表、添加邀请码。

3、添加邀请码的时候需要能设置前缀，一次生成多个邀请码，邀请码长度可自定义，每个邀请码使用次数可设置。

实现步骤

1、在当前主题的functions.php文件中添加以下代码：

说明：该代码的作用是建立数据库以及对数据的获取、增加、删除、更改等操作。

<?php

//第一次启用主题时执行

function ashuwp_load_theme() {

global $pagenow;

if ( is_admin() && ‘themes.php’ == $pagenow && isset( $_GET[‘activated’] ) ){

ashuwp_invitation_code_install();

}

}

add_action( ‘load-themes.php’, ‘ashuwp_load_theme’ );

//建立数据表

function ashuwp_invitation_code_install(){

global $wpdb;

$table_name = $wpdb->prefix . ‘invitation_code’;

if( $wpdb->get_var(“SHOW TABLES LIKE ‘$table_name’”) != $table_name ) :

$sql = ” CREATE TABLE `”.$wpdb->prefix.”invitation_code` (

`id` BIGINT NOT NULL AUTO_INCREMENT PRIMARY KEY ,

`code` varchar(40),

`max` INT NOT NULL,

`users` varchar(20),

`status` tinyint

) ENGINE = MYISAM DEFAULT CHARSET=utf8;”;

require_once(ABSPATH . ‘wp-admin/includes/upgrade.php’);

dbDelta($sql);

endif;

}

//插入数据

function ashuwp_insert_invitation_code( $code, $max = 1, $users, $status){

global $wpdb;

if($code==”){

return false;

}

$code = trim($code);

$code_exists = ashuwp_check_invitation_code($code);

if(!$code_exists){

$insert = “insert into “.$wpdb->prefix.”invitation_code (code,max,users,status) values( ‘$code’, ‘$max’, ”,’1′)”;

$wpdb->query($insert);

return true;

}else{

return false;

}

}

//检查邀请码是否已存在

function ashuwp_check_invitation_code( $code ){

global $wpdb;

$sql = “select * from “.$wpdb->prefix.”invitation_code where code=’$code’”;

$result = $wpdb->get_results($sql);

if(!empty($result)){

return true;

}else{

return false;

}

}

//获取邀请码

function ashuwp_get_invitation_code($per_page=50, $page=1){

global $wpdb;

$page = (int)$page;

$per_page = (int)$per_page;

if(!$page){

$page = 1;

}

if(!$per_page){

$per_page = 50;

}

$begin = $per_page*($page-1);

$end = $per_page*$page;

$sql = “select * from “.$wpdb->prefix.”invitation_code limit $begin,$end”;

$results = $wpdb->get_results($sql,’ARRAY_A’);

return $results;

}

//邀请码的删除、启用、禁用。

function ashuwp_operation_invitation_code( $id, $action ){

global $wpdb;

$id = (int)$id;

if(!$id){

return false;

}

if(!in_array($action,array(‘delete’,’deactive’,’active’))){

return false;

}

if($action ==’delete’){

$sql = “delete from “.$wpdb->prefix .”invitation_code where id=’$id’”;

}

if($action ==’deactive’){

$sql = “update “.$wpdb->prefix .”invitation_code set status=0 where id=’$id’”;

}

if($action ==’active’){

$sql = “update “.$wpdb->prefix .”invitation_code set status=1 where id=’$id’”;

}

$result = $wpdb->query($sql);

if($result){

return true;

}else{

return false;

}

}

2、继续添加以下代码：

说明：这部分代码是制作网站后台部分，包括邀请码列表页面和邀请码增加页面。

<?php

class ashuwp_invitation_code_admin {

static public $instance;

public function __construct(){

add_action( ‘admin_menu’, array(&$this, ‘ashuwp_invitation_code_menu’) );

}

function ashuwp_invitation_code_menu(){

add_menu_page( ‘邀请码’, ‘邀请码’, ‘manage_options’, ‘invitation_code’, array(&$this, ‘invitation_code_list’),”,27);

add_submenu_page(‘invitation_code’, ‘添加邀请码’, ‘添加邀请码’, ‘manage_options’, ‘invitation_code_add’, array(&$this, ‘invitation_code_add’));

}

function invitation_code_list(){

if( isset($_GET[‘code_action’]) && in_array($_GET[‘code_action’],array(‘delete’,’deactive’,’active’)) && isset($_GET[‘code_id’]) ){

$code_id = (int)$_GET[‘code_id’];

if(!$code_id){

return;

}

$result = ashuwp_operation_invitation_code( $code_id, $_GET[‘code_action’] );

}

$code_lists = ashuwp_get_invitation_code(999,1);

?>

<div class=”wrap”>

<h2 class=”wp-heading-inline”>邀请码</h2>

<a href=”<?php echo admin_url( ‘admin.php?page=invitation_code_add’ ); ?>” class=”page-title-action”>添加</a>

<hr class=”wp-header-end”>

<?php

if(isset($result)){

if($result){

?>

<div id=”message” class=”notice notice-success”>操作成功。</div>

<?php

}else{

?>

<div id=”message” class=”notice notice-error”>操作失败。</div>

<?php

}

}

?>

<ul class=”subsubsub”><li class=”all”>全部<span class=”count”>（<?php echo count($code_lists); ?>）</span></ul>

<table class=”wp-list-table widefat fixed”>

<thead>

<tr>

<th scope=”col”>ID</th>

<th scope=”col”>邀请码</th>

<th scope=”col”>统计(最大使用数/已使用)</th>

<th scope=”col”>用户</th>

<th scope=”col”>操作</th>

</tr>

</thead>

<tbody>

<?php

if($code_lists){

foreach($code_lists as $code){

$users = array();

if(!empty($code[‘users’])){

$users = explode( ‘,’, $code[‘users’] );

}

$used = count($users);

?>

<tr>

<td><?php echo $code[‘id’]; ?></td>

<td>

<?php echo $code[‘code’]; ?>

<?php

if(empty($code[‘status’]) || !$code[‘status’]){

echo ‘-已禁用’;

}

?>

</td>

<td>

<?php echo $code[‘max’].’/’.$used; ?>

</td>

<td>

<?php

foreach( $users as $user_id ){

$user = get_user_by(‘id’, $user_id);

if(!empty($user)){

?>

<a href=”<?php echo admin_url( ‘user-edit.php?user_id=’.$user->ID ); ?>”><?php echo $user->user_login; ?></a>,

<?php

}

}

?>

</td>

<td>

<a href=”<?php echo admin_url( ‘admin.php?page=invitation_code&code_action=delete&code_id=’.$code[‘id’] ); ?>”>删除</a>

<?php

if(empty($code[‘status’]) || !$code[‘status’]){

?>

<a href=”<?php echo admin_url( ‘admin.php?page=invitation_code&code_action=active&code_id=’.$code[‘id’] ); ?>”>启用</a>

<?php

}else{

?>

<a href=”<?php echo admin_url( ‘admin.php?page=invitation_code&code_action=deactive&code_id=’.$code[‘id’] ); ?>”>禁用</a>

<?php

}

?>

</td>

</tr>

<?php

}

}

?>

</tbody>

<tfoot>

<tr>

<th scope=”col”>ID</th>

<th scope=”col”>邀请码</th>

<th scope=”col”>统计</th>

<th scope=”col”>用户</th>

<th scope=”col”>操作</th>

</tr>

</tfoot>

</table>

<div class=”tablenav bottom”>

<div class=”tablenav-pages”>

<span class=”pagination-links”>

</span>

</div>

</div>

</div>

<?php

}

function invitation_code_add(){

$data_codes = ashuwp_get_invitation_code(999,1);

$code_list = array();

foreach($data_codes as $code){

$code_list[] = $code[‘code’];

}

if(isset($_REQUEST[‘submit’]) && isset($_REQUEST[‘ashuwp_invitation_code_field’]) && check_admin_referer(‘ashuwp_invitation_code_action’, ‘ashuwp_invitation_code_field’) ) {

$code_prefix = ”;

if(!empty($_POST[‘code_prefix’])){

$code_prefix = trim($_POST[‘code_prefix’]);

}

$code_length = ”;

if(!empty($_POST[‘code_length’])){

$code_length = (int)$_POST[‘code_length’];

}

if(!$code_length){

$code_length = 8;

}

$code_number = 1;

if(!empty($_POST[‘code_number’])){

$code_number = (int)$_POST[‘code_number’];

}

if(!$code_number){

$code_number = 1;

}

$code_counter = ”;

if(!empty($_POST[‘code_counter’])){

$code_counter = (int)$_POST[‘code_counter’];

}

if(!$code_counter){

$code_counter = 1;

}

$i=1;

$code_tem = array();

while ( $i <= $code_number ){

$tem = strtoupper( $code_prefix . wp_generate_password( $code_length, false ) );

if(!in_array($tem,$code_list)){

$i++;

$code_tem[] = $tem;

ashuwp_insert_invitation_code( $tem, $code_counter, ”, 1);

}

}

}

?>

<div class=”wrap”>

<h2 class=”wp-heading-inline”>添加邀请码</h2>

<a href=”<?php echo admin_url( ‘admin.php?page=invitation_code_add’ ); ?>” class=”page-title-action”>添加</a>

<hr class=”wp-header-end”>

<?php

if(!empty($code_tem)){

?>

<div id=”message” class=”notice notice-success”>

<p>邀请码添加成功：</p>

<?php

foreach($code_tem as $text){

echo ‘<p>’.$text.'</p>’;

}

?>

</div>

<?php

}

?>

<form action=”” method=”post”>

<table class=”form-table”>

<tbody>

<tr>

<td><label for=”code_prefix”>邀请码前缀</label></td>

<td>

<input type=”text” id=”code_prefix” name=”code_prefix” class=”regular-text” value=””/>

<p class=”description”>前缀可不填。</p>

</td>

</tr>

<tr>

<td><label for=”code_length”>邀请码字符长度</label></td>

<td>

<input type=”text” id=”code_length” name=”code_length” class=”regular-text” value=””/>

<p class=”description”>字符长度不包括前缀，默认8个字符。</p>

</td>

</tr>

<tr>

<td><label for=”code_number”>邀请码个数</label></td>

<td>

<input type=”text” id=”code_number” name=”code_number” class=”regular-text” value=””/>

<p class=”description”>本次生成多少个邀请码，默认1个。</p>

</td>

</tr>

<tr>

<td><label for=”code_counter”>允许使用的次数</label></td>

<td>

<input type=”text” id=”code_counter” name=”code_counter” class=”regular-text” value=””/>

<p class=”description”>每个邀请码允许使用的次数，默认1次。</p>

</td>

</tr>

</tbody>

</table>

<p class=”submit”>

<?php wp_nonce_field( ‘ashuwp_invitation_code_action’,’ashuwp_invitation_code_field’ ); ?>

<input type=”submit” name=”submit” id=”submit” class=”button button-primary” value=”生成邀请码”>

</p>

</form>

</div>

<?php

}

}

$invitation_code = new ashuwp_invitation_code_admin();

完成效果

1、增加邀请码界面

2、邀请码列表界面