Become a sponsor
本章概要
代码生成引擎的工作原理，包括配置解析、模板渲染、文件写入和自动注册的完整流程。
代码生成引擎
generator_config.py 中的 CodeGenerator 类是代码生成器的执行层，负责接收配置字典、渲染模板、写入文件、自动注册路由和菜单。
核心类：CodeGenerator
python
from generator_config import CodeGenerator
generator = CodeGenerator(config)
generator.generate() # 交互模式（询问迁移和菜单创建）
generator.generate(auto=True) # 自动模式（Web 界面调用）构造参数
| 参数 | 说明 |
|---|---|
config | 配置字典（由 TableSchemaParser.parse() 或 JSON 文件提供） |
构造时自动：
- 设置模板目录：
public/templates/ - 设置输出目录：
application/{app_name}/ - 初始化 Jinja2 模板环境（
trim_blocks=True,lstrip_blocks=True）
generate() 主方法
执行流程
步骤1: 创建输出目录 application/{app_name}/
│
步骤2: 准备模板上下文 _prepare_context()
│ - 处理字段信息、选项、标识
│ - 分类字段：form_fields / list_fields / filter_fields
│
步骤3: 渲染后端模板并写入文件
│ - models.py （ORM 模型）
│ - forms.py （Django 表单验证）
│ - services.py （业务逻辑层）
│ - views.py （视图层）
│ - urls.py （路由配置）
│ - apps.py （应用配置）
│ - admin.py （后台管理）
│
步骤3.5: 创建 __init__.py 和 migrations/__init__.py
│
步骤3.6: 更新 settings.py 的 INSTALLED_APPS
│ - 自动追加 'application.{app_name}'
│
步骤3.7: 更新 application/urls.py
│ - 自动追加路由配置
│
步骤4: 渲染前端模板并写入文件
│ - ui/ 或 ui2/ 目录下的 Vue 组件
│ - API 接口文件
│
步骤5: 数据库迁移（仅交互模式）
│ - python manage.py makemigrations
│ - python manage.py migrate
│
步骤6: 创建菜单权限节点
- 主菜单节点（type=0）
- 8 个权限节点（type=1）参数
| 参数 | 说明 |
|---|---|
auto | 自动模式（True: 跳过交互确认，直接执行迁移和创建菜单） |
模板上下文
_prepare_context()
准备后端模板的渲染上下文：
| 变量 | 类型 | 说明 |
|---|---|---|
app_name | str | 模块名（如 example） |
module_comment | str | 中文名（如 案例） |
module_name | str | 模块名称（下划线移除） |
model_class_name | str | 类名（如 Example） |
model_class_name_camel | str | 驼峰类名（首字母小写） |
route_prefix | str | 路由前缀 |
permission_prefix | str | 权限前缀（如 sys:example） |
display_field | str | 显示字段（如 name） |
has_sort | bool | 有排序字段 |
has_status | bool | 有状态字段 |
has_unique_code | bool | 有唯一编码字段 |
has_image_field | bool | 有图片字段 |
has_rich_text_field | bool | 有富文本字段 |
fields | list | 全部字段 |
form_fields | list | 表单字段 |
list_fields | list | 列表显示字段 |
filter_fields | list | 筛选字段 |
filterable_fields | list | 可过滤字段 |
searchable_fields | list | 可搜索字段 |
editable_fields | list | 可编辑字段 |
_prepare_vue_context()
在后端上下文基础上扩展前端专用变量：
| 变量 | 类型 | 说明 |
|---|---|---|
api_path | str | API 路径（如 tool/example） |
has_export | bool | 有导出功能 |
is_tree_structure | bool | 是否树形结构 |
parent_id_field | str | 父级字段名 |
has_search_form | bool | 显示搜索表单 |
show_selection | bool | 显示复选框（树形结构） |
文件生成
后端文件
| 模板 | 输出路径 | 说明 |
|---|---|---|
models.py.tpl | application/{name}/models.py | ORM 模型 |
forms.py.tpl | application/{name}/forms.py | Django 表单验证 |
services.py.tpl | application/{name}/services.py | 业务逻辑层 |
views.py.tpl | application/{name}/views.py | 视图层 |
urls.py.tpl | application/{name}/urls.py | 路由配置 |
apps.py.tpl | application/{name}/apps.py | 应用配置 |
admin.py.tpl | application/{name}/admin.py | 后台管理 |
前端文件（普通列表）
| 模板 | 输出路径 |
|---|---|
ui/index.vue.tpl | ui/src/views/tool/{name}/index.vue |
ui/edit.vue.tpl | ui/src/views/tool/{name}/edit.vue |
ui/detail.vue.tpl | ui/src/views/tool/{name}/detail.vue |
ui/columns.ts.tpl | ui/src/views/tool/{name}/columns.ts |
ui/querySchemas.ts.tpl | ui/src/views/tool/{name}/querySchemas.ts |
ui/api.ts.tpl | ui/src/api/tool/{name}.ts |
前端文件（树状列表）
当表包含 parent_id / pid 字段时，自动切换为树形模板：
| 模板 | 输出路径 |
|---|---|
ui2/index.vue.tpl | ui/src/views/tool/{name}/index.vue |
ui2/edit.vue.tpl | ui/src/views/tool/{name}/edit.vue |
ui2/detail.vue.tpl | ui/src/views/tool/{name}/detail.vue |
ui2/columns.ts.tpl | ui/src/views/tool/{name}/columns.ts |
ui/api.ts.tpl | ui/src/api/tool/{name}.ts |
树状列表与普通列表共用 api.ts.tpl。
自动注册
应用注册
_update_installed_apps() 在 application/settings.py 的 INSTALLED_APPS 列表末尾追加：
python
'application.example', # 案例管理幂等操作：如果已存在，跳过。
路由注册
_update_main_urls() 在 application/urls.py 的 urlpatterns 列表末尾追加：
python
# 案例模块路由
# 访问地址: /example/
path('example/', include('application.example.urls')),路由前缀支持下划线转斜杠：order_item -> order/item/。
幂等操作：如果路由已存在，跳过。
菜单创建
_create_permission_nodes() 在数据库中创建：
主菜单（type=0）：模块入口菜单
- name: 模块中文名
- path:
/{menu_group}/{module_path} - component:
/{menu_group}/{module_path}/index - parent_id: 配置中的
parent_menu_id
权限节点（type=1）x 8 个：
| 排序 | 权限名称 | 权限标识 |
|---|---|---|
| 1 | 查询分页 | {prefix}:page |
| 5 | 查询列表 | {prefix}:list |
| 10 | 查询详情 | {prefix}:detail |
| 15 | 添加 | {prefix}:add |
| 20 | 更新 | {prefix}:update |
| 25 | 设置状态 | {prefix}:status |
| 30 | 删除 | {prefix}:delete |
| 35 | 批量删除 | {prefix}:batchDelete |
幂等操作：如果主菜单已存在（path + type 匹配），更新信息并删除旧权限节点后重建。
总结
代码生成引擎根据配置字典，通过 Jinja2 模板渲染生成后端（models/forms/services/views/urls/apps/admin）和前端（index.vue/edit.vue/columns.ts/querySchemas.ts/api.ts）全套代码，生成后自动注册应用、路由和权限。