flask数据库之alembic的使用

Posted 2021-04-12 Python学习交流

ORM数据库映射神器——alembic

1.介绍：

alembic是sqlalchemy的作者也即flask的作者开发的。用来做OMR模型与数据库的迁移与映射。alembic使用方式跟git有点了类似，表现在两个方面，第一个，alembic的所有命令都是以alembic开头；第二，alembic的迁移文件也是通过版本进行控制的。

2. 安装：pip install alembic。

3. 按照之前的写法，定义好你需要的ORM模型。比如定义一个叫做User的模型，这里使用的是sqlalchemy的ORM模型：

4. 初始化alembic仓库：在终端中，cd到你的项目目录中，然后执行命令alembic init alembic，创建一个名叫alembic的仓库。

5. 修改配置文件:

①在alembic.ini中设置数据库的连接，sqlalchemy.url = driver://user:pass@localhost/dbname，比如以mysql数据库为例，则配置后的代码为：

sqlalchemy.url = mysql+mysqldb://root:root@localhost/alembic_demo?charset=utf8

②为了使用模型类更新数据库，需要在env.py文件中设置target_metadata，默认为target_metadata=None。使用sys模块把当前项目的路径导入到path中：

6. 自动生成迁移文件：使用alembic revision --autogenerate -m "message"将当前模型中的状态生成迁移文件。

7. 更新数据库：使用alembic upgrade head将刚刚生成的迁移文件，真正映射到数据库中。同理，如果要降级，那么使用alembic downgrade head。

8. 修改ORM模型的代码后，重复6~7的步骤。

9. 命令和参数解释：

· init：创建一个alembic仓库。

· revision：创建一个新的版本文件。

· --autogenerate：自动将当前模型的修改，生成迁移脚本。

· -m：本次迁移做了哪些修改，用户可以指定这个参数，方便回顾。

· upgrade：将指定版本的迁移文件映射到数据库中，会执行版本文件中的upgrade函数。

· downgrade：会执行指定版本的迁移文件中的downgrade函数。

· head：代表当前的迁移脚本的版本号。

· heads：展示当前可用的heads脚本文件。

· history：列出所有的迁移版本及其信息。

· current：展示当前数据库中的版本号。