啥是數(shù)據(jù)遷移

在我們平時的開發(fā)過程中，經(jīng)常需要對一些數(shù)據(jù)進行調(diào)整。一般會有以下幾種場景:

1.需要新的數(shù)據(jù)表

我們的接口自動化平臺雖然已經(jīng)較為完善了，但難免會繼續(xù)迭代一些新的功能，假設我們需要做一個訂閱用例的功能。

大體想一下就可以知道，訂閱用例以后這個數(shù)據(jù)得持久化（即入庫），這樣我在查詢誰訂閱了這條用例的時候，就能獲取到訂閱人，訂閱時間等數(shù)據(jù)。

這也就意味著我們需要一張訂閱表，里面至少得有訂閱人和訂閱的id，以及訂閱時間這3個要素。體現(xiàn)在數(shù)據(jù)庫通俗點說就需要:

// 創(chuàng)建訂閱表, 后面的省略
create table... 

由于業(yè)務的變動，導致新的數(shù)據(jù)表誕生。

2.需要對現(xiàn)有表結(jié)構(gòu)進行調(diào)整

當我們的訂閱表完成以后，有的同學就發(fā)現(xiàn)了，這個訂閱好像不能取消，所以我們此時可能需要一個新的字段: isValid,這個字段用來判斷用戶是否取消訂閱了這個用例，如果我訂閱錯了，或者嫌消息太多，想取消訂閱，那還是得滿足需求的。

包括新增字段/修改字段/刪除字段，這些都會對數(shù)據(jù)表產(chǎn)生影響，導致我們需要改動數(shù)據(jù)庫。

回到ORM

我們目前采用sqlalchemy作為我們的orm，如果只需要修改Python的Model類（操作字段就加在Model類里面操作）該多好。這樣的話，我們依然不需要去寫很基礎的sql語句，就能達到修改表結(jié)構(gòu)的目的。而這個，就是我們今天要講的數(shù)據(jù)遷移。因為數(shù)據(jù)需要發(fā)生變化，orm與數(shù)據(jù)庫的邏輯對不上號了，所以我們需要遷移。

遷移手段

目前市面上，關(guān)于Django(自帶orm)和Flask這塊都很成熟，django因為有自帶的orm顯得更牛逼，在manage.py里面自帶了migrate（遷移）的命令。

而我們今天要講的fastapi，由于不像django那么全面，所以我們采用alembic（sqlalchemy作者編寫）來幫助我們完成數(shù)據(jù)遷移操作。

如果你也用的fastapi+sqlalchemy，那我們就一起來耍耍i！

注: sqlalchemy只自帶create_all(建立全部表)的功能

安裝alembic

大家采用虛擬環(huán)境和全局安裝都可以，我的建議是全局安裝，因為我們可能會在多個項目使用它。

pip install alembic 

初始化項目

我們在python項目的根目錄輸入以下命令:

alembic init alembic 

alembic是我們剛才安裝的工具，init則是初始化的意思，后面的alembic則是遷移文件夾的名字，一般我們會默認叫alembic，以便于其他人一眼就知道是干嘛的。

修改alembic.ini

執(zhí)行完成之后你會發(fā)現(xiàn)根目錄多了個ini配置文件和alembic文件夾，我們需要稍微修改下配置文件:

將alembic.ini中的sqlalchemy.url改為你數(shù)據(jù)庫的jdbc連接地址，以我的為例:

修改alembic/env.py

首先我們找到里面的target_metadata變量，默認是None。接著在target_metadata = None上方加入如下代碼:

import sys
from os.path import abspath, dirname
sys.path.append(dirname(dirname(abspath(__file__)))) 

然后我們需要引入我們的model目錄，由于在pity里面，最后的初始化建表工作都是在curd目錄進行的:

所以我這邊是引入crud里面的Base。這個Base是啥玩意呢？

我們使用sqlalchemy，都需要引入各種model，這些model最終都會被加入到Base.metadata，這樣sqlalchemy就知道你有哪些表需要處理了。

我們繼續(xù)修改env.py，也就是告訴alembic你有哪些數(shù)據(jù)表（上文說的， 數(shù)據(jù)表都在Base.metadata）。

# 注意這個地方是要引入模型里面的Base,不是connect里面的
from app.crud import Base
# 告訴alembic 你的表數(shù)據(jù)在哪
target_metadata = Base.metadata 

開始生成遷移工作

要注意，我們配置這么多東西是為了讓alembic知道你的model都在哪，你的數(shù)據(jù)庫怎么連，這樣它才能去對比差異并生成結(jié)果。

alembic revision --autogenerate -m "test" 

稍作等待，我們可以在alembic/versions目錄看到對應的py文件:

里面會有drop_column(刪除字段)這樣的操作信息。但要注意，這并沒有真正修改數(shù)據(jù)庫。

變更數(shù)據(jù)庫

alembic upgrade head 

執(zhí)行上述命令，alembic就會根據(jù)你當前的版本（應該是你剛才生成的version的py腳本）去執(zhí)行數(shù)據(jù)庫變更操作。

這樣，一個簡單的遷移工作就完成了。接著我們聊聊注意事項。

FAQ

Q. 版本py腳本無變更信息出現(xiàn)

A. 請檢查你的數(shù)據(jù)庫是否真有變更，檢查你的數(shù)據(jù)庫url是否正確，檢查你的model是否引入正確，如果操作都沒問題，可以刪除alembic目錄和ini，重復上述操作（我昨晚就是這樣的）

Q. 為什么字段重命名沒有產(chǎn)生變更

A. 這玩意只校驗了新增/修改/刪除字段，這里的修改指的是字段的nullable這種修改，所以改字段名它不會檢測。

Q. 我數(shù)據(jù)庫里面有表沒有定義到model，為啥變更給俺把表刪除了？

A. 這個問題我也發(fā)現(xiàn)了，應該是有什么配置可以配置不刪除未找到的表，但我目前還沒有去研究，有后續(xù)會在底部留言。

最后，這個玩意相對比較雞肋，建議不要大批量變更?？梢灶l率高一點，比如有一點點改動就用它變更一下，而不要在史詩級改動的時候使用它。我個人的整體感受是，不太好用，但勉強能用。（因為暫時沒有發(fā)現(xiàn)更合適的）

到此這篇關(guān)于python FastApi實現(xiàn)數(shù)據(jù)表遷移流程詳解的文章就介紹到這了,更多相關(guān)python數(shù)據(jù)表遷移內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論