113 lines
4.5 KiB
Python
113 lines
4.5 KiB
Python
# _*_ coding : UTF-8 _*_
|
||
# @Time : 2025/03/03 00:03
|
||
# @UpdateTime : 2025/03/03 00:03
|
||
# @Author : sonder
|
||
# @File : generate.py
|
||
# @Software : PyCharm
|
||
# @Comment : 本程序
|
||
from datetime import datetime
|
||
|
||
from config.constant import MYSQL_TO_TORTOISE_TYPE
|
||
|
||
|
||
class Generate:
|
||
"""
|
||
代码生成成工具
|
||
"""
|
||
|
||
@classmethod
|
||
def convert_mysql_type(cls, mysql_type: str):
|
||
"""
|
||
将 MySQL 字段类型转换为 Tortoise-ORM 的字段类型。
|
||
|
||
:param mysql_type: MySQL 的字段类型，如 "varchar(255)"、"int(11)" 等
|
||
:return: Tortoise-ORM 对应的字段类型，如 "CharField"、"IntField" 等
|
||
"""
|
||
# 提取字段的基本类型（去掉括号及长度信息）
|
||
base_type = mysql_type.split("(")[0].lower()
|
||
|
||
# 处理特殊情况：tinyint(1) -> BooleanField
|
||
if base_type == "tinyint":
|
||
if "(1)" in mysql_type: # tinyint(1) 视为 Boolean
|
||
return "BooleanField", None
|
||
return "IntField", None # 其他 tinyint 作为 IntField 处理
|
||
max_length = None
|
||
if "(" in mysql_type and base_type in {"char", "varchar"}:
|
||
max_length = int(mysql_type.split("(")[1].split(")")[0]) # 提取最大长度
|
||
|
||
field_type = MYSQL_TO_TORTOISE_TYPE.get(base_type, "TextField")
|
||
return field_type, max_length
|
||
|
||
@classmethod
|
||
def prepare_template_data(cls, table_info, columns):
|
||
TYPE_DEFINE = {
|
||
"int": 0,
|
||
"str": "",
|
||
"bool": False,
|
||
"float": 0.0,
|
||
"datetime": "",
|
||
"list": [],
|
||
"dict": {},
|
||
"set": set(),
|
||
"tuple": (),
|
||
"bytes": b"",
|
||
"None": 'null',
|
||
}
|
||
PYTHON_TO_TS = {
|
||
"int": "number",
|
||
"float": "number",
|
||
"str": "string",
|
||
"bool": "boolean",
|
||
"datetime": "string",
|
||
"Optional[int]": "number | null",
|
||
"Optional[str]": "string | null",
|
||
"Optional[bool]": "boolean | null",
|
||
"Optional[float]": "number | null",
|
||
"Optional[datetime]": "string | null",
|
||
"List[int]": "number[]",
|
||
"List[str]": "string[]",
|
||
"List[bool]": "boolean[]",
|
||
"List[datetime]": "string[]",
|
||
"Dict[str, Any]": "Record<string, any>",
|
||
"Any": "any",
|
||
}
|
||
common_column = ["id", "del_flag", "create_by", "update_by", "create_time", "update_time"]
|
||
"""组织数据，供 Jinja2 渲染"""
|
||
return {
|
||
"author": table_info.get('author', ""),
|
||
"prefix": table_info["prefix"],
|
||
"table_name": table_info["table_name"],
|
||
"class_name": table_info["class_name"],
|
||
"table_comment": table_info.get("table_comment", ""),
|
||
"description": table_info.get("description", ""),
|
||
"name": table_info.get('class_name', "").lower(),
|
||
"permission_id": table_info.get('permission_id', ""),
|
||
"columns": [
|
||
{
|
||
"python_name": col["python_name"],
|
||
"field_type": cls.convert_mysql_type(col["column_type"])[0],
|
||
"max_length": cls.convert_mysql_type(col["column_type"])[1],
|
||
"is_common": col["python_name"] in common_column,
|
||
"is_nullable": not col["is_required"],
|
||
"is_unique": col.get("is_unique", False),
|
||
"default": col.get("default", None),
|
||
"column_comment": col.get("column_comment", ""),
|
||
"column_name": col["column_name"],
|
||
"is_required": col.get("is_required", False),
|
||
"is_edit": col.get("is_edit", False),
|
||
"is_list": col.get("is_list", False),
|
||
"is_query": col.get("is_query", False),
|
||
"is_insert": col.get("is_insert", False),
|
||
"is_hide": col.get("is_hide", False),
|
||
"query_way": col.get("query_way", "="),
|
||
"show_type": col.get("show_type", "input"),
|
||
"python_type": col.get("python_type", "str"),
|
||
"define": TYPE_DEFINE.get(col.get("python_type", "str"), None),
|
||
"typescript_type": PYTHON_TO_TS.get(col.get("python_type", "str"), "any"),
|
||
}
|
||
for col in columns
|
||
if col["python_name"] and col["column_type"]
|
||
],
|
||
"current_time": datetime.now().strftime("%Y/%m/%d %H:%M:%S")
|
||
}
|