纯代码实现WordPress邀请码功能

的思路:

  1. 需要新建一个数据表来保存邀请码。
  2. 后台需要两个页面:邀请码列表、添加邀请码。
  3. 添加邀请码的时候需要能设置前缀,一次生成多个邀请码,邀请码长度可自定义,每个邀请码使用次数可设置。

数据表

code:邀请码、max:邀请码使用次数、users:使用这个验证码的所有用户、status:验证码是否可用。

数据库操作

数据库操作部分代码包括:1. 建立数据库。2.对数据的获取、增加、删除、更改等操作。

<?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;
  }
}

网站后台 网站后台部分代码包括:1. 邀请码列表页面。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">
      <h1 class="wp-heading-inline">邀请码</h1>
      <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">
      <h1 class="wp-heading-inline">添加邀请码</h1>
      <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. 邀请码列表

结束语

上面范例代码,仅做参考使用,可根据实际自行优化,具体如何使用邀请码也请自行斟酌。

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏杨建荣的学习笔记

关于MySQL极限值的初步验证纠错

晚上从珠江边回来之后,看到一篇文章说,MySQL有几个极限值,一个表的字段最多只有1017个,我看了以后表示怀疑。怎么快速验证呢,我看到文章的时候已经11点了,...

3175
来自专栏开源FPGA

浅谈Verilog HDL代码编写风格

         消失了好久,没有写文章,也没有做笔记,因为最近再赶一个比赛,时间很紧,昨天周六终于结束了,所以趁着周末这会儿有时间,写点东西,记录下来。首先我...

20010
来自专栏我的博客

TP入门第十二天

1、RBAC权限管理 使用内置RBAC,最主要是懂得官方提供的四个数据表关系,这里重点说明下四个表的关系 thinkphp中RBAC有四个表access、rol...

2634
来自专栏Ceph对象存储方案

对象存储基础概念

对象存储诞生之初 谈到为什么要有对象存储,必须聊聊对象存储诞生之前的两大存储模型:块存储和文件存储。 块存储主要是将存储介质的空间整个映射给主机使用的,主机如果...

3994
来自专栏程序员的碎碎念

redis不难,benny带你入门

关于redis的学习,相信在各大博客、公众号上和教学视频教程里有很多,benny学习了一段时间总算是入门了,在会不定期的更新redis学习日记。

834
来自专栏数据和云

奇异故障:SQL执行反复一慢两快

何剑敏 Oracle ACS华南区售后团队,首席技术工程师。多年从事第一线的数据库运维工作,有丰富项目经验、维护经验和调优经验,专注于数据库的整体运维。 今天...

2774
来自专栏逍遥剑客的游戏开发

C#脚本实践(六): 脚本相对于C++的优势

2063
来自专栏竹清助手

现代化PHP开发

PHP 正在重生。作为一门专注WEB开发的语言,它不断吸取其他语言的优点,如命名空间,闭包,性状,操作码缓存等特性,PSR 规范和Composer 包管理以及...

762
来自专栏更流畅、简洁的软件开发方式

【自然框架】之通用权限(二):人员表组

      继续,这是第二章了。本来想在这一章里面介绍三个表组来着,但是我有点写不好的感觉,还是多分几章吧,这一章就只介绍人员表组。第二章到第五章主要是介绍表结...

19510
来自专栏数据和云

数据库时间出现&#39;0000/00/00&#39;,难道我穿越了?

前几天有个朋友遇到一个问题,在做日期类型数据的运算的时候出现了‘0000-00-00’的结果,不得其解。你是否遇到过同样的问题呢?这样一个并不存在的时间点,难道...

2946

扫码关注云+社区