我們?cè)谏?jí)系統(tǒng)的時(shí)候，經(jīng)常碰到需要更新服務(wù)器端數(shù)據(jù)結(jié)構(gòu)等操作，之前的方式是通過(guò)手工編寫(xiě)alter sql腳本處理，經(jīng)常會(huì)發(fā)現(xiàn)遺漏，導(dǎo)致程序發(fā)布到服務(wù)器上后無(wú)法正常使用。

現(xiàn)在我們可以使用Flask-Migrate插件來(lái)解決之，F(xiàn)lask-Migrate插件是基于Alembic，Alembic是由大名鼎鼎的SQLAlchemy作者開(kāi)發(fā)數(shù)據(jù)遷移工具。

具體操作如下：

1. 安裝Flask-Migrate插件

$ pip install Flask-Migrate

db = SQLAlchemy(app)
migrate = Migrate(app, db)

manager = Manager(app)
manager.add_command('db', MigrateCommand)

3. 初始化

$ python app.py db init

使用Flask-Migrate遷移數(shù)據(jù)庫(kù)
隨著開(kāi)發(fā)進(jìn)度不斷向前，你會(huì)發(fā)現(xiàn)你的數(shù)據(jù)庫(kù)模型需要更改，而當(dāng)這種情況發(fā)生時(shí)需要更新數(shù)據(jù)庫(kù)。

Flask-SQLAlchemy只有當(dāng)數(shù)據(jù)庫(kù)表不存在了才從模型創(chuàng)建它們，所以更新表的唯一途徑就是銷毀舊的表，當(dāng)然這將導(dǎo)致所有數(shù)據(jù)庫(kù)中的數(shù)據(jù)丟失。

有個(gè)更好的解決方案就是使用數(shù)據(jù)庫(kù)遷移框架。和源碼版本控制工具跟蹤更改源碼文件一樣，數(shù)據(jù)庫(kù)遷移框架跟蹤更改數(shù)據(jù)庫(kù)模型，然后將增量變化應(yīng)用到數(shù)據(jù)庫(kù)中。

SQLAlchemy的主要開(kāi)發(fā)人員寫(xiě)了一個(gè)Alembic遷移框架，但我們不直接使用Alembic，F(xiàn)lask應(yīng)用可以使用Flask-Migrate擴(kuò)展，一個(gè)集成了Flask-Script來(lái)提供所有操作命令的輕量級(jí)Alembic包。

4. 創(chuàng)建遷移倉(cāng)庫(kù)

首先，F(xiàn)lask-Migrate必須已經(jīng)安裝到虛擬環(huán)境中：

(venv) $ pip install flask-migrate

下面展示擴(kuò)展如何初始化：

from flask.ext.migrate import Migrate, MigrateCommand 

# ...

migrate = Migrate(app, db)
manager.add_command('db', MigrateCommand)

為了可以使用數(shù)據(jù)庫(kù)遷移命令，F(xiàn)lask-Migrate提供MigrateCommand類來(lái)連接Flask-Script的manager對(duì)象。在這個(gè)示例中使用db來(lái)連接到命令。

在數(shù)據(jù)庫(kù)遷移可以維護(hù)之前，必須通過(guò)init子命令來(lái)創(chuàng)建一個(gè)遷移庫(kù)：

(venv) $ python hello.py db init

 Creating directory /home/flask/flasky/migrations...done
 Creating directory /home/flask/flasky/migrations/versions...done
 Generating /home/flask/flasky/migrations/alembic.ini...done
 Generating /home/flask/flasky/migrations/env.py...done
 Generating /home/flask/flasky/migrations/env.pyc...done
 Generating /home/flask/flasky/migrations/README...done
 Generating /home/flask/flasky/migrations/script.py.mako...done
 Please edit configuration/connection/logging settings in
 '/home/flask/flasky/migrations/alembic.ini' before proceeding.

這個(gè)命令創(chuàng)建一個(gè)migrations文件夾，里面存放了所有遷移腳本。

建議：如果你有克隆在GitHub上的應(yīng)用程序，你現(xiàn)在可以運(yùn)行g(shù)it checkout 5c來(lái)切換到這個(gè)版本的應(yīng)用程序。

5. 創(chuàng)建遷移腳本

在Alembic，數(shù)據(jù)庫(kù)遷移工作由遷移腳本完成。這個(gè)腳本有兩個(gè)函數(shù)，分別叫做upgrade()和downgrade()。upgrade()函數(shù)實(shí)施數(shù)據(jù)庫(kù)更改，是遷移的一部分，downgrade()函數(shù)則刪除它們。通過(guò)添加和刪除數(shù)據(jù)庫(kù)變化的能力，Alembic可以重新配置數(shù)據(jù)庫(kù)從歷史記錄中的任何時(shí)間點(diǎn)。

Alembic遷移可以分別使用revision和migrate命令手動(dòng)或自動(dòng)創(chuàng)建。手動(dòng)遷移通過(guò)由開(kāi)發(fā)人員使用Alembic的Operations對(duì)象指令實(shí)現(xiàn)的空upgrade()和downgrade()函數(shù)創(chuàng)建遷移框架腳本。另一方面，自動(dòng)遷移通過(guò)尋找模型定義和數(shù)據(jù)庫(kù)當(dāng)前狀態(tài)間的不同為upgrade()和downgrade()生成代碼。

警告：自動(dòng)遷移并不總是準(zhǔn)確的，可以忽略一些細(xì)節(jié)。所以應(yīng)該經(jīng)常審查一下自動(dòng)生成的遷移腳本。
migrate子命令創(chuàng)建自動(dòng)遷移腳本：

(venv) $ python hello.py db migrate -m "initial migration"

INFO [alembic.migration] Context impl SQLiteImpl.
INFO [alembic.migration] Will assume non-transactional DDL.
INFO [alembic.autogenerate] Detected added table 'roles'
INFO [alembic.autogenerate] Detected added table 'users'
INFO [alembic.autogenerate.compare] Detected added index
'ix_users_username' on '['username']'
 Generating /home/flask/flasky/migrations/versions/1bc
 594146bb5_initial_migration.py...done

建議：如果你有克隆在GitHub上的應(yīng)用程序，你現(xiàn)在可以運(yùn)行g(shù)it checkout 5c來(lái)切換到這個(gè)版本的應(yīng)用程序。注意，你不需要為這個(gè)應(yīng)用生成migrations，所有的遷移腳本都包含在版本庫(kù)中。
6. 更新數(shù)據(jù)庫(kù)

一旦遷移腳本被審查且接受，就可以使用db upgrade命令更新到數(shù)據(jù)庫(kù)中：

(venv) $ python hello.py db upgrade

INFO [alembic.migration] Context impl SQLiteImpl.
INFO [alembic.migration] Will assume non-transactional DDL.
INFO [alembic.migration] Running upgrade None -> 1bc594146bb5, initial migration

第一次遷移實(shí)際上相當(dāng)于調(diào)用db.create_all()，但在后續(xù)遷移中，upgrade命令對(duì)表實(shí)施更新操作但不影響表中的內(nèi)容。

最新評(píng)論