1、使用 Flask-Migrate 實現數據庫遷移
db.create_all()不會重新創建表或是更新表，需要先使用db.drop_all()刪除數據庫中所有的表之后再調用db.create_all()才能重新創建表，但是這樣的話，原來表中的數據就都被刪除了，這肯定是不行的，這時就出現了數據庫遷移的概念。
在開發過程中，隨著需求的變化，有可能需要添加或修改表的一些字段，但是原表中的數據不能刪除，此時就需要創建新表，并將舊表中的數據遷移至新表中，Flask-Migrate這個擴展就可以在不破壞數據的情況下更新數據庫表的結構，并完成數據從舊表到新表的遷移。
2、Flask-Migrate的使用
可以使用pip install flask-migrate進行安裝。在程序中，我們實例化 Flask_Migrate 提供的 Migrate 類，進行初始化操作。
|
1
2
3
4
5
6
7
8
9
10
11
|
from flask_migrate import Migratefrom flask_sqlalchemy import SQLAlchemyapp = Flask(__name__)···db = SQLAlchemy(app)migrate = Migrate(app, db) |
實例化 Migrate 類，需要傳入 Flask 實例 app 和 SQLAlchemy創建的實例 db。
2.1 數據庫的遷移過程
先定義 User 模型類。
|
1
2
3
4
5
6
7
8
9
|
class User(db.Model): __tablename__ = 'user' id = db.Column(db.Integer, primary_key=True) user_name = db.Column(db.String) password = db.Column(db.String) |
2.1.1 創建遷移環境
在開始遷移數據之前，需要先使用下面的命令創建一個遷移環境：
|
1
|
flask db init |
遷移環境只需創建一次，創建后會在項目根目錄下生成一個 migrations 目錄，其中包含了自動生成的配置文件和遷移版本目錄。
2.1.2 生成遷移腳本
使用如下命令自動生成遷移腳本：
|
1
|
flask db migrate -m "create_table" |
-m 選項添加備注信息，執行后遷移版本目錄生成了遷移腳本。
遷移腳本內有兩個函數：
-
upgrade()：把遷移中的改動應用到數據庫中 -
downgrade()：將改動撤銷
自動生成的遷移腳本會根據模型定義和數據庫當前狀態的差異，生成upgrade()和downgrade()函數的內容，不一定完全正確，有必要再進行檢查一下。
2.1.3 更新數據庫
生成了遷移腳本后，使用flask db upgrade命令可完成對數據庫的更新。執行后即可生成數據庫及表。
如果之后我們需要改動 user 表中的字段，比如添加一個mobile字段，我們只需在 User 模型類中添加該屬性，之后執行flask db migrate -m '注釋'和flask db upgrade命令即可。
|
1
2
3
4
5
6
7
8
9
10
11
|
class User(db.Model): __tablename__ = 'user' id = db.Column(db.Integer, primary_key=True) user_name = db.Column(db.String) password = db.Column(db.String) mobile = db.Column(db.String) |
如果想要回滾遷移的話，可以執行flask db downgrade命令。
3、總結
這里只是介紹如何在 Flask 中進行數據庫遷移，關于在生產環境下，是否需要使用遷移工具或者使用何種工具進行遷移，這里不做討論，至于我的話，在生產環境中，我沒有使用過Flask-Migrate，而是選擇編寫 SQL 腳本來處理數據庫及表的更新或改動，我覺得這樣更不容易出錯，其實各有各的好處，看自己選擇。
到此這篇關于 Flask 數據庫遷移詳情的文章就介紹到這了,更多相關 Flask 數據庫遷移內容請搜索服務器之家以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持服務器之家！
原文鏈接：https://juejin.cn/post/7032081133
