亲宝软件园·资讯

展开

Python数据库反向生成Model

编程学习网 人气: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 字段参数值需要略微调整

加载全部内容

相关教程
猜你喜欢
用户评论