SQLAlchemy中使用Mapped与Protocol的类型兼容问题求解

你的问题核心在于Mapped是不变类型——即使Bar实现了BarProtocol，Mapped[Bar]也无法被视为Mapped[BarProtocol]的兼容类型，导致类型检查失败。以下是两种可行的解决方案：

方案一：匹配实例属性的实际类型（推荐） #

SQLAlchemy的Mapped[T]是用于模型字段声明的类型，当你访问实例的该属性时，实际得到的是T类型的对象（而非Mapped[T]）。因此可以调整Protocol的注解，直接对应实例属性的运行时类型，同时配合mypy-sqlalchemy插件让类型检查工具正确识别SQLAlchemy的类型映射。

修改后的代码：

from typing import Protocol
from sqlalchemy.orm import DeclarativeBase, Mapped

# Protocols：对应实例属性的实际类型
class BarProtocol(Protocol):
    bar: int

class FooProtocol(Protocol):
    @property
    def bar(self) -> BarProtocol:
        ...

def f(foo: FooProtocol) -> BarProtocol:
    return foo.bar

# 实现类保持不变
class Base(DeclarativeBase):
    pass

class Bar(Base):
    bar: Mapped[int]

class Foo(Base):
    bar: Mapped[Bar]

f(Foo())  # 类型检查通过

补充说明 #

需要安装mypy-sqlalchemy插件并在mypy.ini中启用：

[mypy]
plugins = sqlalchemy.ext.mypy.plugin

该插件会自动将模型字段的Mapped[T]解析为实例属性的T类型，让Protocol的结构子类型检查生效。

方案二：使用协变TypeVar兼容Mapped类型 #

如果必须在Protocol中保留Mapped类型注解，可以通过定义协变TypeVar来让Mapped的类型参数支持协变转换，从而允许Mapped[Bar]兼容Mapped[BarProtocol]。

修改后的代码：

from typing import Protocol, TypeVar
from sqlalchemy.orm import DeclarativeBase, Mapped

# 定义协变TypeVar，绑定BarProtocol
TBar = TypeVar("TBar", bound="BarProtocol", covariant=True)

class BarProtocol(Protocol):
    bar: Mapped[int]

class FooProtocol(Protocol[TBar]):
    @property
    def bar(self) -> Mapped[TBar]:
        ...

def f(foo: FooProtocol[BarProtocol]) -> BarProtocol:
    return foo.bar  # 注意：实际运行时foo.bar是Bar实例，需确保逻辑合法

# 实现类保持不变
class Base(DeclarativeBase):
    pass

class Bar(Base):
    bar: Mapped[int]

class Foo(Base):
    bar: Mapped[Bar]

f(Foo())  # 类型检查通过

补充说明 #

协变TypeVar允许子类类型（Bar）替换父类类型（BarProtocol），让Mapped[Bar]被视为Mapped[BarProtocol]的兼容类型。但需注意，函数f中返回foo.bar时，实际得到的是Bar实例（而非Mapped[Bar]），这种写法仅解决类型检查问题，运行时逻辑需确保符合预期。

内容的提问来源于stack exchange，提问作者Roméo Desprès