如何通过alembic.ini配置数据库迁移的连接字符集？

当然可以实现！你只需要做两个简单的修改：在alembic.ini里添加自定义配置项，然后调整env.py来动态拼接这个字符集参数到数据库URL中。

步骤1：在alembic.ini中添加自定义配置 #

打开你的alembic.ini文件，在任意位置（比如现有配置的下方）添加一个自定义配置段，就像你示例里设想的那样：

[various_config]
migrations_encoding = latin1

段名可以自定义，只要后续读取配置时对应上就行。

步骤2：修改env.py来动态拼接字符集参数 #

找到项目alembic目录下的env.py文件，找到获取数据库URL的默认代码块，大概是这样：

config = context.config
sqlalchemy_url = config.get_main_option("sqlalchemy.url")

我们需要修改这段逻辑，读取自定义配置并更新数据库URL：

推荐方案：兼容已有参数的URL处理 #

这个方法会自动识别URL中是否已存在charset参数，存在则替换，不存在则添加，更健壮：

from urllib.parse import urlparse, urlunparse

config = context.config
sqlalchemy_url = config.get_main_option("sqlalchemy.url")

# 读取自定义编码配置，没配置的话默认用utf8
migrations_encoding = config.get_section_option("various_config", "migrations_encoding", fallback="utf8")

# 解析URL结构
parsed_url = urlparse(sqlalchemy_url)
# 拆分并过滤已有查询参数
query_params = parsed_url.query.split("&")
filtered_params = [p for p in query_params if not p.startswith("charset=")]
# 加入新的charset参数
filtered_params.append(f"charset={migrations_encoding}")
new_query = "&".join(filtered_params)

# 重新组装完整URL
new_sqlalchemy_url = urlunparse(
    (
        parsed_url.scheme,
        parsed_url.netloc,
        parsed_url.path,
        parsed_url.params,
        new_query,
        parsed_url.fragment
    )
)

# 将修改后的URL传入配置
context.configure(
    url=new_sqlalchemy_url,
    # 保留其他原有配置项...
)

简化方案：适用于确定URL无charset参数的场景 #

如果你的数据库URL一开始就没有charset参数，也可以用更简洁的写法：

config = context.config
sqlalchemy_url = config.get_main_option("sqlalchemy.url")
migrations_encoding = config.get_section_option("various_config", "migrations_encoding", fallback="utf8")

# 根据是否已有查询参数选择拼接方式
if "?" in sqlalchemy_url:
    new_sqlalchemy_url = f"{sqlalchemy_url}&charset={migrations_encoding}"
else:
    new_sqlalchemy_url = f"{sqlalchemy_url}?charset={migrations_encoding}"

context.configure(
    url=new_sqlalchemy_url,
    # 保留其他原有配置项...
)

完成这两步后，每次运行Alembic迁移命令时，都会自动把你在alembic.ini里配置的migrations_encoding值，注入到数据库连接URL的charset参数中，完全符合你的需求。

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