ThinkPHP 类似Yii的Gii生成Model的功能。

ThinkPHP 类似Yii的Gii生成Model的功能。
自动生成ThinkPhp 3.1 的基础模型.。

 #!/usr/bin/env php
 <?php
 /**
  *
  * THINKPHP 基础模型生成工具
  * 使用方法
  *  命令行: php 本php文件 表名字
  *   然后复制生成好的代码
  *
  *
  */

 $host = 'localhost';
 $user = 'root';
 $pass = 'root';

 $tablePre = 'weixin_';//表前缀
 $dbName = 'weixin2';

 if($argc < 2)
 {
     $s = __FILE__;
     $str = " 使用方法：php $s tableName";
     $num = strlen($str);
     print_r(str_repeat('-',$num)."\r\n ThinkPHP 基础模型生成工具\r\n 使用方法：php $s tableName\r\n".str_repeat('-',$num)."\r\n");
     return;
 }

 $mysql = new mysqli($host, $user, $pass, $dbName);
 $Tii = new ThinkTii($mysql, $tablePre, $argv[1]);

 $search = array(
     '{%fields%}',
     '{%tableName%}',
     '{%trueTableName%}',
     '{%dbName%}',
     '{%className%}',
     '{%_auto%}',
     '{%_validate%}',
     '{%property%}');
 $replace = array(
     $Tii->getFieldString(),
     $Tii->getTableName(),
     $Tii->getTrueTableName(),
     $Tii->getDbName(),
     $Tii->getModelClassName(),
     $Tii->getAutoFill(),
     $Tii->getAutoValidate(),
     $Tii->getProperty());

 $classString = str_replace($search, $replace, $Tii->getTiiTpl());

 echo $classString;

 function arrayToString($array)
 {
     $string = "array( ";

     ksort($array, SORT_NATURAL);
     list($key) = each($array);
     $key === 0 and $i = 0;
     reset($array);

     foreach ($array as $key => $value) {
         if (isset($i) && $key == $i && ++$i) {
             $key = '';
         } else {
             $key = var_export($key, true) . " => ";
             unset($i);
         }
         if (is_array($value)) {
             $string .= $key . arrayToString($value) . ', ';
         } else {
             $string .= $key . var_export($value, true) . ', ';
         }
     }
     $string = trim(trim($string, ' '), ',');
     $string .= ")";

     return $string;
 }

 /**
  * Class ThinkTii
  *
  * 自动生成ThinkPhp 3.1 的模型
  *
  */
 class ThinkTii
 {

     const TYPE_INT = 'integer';
     const TYPE_STRING = 'string ';

     /**
      * @var mysqli
      */

     protected $mysql = null;
     protected $tablePre = '';
     protected $dbName = '';
     protected $isHump = true;
     protected $host = 'localhost';
     protected $user = 'root';
     protected $pass = 'root';
     protected $tableName = '';
     protected $trueTableName = '';

     public function __construct($mysql, $tablePre, $tableName)
     {
         $this->tableName = $tableName;
         $this->tablePre = $tablePre;
         $this->trueTableName = $this->tablePre . $tableName;//要生成哪张表，完整表名
         $this->mysql = $mysql;
     }

     public function getDbName()
     {
         $row = $this->mysql->query("select database();")->fetch_row();
         $this->dbName = $row[0];
         return $this->dbName;
     }

     public function getTableName()
     {
         return $this->tableName;
     }

     /**
      * 小驼峰转大驼峰
      * @param $name
      * @return mixed|string
      */
     public function humpSmallToBig($name)
     {
         $str = str_replace('_', ' ', $name);
         $str = ucwords($str);
         $str = str_replace(' ', '', $str);
         return $str;
     }

     public function getDesc($tableName)
     {
         $sql = "desc " . $this->getTrueTableName($tableName);
         $this->mysql->set_charset('utf-8');
         $query = $this->mysql->query($sql);
         $fetch = array();
         while (is_array($row = $query->fetch_array(MYSQL_ASSOC))) {
             $fetch[] = $row;
         }
         return $fetch;
     }

     public function getTiiFields($tableName)
     {
         $fetch = $this->getDesc($tableName);

         $fields = array();
         foreach ($fetch as $value) {
             $fields[] = $value['Field'];
             if (strpos($value['Key'], 'PRI') !== false) {
                 $fields['_pk'] = $value['Field'];
             }
             if ($value['Extra'] == 'auto_increment') {
                 $fields['_autoinc'] = true;
             }
         }
         return $fields;
     }

     public function getAutoFill()
     {
         $fetch = $this->getDesc($this->getTrueTableName());
         $array = array();
         foreach ($fetch as $field) {
             if ($field['Default'] !== null) {
                 $array[] = array($field['Field'], $field['Default']);
             }
         }
         return arrayToString($array);
     }

     public function getAutoValidate()
     {
         $fetch = $this->getDesc($this->getTrueTableName());
         $requires = $urls = $emails = $numbers = $array = $numbers = $number1s = array();
         foreach ($fetch as $field) {
             $NotNull = false;
             if ($field['Null'] == "NO" && $field['Default'] === null) {
                 $NotNull = true;
                 //$array[] = array($field['Field'], 'require', $field['Field'] . ' Can not be a null！', 1);
                 $requires[] = $field['Field'];
             }
             if ($field['Key'] == "UNI") {
                 $array[] = array($field['Field'], '', '值已经存在！', 1, 'unique');
             }

             switch ($this->getType($field['Type'])) {
                 case self::TYPE_INT:
                     if ($NotNull) {
                         $number1s[] = $field['Field'];
                     } else {
                         $numbers[] = $field['Field'];
                     }
                     break;
                 case self::TYPE_STRING:
                     if (strpos($field['Field'], 'mail')) {
                         $emails[] = $field['Field'];
                     } elseif (strpos($field['Field'], 'url')) {
                         $urls[] = $field['Field'];
                     }
                     break;
                 case 'enum':
                     $string = rtrim(str_replace(array('enum('), '', $field['Type']), ')');
                     $string = explode(',', $string);
                     $_tmp = array();
                     foreach ($string as $str) {
                         $_tmp[] = trim($str, "'");
                     }
                     $array[] = array($field['Field'], $_tmp, '值的范围不正确！', 2, 'in');
                     unset($_tmp);
                     break;
             }
         }
         empty($numbers) or $array[] = array(implode(',', $numbers), 'number', ' 格式不对');
         empty($number1s) or $array[] = array(implode(',', $number1s), 'number', ' 格式不对', 1);
         empty($emails) or $array[] = array(implode(',', $emails), 'email', ' 格式不对');
         empty($urls) or $array[] = array(implode(',', $urls), 'url', ' 格式不对');
         empty($requires) or $array[] = array(implode(',', $requires), 'require', ' Can not be a null！', 1);

         return arrayToString($array);
     }

     public function getProperty()
     {
         $fetch = $this->getDesc($this->getTrueTableName());
         $property = array();
         foreach ($fetch as $field) {
             $type = $this->getType($field['Type']);
             $type = $type == 'enum' ? self::TYPE_STRING : $type;
             $property[] = " * @property $type \${$field['Field']}";
         }
         return implode("\r\n", $property);
     }

     protected function getType($typeString)
     {
         list($type) = explode('(', $typeString);
         $types = array(
             self::TYPE_INT => array('int'),
             self::TYPE_STRING => array('text', 'char', 'varchar')
         );

         foreach ($types as $key => $value) {
             if (in_array($type, $value)) {
                 return $key;
             }
         }
         return $type;
     }

     public function getFieldString()
     {
         $fieldString = arrayToString($this->getTiiFields($this->tableName));
         return $fieldString;
     }

     public function getTrueTableName()
     {
         return $this->trueTableName;
     }

     public function getModelClassName()
     {
         if ($this->isHump) {
             $className = $this->humpSmallToBig($this->tableName);
         } else {
             $className = $this->tableName;
         }
         return $className;
     }

     public function getTiiTpl()
     {
         $time = date('Y-m-d H:s:i');
         $t = <<<__END
 <?php

 /**
  *
  * Class {%className%}Model
  *  Tii自动生成。请更具自己实际项目需求进行更改
  *      自动验证\$_validate 和自动完成 \$_auto 是更具数据库表默认设置进行生成的。
  *
  * 按照字段自动生成的类属性。方便IDE代码自动补全
  *
 {%property%}
  *
  * @lastTime $time
  * @author niy <136045277@qq.com>
  *
  */
 class {%className%}Model extends Model{
     /**
      * @var string 表名
      */
     protected \$tableName = '{%tableName%}';
     /**
      * @var string 真实的表名
      */
     protected \$trueTableName = '{%trueTableName%}';
     /**
      * @var string 该模型属于哪个数据库
      */
     protected \$dbName = '{%dbName%}';
     /**
      * @var array 本章表的字段
      */
     protected \$fields = {%fields%};

     /**
      * 自动完成
      * @url http://doc.thinkphp.cn/manual/auto_operate.html
      * @var array
      */
     protected \$_auto = {%_auto%};

     /**
      * 自动验证
      * @url http://doc.thinkphp.cn/manual/auto_validate.html
      * @var array
      */
     protected \$_validate = {%_validate%};

     /**
      * 以免破坏程序罗辑，以下函数不需要的请删除。
      */
     protected function _before_insert(&\$data, \$options){}
     protected function _after_insert(\$data,\$options) {}
     protected function _before_update(&\$data,\$options) {}
     protected function _after_update(\$data,\$options) {}
     protected function _after_select(&\$resultSet,\$options) {}
     protected function _after_delete(\$data,\$options) {}
     protected function _after_find(&\$result,\$options) {}
 }

 __END;
         return $t;
     }

 }