__eq__是否属于Protocol？如何为泛型树添加相等检查类型约束？

问题背景 #

我们定义了如下代数数据类型（ADT）来表示树结构：

from __future__ import annotations
from dataclasses import dataclass
from typing import TypeVar, Generic

T = TypeVar("T")

@dataclass(frozen=True)
class Branch(Generic[T]):
    value: T
    left: Tree[T]
    right: Tree[T]

Tree = Branch[T] | None

接着实现了检查树中是否包含指定元素的函数：

def contains(t: Tree[T], x: T) -> bool:
    match t:
        case None:
            return False
        case Branch(value, left, right):
            return x == value or contains(left, x) or contains(right, x)

这段代码能通过类型检查，但存在一个隐患：无法保证泛型参数T支持相等比较操作（==）。

作为Python新手，我了解Protocol用于定义共享功能，想知道：__eq__是否属于某个标准库中的Protocol，可以用来约束泛型参数T（类似Haskell的Eq类型类）？如果没有，还有哪些提升类型安全的方法？

解答 #

1. 自定义Protocol约束T支持相等检查 #

Python标准库中没有直接对应Haskell Eq类型类的现成Protocol，但你可以自己定义一个包含__eq__方法的Protocol，以此约束泛型参数T：

from __future__ import annotations
from dataclasses import dataclass
from typing import TypeVar, Generic, Protocol, Any

# 定义支持相等比较的Protocol
class EqProtocol(Protocol):
    def __eq__(self, other: Any) -> bool: ...

# 将TypeVar的bound设置为EqProtocol，限制T必须实现__eq__
T = TypeVar("T", bound=EqProtocol)

@dataclass(frozen=True)
class Branch(Generic[T]):
    value: T
    left: Tree[T]
    right: Tree[T]

Tree = Branch[T] | None

def contains(t: Tree[T], x: T) -> bool:
    match t:
        case None:
            return False
        case Branch(value, left, right):
            return x == value or contains(left, x) or contains(right, x)

这样配置后，mypy等类型检查工具会自动校验传入contains函数的T类型是否实现了__eq__方法，提前避免运行时可能出现的不支持相等比较的错误。

2. 其他提升类型安全的方法 #

• 利用object作为约束：如果不需要严格区分“自定义相等逻辑”和“默认身份比较”，可以直接将TypeVar的bound设为object——因为Python中所有类型都继承自object，而object自带__eq__方法（默认比较对象身份）。这种方式更简单，但不够明确：

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

• 运行时校验：在函数开头添加运行时检查，确保传入的类型支持相等比较。虽然这是运行时保障，但可以作为静态类型检查的补充：

  def contains(t: Tree[T], x: T) -> bool:
      if not hasattr(x, "__eq__"):
          raise TypeError(f"Type {type(x)} does not support equality comparison")
      # 原函数逻辑...
  

• 启用严格类型检查：使用mypy、pyright等工具时开启严格模式（比如mypy的--strict参数），能更全面地捕获潜在的类型问题，包括隐式的类型转换、未定义属性访问等。
• 自定义类型守卫：如果需要更复杂的类型校验逻辑，可以定义类型守卫函数，在运行时验证类型的同时，帮助类型检查器更准确地推断类型：

  from typing import TypeGuard
  
  def is_eq_supported(obj: Any) -> TypeGuard[EqProtocol]:
      return hasattr(obj, "__eq__")
  
  def contains(t: Tree[T], x: T) -> bool:
      if not is_eq_supported(x):
          raise TypeError(f"Type {type(x)} does not support equality comparison")
      # 原函数逻辑...
  

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