20 lines
598 B
Python
20 lines
598 B
Python
"""
|
||
自动生成数据库表和sqlalchemy对应的Model
|
||
"""
|
||
import os.path
|
||
import subprocess
|
||
|
||
from db import DB_URL
|
||
from log import PROJECT_ROOT
|
||
|
||
if __name__ == '__main__':
|
||
table = input('请输入表名：')
|
||
out_file = os.path.join(PROJECT_ROOT, 'db', f'{table}.py')
|
||
command = f'sqlacodegen {DB_URL} --outfile={out_file} --tables={table}'
|
||
|
||
try:
|
||
subprocess.run(command, shell=True, check=True)
|
||
print(f'{table}.py文件生成成功！请检查并复制到合适的文件中！')
|
||
except Exception as e:
|
||
print(f'生成{table}.py文件时发生错误: {e}')
|