AlloyDB使用SQLAlchemy创建数据库时事务块错误的解决求助

PostgreSQL（包括AlloyDB）的CREATE DATABASE命令不允许在事务块内执行，而SQLAlchemy默认会为连接开启事务，这就是你遇到错误的核心原因。以下是几种可行的解决办法：

方法1：使用底层psycopg2连接手动执行 #

绕过SQLAlchemy的事务管理，直接用psycopg2原生连接执行创建语句：

from sqlalchemy import create_engine

# 先连接到AlloyDB的默认数据库（比如postgres）
engine = create_engine("postgresql+psycopg2://<用户名>:<密码>@<主机>:<端口>/postgres")

# 获取原生连接并手动控制事务
raw_conn = engine.raw_connection()
try:
    cur = raw_conn.cursor()
    # 执行创建数据库语句
    cur.execute("CREATE DATABASE <你的数据库名>;")
    cur.close()
    raw_conn.commit()
finally:
    raw_conn.close()

方法2：配置SQLAlchemy连接为自动提交模式 #

在创建引擎时设置isolation_level="AUTOCOMMIT"，让执行语句时不被包裹在事务块中：

from sqlalchemy import create_engine
from sqlalchemy.sql import text

engine = create_engine(
    "postgresql+psycopg2://<用户名>:<密码>@<主机>:<端口>/postgres",
    isolation_level="AUTOCOMMIT"
)

with engine.connect() as conn:
    conn.execute(text("CREATE DATABASE <你的数据库名>;"))

方法3：Alembic中的特殊处理 #

如果要在Alembic流程中创建数据库，不要把CREATE DATABASE放在迁移脚本里（迁移脚本针对已有数据库），建议单独写一个初始化脚本，用上面两种方法之一完成数据库创建，之后再运行Alembic迁移。

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