小技巧 | 聊聊 Python 中数据库反向生成 Model 最优方案

AirPython

发布于 2021-07-12 14:34:20

3120

发布于 2021-07-12 14:34:20

文章被收录于专栏：Python 自动化

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))