SQLalchemy 在设置角色时不提交更改

Posted 2023-03-14

【中文标题】SQLalchemy 在设置角色时不提交更改

【英文标题】：SQLalchemy not committing changes when setting role

【发布时间】：2019-10-03 00:14:46

【问题描述】：

我正在使用 sqlalchemy 引擎创建表，但即使我的创建语句执行没有错误，当我尝试预先设置角色时，这些表也不会显示在数据库中。

url = 'postgresql://:@:/'
url = url.format(user, password, host, port, db)

engine = sqlalchemy.create_engine(url)

# works fine
engine.execute("CREATE TABLE testpublic (id int, val text); \n\nINSERT INTO testpublic VALUES (1,'foo'), (2,'bar'), (3,'baz');")
r = engine.execute("select * from testpublic")
r.fetchall() # returns expected tuples
engine.execute("DROP TABLE testpublic;")

# appears to succeed/does NOT throw any error
engine.execute("SET ROLE read_write; CREATE table testpublic (id int, val text);")

# throws error "relation testpublic does not exist"
engine.execute("select * FROM testpublic")

对于上下文，我使用的是 python 3.6、sqlalchemy 版本 1.2.17 和 postgres 11.1，并且角色“read_write”绝对存在，并且拥有公开创建表的所有必要权限（我在运行上面的确切序列时没有问题pgadmin)。

有谁知道为什么会这样以及如何解决？

【参考方案1】：

这里的问题是 sqlalchemy 如何决定在每个语句之后发出提交。

如果将文本传递给engine.execute，sqlalchemy 将尝试使用以下正则表达式确定文本是 DML 还是 DDL。 You can find it in the sources here

AUTOCOMMIT_REGEXP = re.compile(
    r"\s*(?:UPDATE|INSERT|CREATE|DELETE|DROP|ALTER)", re.I | re.UNICODE
)

这仅检测位于文本开头的单词，忽略任何前导空格。因此，当您第一次尝试# works fine 时，第二个示例未能识别出需要在执行语句后发出提交，因为第一个单词是SET。

相反，sqlalchemy 发出回滚，所以它@98​​7654327@。

最简单的解决方案是手动提交。

示例：

engine.execute("SET ROLE read_write; CREATE table testpublic (id int, val text); COMMIT;")

或者，将 sql 包装在 text 中并设置 autocommit=True、as shown in the documentation

stmt = text('set role read_write; create table testpublic (id int, val text);').execution_options(autocommit=True)
e.execute(stmt)