如何在Alembic中按需启用或停用自动生成的Drop操作？

首先找到Alembic的迁移脚本模板文件，默认路径是alembic/templates/generic/script.py.mako（如果用了自定义模板，就找你指定的模板路径）。

打开这个mako文件，找到生成DROP TABLE操作的代码块——大概是这样的：

% for table_name in drop_tables:
    op.drop_table('${table_name}')
% endfor

给这段代码加个条件判断，只在你指定的场景下才生成drop操作。比如通过配置项来控制：

% if context.config.get_main_option("alembic.allow_drops", "false") == "true":
    % for table_name in drop_tables:
        op.drop_table('${table_name}')
    % endfor
% endif

同理，如果你还要控制DROP COLUMN、DROP CONSTRAINT这类删除操作，找到模板里对应的代码段（比如drop_columns、drop_constraints循环），同样套上这个条件判断。

接下来配置开关：

1. 在alembic.ini里加一行默认配置，比如：

alembic.allow_drops = false

这样默认不会生成任何drop操作，避免误删。

2. 当你需要在特定场景（比如开发环境）允许生成drop操作时，要么临时修改ini里的值，要么用命令行参数覆盖：

alembic revision --autogenerate -x allow_drops=true

另外，修改模板前建议先备份原文件，防止后续Alembic版本更新覆盖你的自定义修改。

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