Python数据库反向生成Model

时间:2022-09-28 编程学习网人气:0

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 字段参数值需要略微调整

加载全部内容