zwf
4729698049
git-subtree-dir: paste-framework git-subtree-split: 34e8684c4bc3cebbe177509f42ab4ef5b5425a7a
51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
"""
|
|
生成数据模型代码，注意这里显式排除了在配置文件 RBAC 中配置的数据表。
|
|
如果不需要排除，可直接使用 sqlacodegen 自带的命令。
|
|
"""
|
|
import subprocess
|
|
from os import path
|
|
|
|
import pandas as pd
|
|
|
|
from paste.core import config
|
|
from paste.core.logging import echo_log
|
|
from paste.db.basetable import BaseTable
|
|
|
|
exclude_tables = list(config.get_config('rbac.table').values())
|
|
"""
|
|
需要排除的数据表，默认排除 RBAC 数据表，这部分表格在 RBAC 模块中已经配置好了。
|
|
"""
|
|
|
|
|
|
def db_engin():
|
|
return config.get_config('db_engine.engine')
|
|
|
|
|
|
async def sqlacodegen(is_exclude_rbac_table: bool = True):
|
|
"""
|
|
生成代码文件。
|
|
|
|
:param is_exclude_rbac_table: 是否排除 RBAC 相关的数据表，默认排除不生成数据模型
|
|
"""
|
|
_table_names = await BaseTable.tables_in_db()
|
|
|
|
# 剔除 RBAC 数据表
|
|
if _table_names and is_exclude_rbac_table:
|
|
# 转换为 DataFrame
|
|
_tables_df = pd.DataFrame(_table_names)
|
|
# 剔除不要包含的表
|
|
_name_df: pd.DataFrame = _tables_df.loc[~_tables_df.iloc[:, 0].isin(exclude_tables)]
|
|
# 转换剩余数据为'表名'字符串列表
|
|
_table_names = _name_df[0].tolist()
|
|
|
|
if len(_table_names) == 0:
|
|
return
|
|
|
|
echo_log(f"将为以下表生成数据模型：{_table_names}")
|
|
_tables = f"--tables={','.join(_table_names)}"
|
|
# 默认创建在当前目录的 models 目录中
|
|
_outfile = f"--outfile={path.join(path.curdir, 'models', 'db_models.py')}"
|
|
|
|
_engin = db_engin()
|
|
subprocess.call(['sqlacodegen', _engin, _tables, _outfile])
|
|
echo_log(f"生成完成.") |