如何重写Alembic的Revision ID生成函数？改用时间戳方案

完全可以通过猴子补丁的方式替换Alembic默认的随机revision ID生成逻辑，切换回你熟悉的时间戳方案。下面是具体的实现步骤，确保能覆盖所有生成revision的场景：

实现猴子补丁 #

找到你项目里的Alembic配置文件env.py，在文件开头（建议在from alembic import context之后）添加以下代码：

import time
from alembic import util

def timestamp_based_rev_id():
    # 生成和旧版本Alembic一致的十六进制时间戳ID
    return hex(int(time.time()))[2:]

# 替换默认的随机ID生成函数
util.rev_id = timestamp_based_rev_id

为什么这么做？ #

Alembic的util.rev_id函数是全局负责生成revision ID的入口——不管是通过命令行执行alembic revision，还是通过代码调用Alembic的API创建revision，最终都会调用这个函数。直接替换这个函数就能让时间戳ID的生成逻辑全局生效。

验证效果 #

保存修改后，运行一次revision生成命令测试：

alembic revision -m "test timestamp-based id"

打开生成的revision文件，你会看到ID已经变成了基于当前时间戳的十六进制字符串，和你之前使用的格式完全一致。

注意事项 #

• 确保这段补丁代码在Alembic执行任何生成ID的逻辑前被加载，env.py是最理想的位置，因为Alembic启动时会优先加载这个文件。
• 这个修改不会影响已有的revision ID，只会作用于新生成的revision。

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