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

  1 #!/usr/bin/env php
  2 <?php
  3 /**
  4  *
  5  * THINKPHP 基础模型生成工具
  6  * 使用方法
  7  *  命令行: php 本php文件 表名字
  8  *   然后复制生成好的代码
  9  *
 10  *
 11  */
 12 
 13 
 14 $host = ‘localhost‘;
 15 $user = ‘root‘;
 16 $pass = ‘root‘;
 17 
 18 $tablePre = ‘weixin_‘;//表前缀
 19 $dbName = ‘weixin2‘;
 20 
 21 
 22 if($argc < 2)
 23 {
 24     $s = __FILE__;
 25     $str = " 使用方法：php $s tableName";
 26     $num = strlen($str);
 27     print_r(str_repeat(‘-‘,$num)."\r\n ThinkPHP 基础模型生成工具\r\n 使用方法：php $s tableName\r\n".str_repeat(‘-‘,$num)."\r\n");
 28     return;
 29 }
 30 
 31 
 32 
 33 
 34 $mysql = new mysqli($host, $user, $pass, $dbName);
 35 $Tii = new ThinkTii($mysql, $tablePre, $argv[1]);
 36 
 37 
 38 $search = array(
 39     ‘{%fields%}‘,
 40     ‘{%tableName%}‘,
 41     ‘{%trueTableName%}‘,
 42     ‘{%dbName%}‘,
 43     ‘{%className%}‘,
 44     ‘{%_auto%}‘,
 45     ‘{%_validate%}‘,
 46     ‘{%property%}‘);
 47 $replace = array(
 48     $Tii->getFieldString(),
 49     $Tii->getTableName(),
 50     $Tii->getTrueTableName(),
 51     $Tii->getDbName(),
 52     $Tii->getModelClassName(),
 53     $Tii->getAutoFill(),
 54     $Tii->getAutoValidate(),
 55     $Tii->getProperty());
 56 
 57 $classString = str_replace($search, $replace, $Tii->getTiiTpl());
 58 
 59 echo $classString;
 60 
 61 
 62 function arrayToString($array)
 63 {
 64     $string = "array( ";
 65 
 66     ksort($array, SORT_NATURAL);
 67     list($key) = each($array);
 68     $key === 0 and $i = 0;
 69     reset($array);
 70 
 71     foreach ($array as $key => $value) {
 72         if (isset($i) && $key == $i && ++$i) {
 73             $key = ‘‘;
 74         } else {
 75             $key = var_export($key, true) . " => ";
 76             unset($i);
 77         }
 78         if (is_array($value)) {
 79             $string .= $key . arrayToString($value) . ‘, ‘;
 80         } else {
 81             $string .= $key . var_export($value, true) . ‘, ‘;
 82         }
 83     }
 84     $string = trim(trim($string, ‘ ‘), ‘,‘);
 85     $string .= ")";
 86 
 87     return $string;
 88 }
 89 
 90 
 91 /**
 92  * Class ThinkTii
 93  *
 94  * 自动生成ThinkPhp 3.1 的模型
 95  *
 96  */
 97 class ThinkTii
 98 {
 99 
100     const TYPE_INT = ‘integer‘;
101     const TYPE_STRING = ‘string ‘;
102 
103 
104     /**
105      * @var mysqli
106      */
107 
108     protected $mysql = null;
109     protected $tablePre = ‘‘;
110     protected $dbName = ‘‘;
111     protected $isHump = true;
112     protected $host = ‘localhost‘;
113     protected $user = ‘root‘;
114     protected $pass = ‘root‘;
115     protected $tableName = ‘‘;
116     protected $trueTableName = ‘‘;
117 
118 
119     public function __construct($mysql, $tablePre, $tableName)
120     {
121         $this->tableName = $tableName;
122         $this->tablePre = $tablePre;
123         $this->trueTableName = $this->tablePre . $tableName;//要生成哪张表，完整表名
124         $this->mysql = $mysql;
125     }
126 
127     public function getDbName()
128     {
129         $row = $this->mysql->query("select database();")->fetch_row();
130         $this->dbName = $row[0];
131         return $this->dbName;
132     }
133 
134     public function getTableName()
135     {
136         return $this->tableName;
137     }
138 
139 
140     /**
141      * 小驼峰转大驼峰
142      * @param $name
143      * @return mixed|string
144      */
145     public function humpSmallToBig($name)
146     {
147         $str = str_replace(‘_‘, ‘ ‘, $name);
148         $str = ucwords($str);
149         $str = str_replace(‘ ‘, ‘‘, $str);
150         return $str;
151     }
152 
153 
154     public function getDesc($tableName)
155     {
156         $sql = "desc " . $this->getTrueTableName($tableName);
157         $this->mysql->set_charset(‘utf-8‘);
158         $query = $this->mysql->query($sql);
159         $fetch = array();
160         while (is_array($row = $query->fetch_array(MYSQL_ASSOC))) {
161             $fetch[] = $row;
162         }
163         return $fetch;
164     }
165 
166 
167     public function getTiiFields($tableName)
168     {
169         $fetch = $this->getDesc($tableName);
170 
171         $fields = array();
172         foreach ($fetch as $value) {
173             $fields[] = $value[‘Field‘];
174             if (strpos($value[‘Key‘], ‘PRI‘) !== false) {
175                 $fields[‘_pk‘] = $value[‘Field‘];
176             }
177             if ($value[‘Extra‘] == ‘auto_increment‘) {
178                 $fields[‘_autoinc‘] = true;
179             }
180         }
181         return $fields;
182     }
183 
184 
185     public function getAutoFill()
186     {
187         $fetch = $this->getDesc($this->getTrueTableName());
188         $array = array();
189         foreach ($fetch as $field) {
190             if ($field[‘Default‘] !== null) {
191                 $array[] = array($field[‘Field‘], $field[‘Default‘]);
192             }
193         }
194         return arrayToString($array);
195     }
196 
197 
198     public function getAutoValidate()
199     {
200         $fetch = $this->getDesc($this->getTrueTableName());
201         $requires = $urls = $emails = $numbers = $array = $numbers = $number1s = array();
202         foreach ($fetch as $field) {
203             $NotNull = false;
204             if ($field[‘Null‘] == "NO" && $field[‘Default‘] === null) {
205                 $NotNull = true;
206                 //$array[] = array($field[‘Field‘], ‘require‘, $field[‘Field‘] . ‘ Can not be a null！‘, 1);
207                 $requires[] = $field[‘Field‘];
208             }
209             if ($field[‘Key‘] == "UNI") {
210                 $array[] = array($field[‘Field‘], ‘‘, ‘值已经存在！‘, 1, ‘unique‘);
211             }
212 
213             switch ($this->getType($field[‘Type‘])) {
214                 case self::TYPE_INT:
215                     if ($NotNull) {
216                         $number1s[] = $field[‘Field‘];
217                     } else {
218                         $numbers[] = $field[‘Field‘];
219                     }
220                     break;
221                 case self::TYPE_STRING:
222                     if (strpos($field[‘Field‘], ‘mail‘)) {
223                         $emails[] = $field[‘Field‘];
224                     } elseif (strpos($field[‘Field‘], ‘url‘)) {
225                         $urls[] = $field[‘Field‘];
226                     }
227                     break;
228                 case ‘enum‘:
229                     $string = rtrim(str_replace(array(‘enum(‘), ‘‘, $field[‘Type‘]), ‘)‘);
230                     $string = explode(‘,‘, $string);
231                     $_tmp = array();
232                     foreach ($string as $str) {
233                         $_tmp[] = trim($str, "‘");
234                     }
235                     $array[] = array($field[‘Field‘], $_tmp, ‘值的范围不正确！‘, 2, ‘in‘);
236                     unset($_tmp);
237                     break;
238             }
239         }
240         empty($numbers) or $array[] = array(implode(‘,‘, $numbers), ‘number‘, ‘ 格式不对‘);
241         empty($number1s) or $array[] = array(implode(‘,‘, $number1s), ‘number‘, ‘ 格式不对‘, 1);
242         empty($emails) or $array[] = array(implode(‘,‘, $emails), ‘email‘, ‘ 格式不对‘);
243         empty($urls) or $array[] = array(implode(‘,‘, $urls), ‘url‘, ‘ 格式不对‘);
244         empty($requires) or $array[] = array(implode(‘,‘, $requires), ‘require‘, ‘ Can not be a null！‘, 1);
245 
246         return arrayToString($array);
247     }
248 
249 
250     public function getProperty()
251     {
252         $fetch = $this->getDesc($this->getTrueTableName());
253         $property = array();
254         foreach ($fetch as $field) {
255             $type = $this->getType($field[‘Type‘]);
256             $type = $type == ‘enum‘ ? self::TYPE_STRING : $type;
257             $property[] = " * @property $type \${$field[‘Field‘]}";
258         }
259         return implode("\r\n", $property);
260     }
261 
262 
263     protected function getType($typeString)
264     {
265         list($type) = explode(‘(‘, $typeString);
266         $types = array(
267             self::TYPE_INT => array(‘int‘),
268             self::TYPE_STRING => array(‘text‘, ‘char‘, ‘varchar‘)
269         );
270 
271         foreach ($types as $key => $value) {
272             if (in_array($type, $value)) {
273                 return $key;
274             }
275         }
276         return $type;
277     }
278 
279 
280     public function getFieldString()
281     {
282         $fieldString = arrayToString($this->getTiiFields($this->tableName));
283         return $fieldString;
284     }
285 
286 
287     public function getTrueTableName()
288     {
289         return $this->trueTableName;
290     }
291 
292     public function getModelClassName()
293     {
294         if ($this->isHump) {
295             $className = $this->humpSmallToBig($this->tableName);
296         } else {
297             $className = $this->tableName;
298         }
299         return $className;
300     }
301 
302 
303     public function getTiiTpl()
304     {
305         $time = date(‘Y-m-d H:s:i‘);
306         $t = <<<__END
307 <?php
308 
309 /**
310  *
311  * Class {%className%}Model
312  *  Tii自动生成。请更具自己实际项目需求进行更改
313  *      自动验证\$_validate 和自动完成 \$_auto 是更具数据库表默认设置进行生成的。
314  *
315  * 按照字段自动生成的类属性。方便IDE代码自动补全
316  *
317 {%property%}
318  *
319  * @lastTime $time
320  * @author niy <[email protected]>
321  *
322  */
323 class {%className%}Model extends Model{
324     /**
325      * @var string 表名
326      */
327     protected \$tableName = ‘{%tableName%}‘;
328     /**
329      * @var string 真实的表名
330      */
331     protected \$trueTableName = ‘{%trueTableName%}‘;
332     /**
333      * @var string 该模型属于哪个数据库
334      */
335     protected \$dbName = ‘{%dbName%}‘;
336     /**
337      * @var array 本章表的字段
338      */
339     protected \$fields = {%fields%};
340 
341     /**
342      * 自动完成
343      * @url http://doc.thinkphp.cn/manual/auto_operate.html
344      * @var array
345      */
346     protected \$_auto = {%_auto%};
347 
348     /**
349      * 自动验证
350      * @url http://doc.thinkphp.cn/manual/auto_validate.html
351      * @var array
352      */
353     protected \$_validate = {%_validate%};
354 
355 
356     /**
357      * 以免破坏程序罗辑，以下函数不需要的请删除。
358      */
359     protected function _before_insert(&\$data, \$options){}
360     protected function _after_insert(\$data,\$options) {}
361     protected function _before_update(&\$data,\$options) {}
362     protected function _after_update(\$data,\$options) {}
363     protected function _after_select(&\$resultSet,\$options) {}
364     protected function _after_delete(\$data,\$options) {}
365     protected function _after_find(&\$result,\$options) {}
366 }
367 
368 __END;
369         return $t;
370     }
371 
372 
373 }