273 lines
9.3 KiB
Python
273 lines
9.3 KiB
Python
import os
|
|
import zipfile
|
|
from io import BytesIO
|
|
from pathlib import Path
|
|
from mako.lookup import TemplateLookup
|
|
import black
|
|
|
|
class CodeGenerator:
|
|
def __init__(self, template_root='templates'):
|
|
self.template_root = template_root
|
|
self.lookup = TemplateLookup(directories=[template_root], input_encoding='utf-8')
|
|
|
|
def generate_files(self, entity_config, prj_template_dir, params, language='python'):
|
|
"""
|
|
生成所有匹配的模板文件
|
|
:param entity_config: 实体配置
|
|
:param prj_template_dir: 项目对应的文件路径
|
|
:param params: 生成参数
|
|
:return: 生成的文件字典 {相对路径: 文件内容}
|
|
"""
|
|
entity_config = type("EntityConfig", (object,), entity_config)
|
|
generated_files = []
|
|
template_dir = os.path.join(self.template_root, prj_template_dir)
|
|
|
|
# 遍历模板目录
|
|
for root, _, files in os.walk(template_dir):
|
|
for file in files:
|
|
if not file.endswith('.mako'): continue
|
|
template_path = os.path.join(root, file)
|
|
relative_path = os.path.relpath(template_path, self.template_root)
|
|
|
|
# 渲染模板
|
|
template = self.lookup.get_template(relative_path)
|
|
content = template.render(entity=entity_config, params=params, generator=self)
|
|
|
|
if relative_path.endswith('.py.mako'):
|
|
content = self.post_process(content)
|
|
|
|
# 计算输出路径
|
|
output_path = self._get_output_path(relative_path, entity_config, params)
|
|
# generated_files[output_path] = content
|
|
generated_files.append(dict(
|
|
path=output_path,
|
|
content=content
|
|
))
|
|
|
|
return generated_files
|
|
|
|
def post_process(self, content):
|
|
try:
|
|
# 尝试使用 Black 格式化代码
|
|
formatted_code = black.format_str(content, mode=black.Mode(line_length=100))
|
|
return formatted_code
|
|
except Exception as e:
|
|
print(f"Black 格式化失败: {e}")
|
|
return content
|
|
|
|
def _get_output_path(self, template_path, entity_config, params):
|
|
"""
|
|
根据模板路径计算输出路径
|
|
:param template_path: 模板相对路径
|
|
:return: 输出文件相对路径(无.mako后缀)
|
|
"""
|
|
# 移除.mako后缀
|
|
if template_path.endswith('.mako'):
|
|
output_path = template_path[:-5]
|
|
else:
|
|
output_path = template_path
|
|
|
|
# 替换实体名称占位符
|
|
entity_name = entity_config.name
|
|
output_path = output_path.replace('Entity', entity_name.capitalize())
|
|
output_path = output_path.replace('entity', entity_name.lower())
|
|
|
|
return output_path
|
|
|
|
def _process_relations(self, entity):
|
|
"""处理实体关系并返回处理后的结果"""
|
|
if not hasattr(entity, 'relations') or not entity.relations:
|
|
return []
|
|
|
|
processed = []
|
|
for rel in entity.relations:
|
|
# 确保关系有必要的字段
|
|
rel.setdefault('through', None)
|
|
rel.setdefault('description', '')
|
|
|
|
# 计算反向引用名称
|
|
rel['reverse_name'] = self._get_reverse_relation_name(entity.name, rel)
|
|
processed.append(rel)
|
|
return processed
|
|
|
|
def _get_reverse_relation_name(self, entity_name, relation):
|
|
"""生成反向引用名称"""
|
|
if relation['type'] == 'one-to-many':
|
|
return entity_name.lower()
|
|
elif relation['type'] == 'many-to-one':
|
|
return relation['name']
|
|
elif relation['type'] == 'many-to-many':
|
|
return f"{entity_name.lower()}s"
|
|
return None
|
|
|
|
def create_zip(self, generated_files):
|
|
"""
|
|
将生成的文件打包为zip
|
|
:param generated_files: {路径: 内容} 字典
|
|
:return: zip文件字节流
|
|
"""
|
|
zip_buffer = BytesIO()
|
|
with zipfile.ZipFile(zip_buffer, 'w', zipfile.ZIP_DEFLATED) as zip_file:
|
|
for item in generated_files:
|
|
path, content = item.get('path'), item.get('content')
|
|
zip_file.writestr(path, content)
|
|
zip_buffer.seek(0)
|
|
return zip_buffer
|
|
|
|
def create_local_code(self, generated_files, output_dir):
|
|
save_dir = Path(output_dir)
|
|
|
|
for item in generated_files:
|
|
# 确保输出目录存在
|
|
path, content = item.get('path'), item.get('content')
|
|
output_file = save_dir / path
|
|
output_file.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
with open(output_file, 'w', encoding='utf-8', newline='') as f:
|
|
f.write(content)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# 使用示例
|
|
generator = CodeGenerator()
|
|
|
|
default_frontend_config = {
|
|
"editable": True,
|
|
"listable": True,
|
|
"detailable": True,
|
|
"sortable": True,
|
|
"filterable": True,
|
|
"filter_operator": 'equal', # equal | contains 默认过滤操作符为包含
|
|
"display_type": 'text', # 默认显示类型为文本
|
|
}
|
|
|
|
entity_config = {
|
|
"name": "user", # 实体名称（小写）
|
|
"author": "Ly997", # 实体名称（小写）
|
|
"description": "系统用户管理", # 实体描述
|
|
"version": "1.0.0", # 版本号
|
|
"fields": [
|
|
{
|
|
"name": "id",
|
|
"type": "Long",
|
|
"primary_key": True,
|
|
"description": "用户ID",
|
|
"required": True
|
|
},
|
|
{
|
|
"name": "username",
|
|
"type": "String",
|
|
"length": 50,
|
|
"description": "用户名",
|
|
"required": True,
|
|
"unique": True,
|
|
"validations": [
|
|
{"type": "min_length", "value": 4},
|
|
{"type": "max_length", "value": 20}
|
|
]
|
|
},
|
|
{
|
|
"name": "email",
|
|
"type": "String",
|
|
"description": "电子邮箱",
|
|
"required": True,
|
|
"validations": [
|
|
{"type": "email"}
|
|
]
|
|
},
|
|
{
|
|
"name": "password_hash",
|
|
"type": "String",
|
|
"description": "密码哈希",
|
|
"required": True,
|
|
"secret": True # 标记为敏感字段
|
|
},
|
|
{
|
|
"name": "is_active",
|
|
"type": "Boolean",
|
|
"description": "是否激活",
|
|
"default": True
|
|
},
|
|
{
|
|
"name": "created_at",
|
|
"type": "DateTime",
|
|
"description": "创建时间",
|
|
"auto_now_add": True
|
|
},
|
|
{
|
|
"name": "updated_at",
|
|
"type": "DateTime",
|
|
"description": "更新时间",
|
|
"auto_now": True
|
|
}
|
|
],
|
|
"relations": [
|
|
{
|
|
"name": "roles",
|
|
"type": "many-to-many",
|
|
"target": "role",
|
|
"through": "user_roles",
|
|
"description": "用户角色关联"
|
|
},
|
|
{
|
|
"name": "posts",
|
|
"type": "one-to-many",
|
|
"target": "post",
|
|
"description": "用户发表的文章"
|
|
}
|
|
],
|
|
"api": {
|
|
"operations": ["create", "read", "update", "delete", "list"],
|
|
"base_path": "/users",
|
|
"auth_required": True,
|
|
"permissions": {
|
|
"create": ["admin", "manager"],
|
|
"delete": ["admin"]
|
|
}
|
|
},
|
|
"menu": {
|
|
"label": "用户管理",
|
|
"icon": "user",
|
|
"order": 10,
|
|
"submenu": [
|
|
{
|
|
"label": "用户列表",
|
|
"path": "/users",
|
|
"icon": "list"
|
|
},
|
|
{
|
|
"label": "角色管理",
|
|
"path": "/roles",
|
|
}
|
|
]
|
|
}
|
|
}
|
|
params = {
|
|
"package": "com.example",
|
|
"author": "John Doe"
|
|
}
|
|
|
|
# 在代码中合并默认配置和字段的前端配置
|
|
for field in entity_config["fields"]:
|
|
frontend = {**default_frontend_config, **field}
|
|
field.update(frontend)
|
|
|
|
import json
|
|
with open('tmp.json', 'w', encoding='utf-8') as f:
|
|
json.dump(entity_config, f, ensure_ascii=False, indent=4)
|
|
|
|
# 生成所有文件
|
|
generated_files = generator.generate_files(entity_config, 'relation-demo', params)
|
|
# generated_files = generator.generate_files(entity_config, 'vue-fastapi-admin', params)
|
|
generator.create_local_code(generated_files, 'tmp')
|
|
# 打包为zip
|
|
zip_data = generator.create_zip(generated_files)
|
|
|
|
file_path = './tmp.zip'
|
|
with open(file_path, 'wb') as f:
|
|
f.write(zip_data.getvalue())
|
|
print(f"文件已保存到 {file_path}")
|
|
|
|
# 在Flask中返回zip文件示例
|
|
# return send_file(zip_data, mimetype='application/zip', as_attachment=True, download_name='generated_code.zip')
|