优化代码生成

This commit is contained in:
JaguarJack
2020-04-29 09:02:20 +08:00
parent 7f72784bfb
commit 64c03bdfd3
5 changed files with 48 additions and 31 deletions

View File

@@ -8,6 +8,26 @@ use think\facade\Db;
class Model extends Factory
{
public function done($params)
{
$file = $this->getGeneratePath($params['model']);
file_put_contents($file, $this->getContent($params));
if (!file_exists($file)) {
throw new FailedException('create model failed');
}
return true;
}
/**
* get contents
*
* @time 2020年04月29日
* @param $params
* @return string|string[]
*/
public function getContent($params)
{
// TODO: Implement done() method.
$template = new Template();
@@ -23,27 +43,17 @@ class Model extends Factory
}
$content = $template->useTrait($extra['soft_delete']) .
$template->name($table) .
$template->field($this->parseField($table));
$template->name($table) .
$template->field($this->parseField($table));
$class = $template->header() .
$template->nameSpace($namespace) .
$template->uses($extra['soft_delete']) .
$template->createModel($modelName, $table);
$template->nameSpace($namespace) .
$template->uses($extra['soft_delete']) .
$template->createModel($modelName, $table);
$file = $this->getGeneratePath($params['model']);
file_put_contents($file, str_replace('{CONTENT}', $content, $class));
if (!file_exists($file)) {
throw new FailedException('create model failed');
}
return true;
return str_replace('{CONTENT}', $content, $class);
}
/**
* parse field
*
@@ -53,6 +63,10 @@ class Model extends Factory
*/
protected function parseField($table)
{
if (!$this->hasTableExists($table)) {
return false;
}
$columns = Db::query('show full columns from ' .
config('database.connections.mysql.prefix') . $table);