SQLAlchemy对接Snowflake自增主键写入报NULL identity key错误

两个报错的根因非常明确：

• 报NULL identity key是因为SQLAlchemy没有识别到Snowflake表的自增主键是数据库侧自动生成的，提交flush时拿不到生成的主键值，直接判定主键为空抛出错误。
• 手动加Sequence("id_seq")后报invalid identifier 'ID_SEQ.NEXTVAL'是因为Snowflake的表级autoincrement是独立的IDENTITY属性，和独立序列对象不互通。你配置Sequence后，SQLAlchemy会主动查询该序列的下一个值作为主键，但你从来没有创建过这个名为id_seq的独立序列，自然会报不存在的错误。就算手动建了这个序列，也会和表自带的自增逻辑冲突，极易出现主键重复问题。

1. 修正模型主键定义 #

优先使用SQLAlchemy 1.4+版本支持的标准Identity语法定义自增主键，Snowflake方言对该语法的适配最完善，不会生成多余的序列查询逻辑：

from sqlalchemy import Column, String, BigInteger, Identity
# 其余原有导入保持不变

class Location(Base):
    __tablename__ = "location"
    # Snowflake的NUMBER(38,0)数值范围超过普通Integer，用BigInteger映射更稳妥
    # Identity配置和Snowflake原生autoincrement逻辑完全对齐，不需要额外创建序列
    id = Column(BigInteger, Identity(start=1, increment=1), primary_key=True)
    address = Column(String)
    latitude = Column(String, unique=True, nullable=False)
    longitude = Column(String, unique=True, nullable=False)

    # 以下关系定义不需要修改
    buildings = relationship("Building", back_populates="location")
    quotes = relationship("Quote", back_populates="location")
    binds = relationship("Bind", back_populates="location")

如果你用的SQLAlchemy版本低于1.4，不支持Identity语法，可以用以下配置替代，显式告诉SQLAlchemy该列是自增的服务端生成值：

id = Column(BigInteger, primary_key=True, autoincrement=True)

2. 检查依赖版本 #

先升级相关依赖到最新稳定版，老版本的snowflake-sqlalchemy对自增列的主键返回逻辑存在已知bug：

pip install --upgrade snowflake-sqlalchemy sqlalchemy alembic

3. 对齐迁移脚本 #

如果你用Alembic管理迁移，修改模型后生成新的迁移脚本，检查生成的建表/修改列语句，确保id列带有autoincrement属性，不要生成任何创建id_seq序列的代码，最终生成的LOCATION表结构和你给出的示例一致即可。

你原来的create_location插入逻辑不需要做任何修改，插入时不需要手动传入id值，commit后SQLAlchemy会自动从Snowflake返回的结果中拿到自动生成的主键值，赋值到location实例上，不会再出现空主键的报错。

内容的提问来源于stack exchange，提问作者Henry Alexander Polindara