SQL Alchemy无法在事务块内运行

Question

我正在尝试从python脚本在redshift中运行查询，但出现错误：

sqlalchemy.exc.InternalError: (psycopg2.InternalError) ALTER EXTERNAL TABLE cannot run inside a transaction block

这是我的代码：

engine = create_engine(SQL_ENGINE % urlquote(REDSHIFT_PASS))
partition_date = (date.today() - timedelta(day)).strftime("%Y%m%d")
query = """alter table  {table_name} add partition (dt={date_partition}) location 's3://dft-dwh-files/raw_data/google_analytics/revenue_per_channel/{date_partition}/';""".format(date_partition=partition_date,table_name=table_name)
conn = engine.connect()
conn.execute(query).execution_options(autocommit=True)

我怎样才能解决这个问题？

Answer 1

对于PostgreSQL，您需要将隔离级别设置为AUTOCOMMIT，而不是SQLAlchemy autocommit：

conn.execution_options(isolation_level="AUTOCOMMIT").execute(query)

Answer 2

这个解决方案对我有用，因为使用 sqlalchemy :session not :conn as @univerio answer

我在这里引用他们的回答

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
engine = create_engine('postgresql+psycopg2://USER:PASSWORD@127.0.0.1:5432/DB_OR_TEMPLATE')
session = sessionmaker(bind=engine)()
session.connection().connection.set_isolation_level(0)
session.execute('CREATE DATABASE test')
session.connection().connection.set_isolation_level(1)

如果你没有任何数据库，你应该使用 template1

"""Isolation level values."""
ISOLATION_LEVEL_AUTOCOMMIT     = 0
ISOLATION_LEVEL_READ_COMMITTED = 1
ISOLATION_LEVEL_SERIALIZABLE   = 2