如何通过SQLAlchemy自动更新ORM管理的数据库表结构？

metadata.create_all(engine)的局限性在于仅会创建不存在的表，不会修改已有表的结构，所以你添加新列后旧数据库不会自动同步。要实现表结构的可控自动更新，推荐使用SQLAlchemy官方的迁移工具Alembic，以下是具体步骤：

一、安装Alembic #

pip install alembic

二、初始化Alembic项目 #

在项目根目录下执行：

alembic init alembic

执行后会生成alembic目录和alembic.ini配置文件。

三、配置Alembic #

1. 打开alembic.ini，找到sqlalchemy.url配置项，替换成你的数据库连接字符串：

sqlalchemy.url = sqlite:///database.db

2. 打开alembic/env.py，找到target_metadata变量，修改为指向你的模型Base的metadata：

# 导入你定义Base的模块（替换成实际模块名）
from your_model_module import Base

target_metadata = Base.metadata

四、生成迁移脚本 #

当你修改模型（比如添加新列）后，执行以下命令生成迁移脚本：

alembic revision --autogenerate -m "add new column to tests table"

这条命令会对比当前模型与数据库的结构差异，自动生成对应的SQL迁移脚本，存放在alembic/versions目录下。

五、执行迁移更新数据库 #

生成脚本后，执行以下命令将变更应用到数据库：

alembic upgrade head

数据库表结构会同步到最新的模型定义。

开发环境临时方案（不推荐生产使用） #

如果仅在开发阶段快速测试，不想用Alembic，可以借助sqlalchemy_utils工具先删除旧表再重建，但这种方式会丢失数据，仅适合开发场景：

from sqlalchemy_utils import database_exists, create_database, drop_database

if database_exists(engine.url):
    drop_database(engine.url)
create_database(engine.url)
metadata.create_all(engine)

注意：生产环境绝对不能使用该方式，会丢失所有数据！

内容的提问来源于stack exchange，提问作者kotek