如何在SQLAlchemy 2.0中用dataclass_transform标记必填字段？

你遇到的核心问题是：当给Mapped类型字段赋值mapped_column(...)时，类型检查器会默认认为该字段有默认值，从而标记为可选参数，但实际上数据库层面nullable=False要求它是必填项。以下是两种可靠的解决方式：

方法1：用Annotated+Required（Python 3.10+推荐） #

从typing模块导入Required，将字段类型包装为Annotated[Mapped[int], Required]，明确告诉类型检查器这个字段是必填项，即使已经通过mapped_column定义了数据库属性。

修改后的代码示例：

from typing import Annotated, Required
from sqlalchemy import ForeignKey
from sqlalchemy.orm import Mapped, mapped_column, DeclarativeBase, Session
import typing_extensions

@typing_extensions.dataclass_transform(kw_only_default=True)
class Base(DeclarativeBase):
    pass

class External(Base):
    __tablename__ = "external"
    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)

class TestModel(Base):
    __tablename__ = "test_table"
    name: Mapped[str]
    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    external_id: Annotated[Mapped[int], Required] = mapped_column(
        ForeignKey("external.id"), nullable=False
    )

def test_test_model(session: Session) -> None:
    TEST_NAME = "name"
    external = External()
    session.add(external)
    session.commit()
    
    model3 = TestModel(name=TEST_NAME) # 此处会被Intellisense报错，提示缺少external_id参数
    session.add(model3)
    session.commit()

修改后，类型检查器会像识别name字段一样，把external_id标记为必填，提前在编码阶段发现问题，避免运行时数据库报错。

方法2：拆分类型声明与列映射（兼容旧Python版本） #

如果项目仍在使用Python 3.9及以下版本，可以将mapped_column的定义放到Annotated中，不给字段赋值，让类型检查器识别它为必填项：

from typing import Annotated
from sqlalchemy import ForeignKey
from sqlalchemy.orm import Mapped, mapped_column, DeclarativeBase, Session
import typing_extensions

@typing_extensions.dataclass_transform(kw_only_default=True)
class Base(DeclarativeBase):
    pass

class External(Base):
    __tablename__ = "external"
    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)

class TestModel(Base):
    __tablename__ = "test_table"
    name: Mapped[str]
    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    external_id: Annotated[Mapped[int], mapped_column(ForeignKey("external.id"), nullable=False)]

def test_test_model(session: Session) -> None:
    TEST_NAME = "name"
    external = External()
    session.add(external)
    session.commit()
    
    model3 = TestModel(name=TEST_NAME) # 同样会被类型检查器报错
    session.add(model3)
    session.commit()

这种写法不需要依赖Python 3.10+的Required，但需要确保你的类型检查器（如Mypy、Pyright）安装了SQLAlchemy的类型插件，能正确识别Annotated中的mapped_column定义。

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