使用alembic为SQLAlchemy迁移数据

用惯了django的迁移命令后, 发现SQLAlchemy默认没有对应的迁移命令, 但是SQLAlchemy作者为SQLAlchemy开发了迁移工具: Alembic, 其官方文档见: Alembic Tutorial

使用这个工具的需要前提安装数据库链接库,如:pymysql

安装#

我这里使用virtualenv创建虚拟环境, 所以使用pip命令安装:

fast →pip install alembicrestart ↻

初始化项目并配置alembic#

在项目的根目录执行初始化命令:

fast →alembic init alembicrestart ↻

上述命令的第二个alembic指的是存放配置和迁移历史等数据的目录, 设置自己喜欢的目录名

此时alembic会生成以下目录与文件:

<project-name>
+--- alembic
|   +--- env.py
|   +--- README
|   +--- script.py.mako
|   +--- versions
|   
+--- alembic.ini
 

接下来需要配置alembic.ini和env.py:

1. alembic.ini配置数据库链接
   修改sqlalchemy.url配置, 这个和创建engine的url是一样的, 如:

   sqlalchemy.url = mysql+pymysql://demo_user:demo123456@127.0.0.1:3306/demo_db
   

2. env.py配置metadata

   from orm.database import Base
    
   target_metadata = Base.metadata
   

   这里的Base是BaseModel即由declarative_base创建的基类

第一次执行迁移命令#

执行: alembic upgrade head命令,
该命令指的是: 将数据库升级(切换)到最新版本, 该过程会创建没有表
你可以类比git

执行迁移命令#

假如不是第一次执行迁移命令了, 你可以执行以下命令:
alembic revision --autogenerate -m "init db"

-m指定此次迁移的注释, alembic会根据这个注释在versions目录生成一个py文件, 用于记录操作

其他命令#

查看迁移历史#

fast →alembic history --verboseRev: fe3dcb76f464 (head)Parent: 8bb0bf479f9dPath: E:\code\Project\alembic\versions\fe3dcb76f464_测试.py    Create Date: 2022-02-19 21:59:22.473262 Rev: 8bb0bf479f9dParent: <base>Path: E:\code\Project\alembic\versions\8bb0bf479f9d_init_db.py     init db     Revision ID: 8bb0bf479f9d    Revises:    Create Date: 2022-02-19 21:22:46.521491 restart ↻

我们主要需要关注head即头指针的位置和 Revision ID即该迁移的哈希值, 对应着该版本的指针

查看当前指针的指向#

fast →alembic current2022-02-19 22:12:39,749 INFO sqlalchemy.engine.Engine BEGIN (implicit)...2022-02-19 22:12:39,755 INFO sqlalchemy.engine.Engine COMMITINFO  [alembic.runtime.migration] Context impl SQLiteImpl.INFO  [alembic.runtime.migration] Will assume non-transactional DDL.8bb0bf479f9drestart ↻

最后一行的值为当前版本的指向

切换到对应的迁移版本#

这里对应向上和向下切换
通过Parent确定此次迁移的父版本

# 向上切换
alembic upgrade <version>
 
# 向下切换
alembic downgrade <version>

如:

fast →alembic downgrade 8bb0bf479f9dalembic upgrade fe3dcb76f464restart ↻

当然也可以使用命令直接切换到最新/老的版本:

# 
fast →切换到最新版本alembic upgrade headalembic downgrade base restart ↻

生成sql文件#

在某些不适合在线更新的情况，可以采用生成sql脚本的形式，进行离线更新：

alembic upgrade <version> --sql > migration.sql

fast →alembic upgrade 8bb0bf479f9d--sql > migration.sqlrestart ↻

从特定起始版本生成sql脚本：

alembic upgrade <vsersion>:<vsersion> --sql > migration.sql

fast →alembic upgrade 8bb0bf479f9d:fe3dcb76f464--sql > migration.sql restart ↻