FastApi框架-Alembic实现数据迁移

FastApi框架-Alembic实现数据迁移

安装Alembic

pip install alembic

初始化项目

进入项目根目录，执行如下命令

alembic init alembic

init后面的alembic是迁移文件夹的名字，可以随便定义，一般默认alembic，执行成功后，项目目录下会多出alembic文件夹和alembic.ini文件

修改alembic.ini

找到sqlalchemy.url，修改为自己的数据库连接

sqlalchemy.url = mysql+pymysql://root:12345678@127.0.0.1:3306/test_craft

修改env.py

找到target_metadata = None，进行更改

from model import ModelBase
target_metadata = ModelBase.metadata

此处的ModelBase是ModelBase = declarative_base()

生成迁移文件

执行如下命令：

alembic revision --autogenerate -m "test"

执行完成，可以在alembic/versions目录看到对应的py文件

变更数据库

执行如下命令

alembic upgrade head

执行完成，查看数据库，会发现已经按照上述迁移文件创建成功users表