SQLAlchemy中MySQL表tenant_id主键与tenant_index自增列配置问题求助

问题根源 #

你遇到的问题是：多数数据库（如MySQL）的自增特性默认与主键绑定，当你移除tenant_index的primary_key=True后，SQLAlchemy不会自动将其标记为自增列，导致插入数据时数据库找不到该字段的默认值，从而抛出Field 'tenant_index' doesn't have a default value错误。

正确实现方式 #

需要显式配置tenant_index为自增列，同时保留tenant_id作为主键，以下是不同场景的代码示例：

1. 多数据库兼容方案（使用Sequence） #

如果项目需要兼容多种数据库（如PostgreSQL、MySQL等），可以用Sequence定义自增序列：

from sqlalchemy import Column, Integer, String, Sequence
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class Tenant(Base):
    __tablename__ = 'tenants'
    
    tenant_id = Column(String(50), primary_key=True)  # 自定义主键
    tenant_index = Column(Integer, Sequence('tenant_index_seq'), nullable=False)

创建表后，插入数据时无需手动指定tenant_index，数据库会自动通过序列生成递增数值。

2. MySQL专属优化方案 #

针对MySQL，可直接用autoincrement=True标记字段，但需注意非主键自增字段必须添加唯一约束：

from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class Tenant(Base):
    __tablename__ = 'tenants'
    
    tenant_id = Column(String(50), primary_key=True)
    tenant_index = Column(Integer, autoincrement=True, nullable=False, unique=True)

MySQL要求非主键的自增字段必须唯一，因此unique=True是必要配置。

插入数据示例 #

配置完成后，插入数据只需指定tenant_id即可：

new_tenant = Tenant(tenant_id="TENANT_001")
session.add(new_tenant)
session.commit()

迁移注意事项 #

如果使用Alembic进行数据库迁移，需确保迁移脚本正确生成了自增列的定义，避免手动修改脚本导致配置失效。

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