在Redshift中使用SQLAlchemy创建含BOOLEAN列的表失败

成功连接Redshift集群后，尝试用SQLAlchemy的声明式基类确保目标表存在，运行代码后出现语法错误。

原始代码 #

from sqlalchemy.engine import URL, create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Boolean, SMALLINT, Column


Base = declarative_base()


class Table(Base):
    __tablename__ = "a_table"

    id = Column(SMALLINT, primary_key=True)
    a_boolean_column = Column(Boolean)


# example values
engine = create_engine(
    URL.create(
        drivername="postgresql+psycopg2",
        username="username",
        password="password",
        host="127.0.0.1",
        port=5439,
        database="dev",
    )
)

base.metadata.create_all(engine)

错误信息 #

sqlalchemy.exc.ProgrammingError: (psycopg2.errors.SyntaxError) syntax error at or near "a_boolean_column"
LINE 12:  a_boolean_column BOOLEAN, 
          ^

[SQL: 
CREATE TABLE a_table (
        id INTEGER NOT NULL, 
        a_boolean_column BOOLEAN, 
        PRIMARY KEY (id),
)

]

问题分析 #

从生成的SQL可以看到，PRIMARY KEY (id)后面多了一个逗号，这不符合Redshift的SQL语法规范。出现这个问题的核心原因是使用了通用的postgresql+psycopg2驱动，该驱动生成的SQL允许末尾逗号，但Redshift不支持这种写法。另外代码中存在笔误：定义的基类是大写的Base，但最后调用时误用了小写的base。

解决方法 #

1. 切换为Redshift专属驱动：将drivername改为redshift+psycopg2，让SQLAlchemy使用sqlalchemy-redshift提供的方言生成兼容Redshift的SQL，避免多余的末尾逗号。
2. 修正大小写笔误：将base.metadata.create_all(engine)改为Base.metadata.create_all(engine)，确保调用正确的基类元数据。

修正后的代码 #

from sqlalchemy.engine import URL, create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Boolean, SMALLINT, Column


Base = declarative_base()


class Table(Base):
    __tablename__ = "a_table"

    id = Column(SMALLINT, primary_key=True)
    a_boolean_column = Column(Boolean)


# example values
engine = create_engine(
    URL.create(
        drivername="redshift+psycopg2",  # 改为Redshift专属驱动
        username="username",
        password="password",
        host="127.0.0.1",
        port=5439,
        database="dev",
    )
)

Base.metadata.create_all(engine)  # 修正大小写笔误

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