小技巧 | 聊聊 Python 中資料庫反向生成 Model 最優方案

2021-06-16 11:05:33

來源：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_enginefrom sqlalchemy.orm import sessionmakerengine = 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 欄位參數值需要略微調整

小技巧 | 聊聊 Python 中資料庫反向生成 Model 最優方案

熱門文章