如何为支持富比较且返回bool值的同类型对象添加Python类型提示？

当需要给支持富比较（如<、>等）且返回bool的函数添加类型提示时，直接使用Any或无约束TypeVar会触发类型检查报错——类型检查器无法确认参数类型支持比较操作，也无法保证比较结果为bool。以下是可行的解决方案：

问题回顾 #

使用Any的问题 #

from typing import Any

def f(a: Any, b: Any) -> bool:
    return a < b

# 报错：Returning Any from function declared to return "bool"  [no-any-return]

Any完全放松了类型约束，类型检查器无法推断a < b的返回值类型，因此报错。

使用无约束TypeVar的问题 #

from typing import TypeVar

T = TypeVar("T")

def f(a: T, b: T) -> bool:
    return a < b

# 报错1：Returning Any from function declared to return "bool"  [no-any-return]
# 报错2：Unsupported left operand type for < ("T")  [operator]

无约束的TypeVar仅保证参数类型一致，但无法确认该类型支持<操作，也无法保证返回值为bool。

解决方案 #

方案1：使用标准库的SupportsRichComparison（Python 3.10+） #

Python 3.10及以上版本的typing（或collections.abc）提供了SupportsRichComparison协议，它定义了支持所有富比较操作的类型。通过将TypeVar绑定到该协议，类型检查器就能识别参数的合法性：

from typing import TypeVar, SupportsRichComparison

T = TypeVar("T", bound=SupportsRichComparison)

def f(a: T, b: T) -> bool:
    return a < b

方案2：自定义Protocol（灵活兼容多版本） #

如果需要更精确地指定支持的比较操作，或者兼容Python 3.10以下版本，可以自定义Protocol来明确定义富比较方法的签名：

from typing import TypeVar, Protocol

class Comparable(Protocol):
    def __lt__(self, other: "Comparable") -> bool: ...
    def __le__(self, other: "Comparable") -> bool: ...
    def __gt__(self, other: "Comparable") -> bool: ...
    def __ge__(self, other: "Comparable") -> bool: ...
    def __eq__(self, other: object) -> bool: ...
    def __ne__(self, other: object) -> bool: ...

T = TypeVar("T", bound=Comparable)

def f(a: T, b: T) -> bool:
    return a < b

方案3：针对单一比较操作的简化版 #

如果只需要支持特定的比较操作（比如仅<），可以简化Protocol定义：

from typing import TypeVar, Protocol

class SupportsLT(Protocol):
    def __lt__(self, other: "SupportsLT") -> bool: ...

T = TypeVar("T", bound=SupportsLT)

def f(a: T, b: T) -> bool:
    return a < b

以上方案均能让类型检查器确认参数支持指定的比较操作，同时保证返回值为bool，解决原有的报错问题。

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