來源:AirPython
作者:星安果
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 字段參數值需要略微調整