小技巧 | 聊聊 Python 中数据库反向生成 Model 最优方案
1. 前言
大家好，我是安果！
熟悉 Django 的朋友应该知道，我们可以通过「 inspectdb 」命令将数据库表反向生成 Model 并写入到文件中去
比如，Django 项目映射数据库中有一张 student 表，我们希望反向生成 Model 并写入到本地文件 models.py 文件中
只需要输入下面命令：
# 反向生成Model
# 进入到项目根目录，输入下面的命令
# 其中
# student：映射数据库的student数据表
# 写入到index App 下的models.py文件内
python3 manage.py inspectdb student > index/models.py但如果是一个普通 Python 项目，如何反向生成 Model 呢？
2. 普通项目反向生成 Model
假设项目使用「 sqlalchemy 」连接操作数据库
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
engine = create_engine('mysql+pymysql://{}:{}@{}:{}/{}'.format(user, password, host, port, database), echo = True)
session = sessionmaker(bind=self.engine)()
...
接着，我们安装依赖「 sqlacodegen 」
# 安装依赖
pip3 install sqlacodegen
这样，我们就可以使用 sqlacodegen 命令关联数据库表，在本地一键生成 Model 了
def generate_orm_table(db_ip, db_name, table_name, output, username='root', pwd='root', port=3306):
"""
生成orm数据表
:return:
"""
exec_cmd(
'sqlacodegen --tables {} --outfile {} mysql+pymysql://{}:{}@{}:{}/{}?charset=utf8'.format(
table_name, output, username, pwd, db_ip, port, db_name))
if __name__ == '__main__':
# 生成orm数据库表
table_name = 'zlb_users'
# 生成Model
generate_orm_table("192.168.*.*", "db_name", "table_name", "./../model/{}.py".format(table_name))
3. 最后
虽然可以通过命令行一键生成 Model，但是生成的部分字段可能存在误差，需要手动进行调整一下
比如，数据库如果某个字段为时间 datetime，反向生成 Model 中的 DateTimeField 字段参数值需要略微调整
赞 (0)