如何让Alembic自动生成外键关联？迁移缺失外键问题排查

你没生成外键的核心原因是只定义了ORM层面的relationship，但没有在数据库层面定义外键约束。relationship只是SQLAlchemy用来关联模型对象的工具，不会自动生成数据库外键，必须显式通过ForeignKey定义对应的外键列。

修正后的模型代码 #

from typing import List
from sqlalchemy import String, ForeignKey
from sqlalchemy.orm import DeclarativeBase, mapped_column, Mapped, relationship


class Base(DeclarativeBase):
    pass


class DBUnitCategory(Base):
    __tablename__ = 'unit_category'

    id: Mapped[int] = mapped_column(primary_key=True)
    name: Mapped[str] = mapped_column(String(20))

    # 属性名改为units更语义化，对应多个DBUnit实例
    units: Mapped[List["DBUnit"]] = relationship(back_populates="category")

class DBUnit(Base):
    __tablename__ = 'unit'

    id: Mapped[int] = mapped_column(primary_key=True)
    name: Mapped[str] = mapped_column(String(12))
    # 定义数据库外键列，关联unit_category表的id主键
    category_id: Mapped[int] = mapped_column(ForeignKey("unit_category.id"))
    # ORM关联属性，对应DBUnitCategory的units属性
    category: Mapped[DBUnitCategory] = relationship(back_populates="units")

关键说明 #

1. 必须导入ForeignKey并在子表（unit）中定义外键列，明确指定关联父表的主键字段
2. relationship的属性名建议语义化命名，避免和列名混淆，提升代码可读性
3. 该问题和SQLite无关，SQLite完全支持外键约束，只要模型定义正确，Alembic会自动生成对应的外键约束代码

重新执行alembic revision --autogenerate命令，生成的迁移文件就会包含外键约束内容。

内容的提问来源于stack exchange，提问作者S.Comeau