使用数据模型对象实现运算符重载并兼容 Pyright 类型检查

from typing import Callable as Fn, Any, overload
import operator

class Apply:
    """封装运算符逻辑并定义其重载签名的可调用对象。"""

    def __init__(self, op: Fn[[Any, Any], Any], obj: Any) -> None:
        self.op = op
        self.obj = obj

    # 示例：两个模拟的重载签名，实际应用中可根据需求扩展
    @overload
    def __call__(self, x: int) -> str: ...
    @overload
    def __call__(self, x: str) -> int: ...

    def __call__(self, x: int | str) -> str | int:
        # 实际的运算符逻辑可以在这里实现，或者委托给 self.op
        # 为了演示，这里省略具体实现
        if isinstance(x, int):
            return str(x)  # 示例返回类型
        else:
            return int(x) # 示例返回类型

class Op:
    """用于实现运算符的描述符。"""

    def __init__(self, op: Fn[[Any, Any], Any]) -> None:
        self.op = op

    def __get__(self, obj: Any, _: Any) -> Apply:
        # 当描述符被访问时，返回一个 Apply 实例
        return Apply(self.op, obj)

class Foo:
    # 将运算符特殊方法绑定到 Op 描述符实例
    __add__ = Op(operator.add)
    __mul__ = Op(operator.mul)

# 实例化类并测试
foo = Foo()

# 直接通过属性访问描述符返回的 Apply 实例并调用
a: str = foo.__add__(2)    # Pyright 和 Mypy 均能正确推断为 str
b: int = foo.__mul__("2")  # Pyright 和 Mypy 均能正确推断为 int

# 通过运算符语法调用
_ = foo + 1    # Pyright 报告类型错误
_ = foo * "2"  # Pyright 报告类型错误

from typing import Callable as Fn, Any, overload
import operator

# Apply 类保持不变
class Apply:
    """封装运算符逻辑并定义其重载签名的可调用对象。"""

    def __init__(self, op: Fn[[Any, Any], Any], obj: Any) -> None:
        self.op = op
        self.obj = obj

    @overload
    def __call__(self, x: int) -> str: ...
    @overload
    def __call__(self, x: str) -> int: ...

    def __call__(self, x: int | str) -> str | int:
        if isinstance(x, int):
            return str(x)
        else:
            return int(x)

class Op:
    """用于实现运算符的描述符（Pyright 兼容版本）。"""

    def __init__(self, op: Fn[[Any, Any], Any]) -> None:
        self.op = op

    def __get__(self, obj: Any, _: Any) -> Apply:
        return Apply(self.op, obj)

    # 关键的辅助注解：告诉 Pyright，Op 实例在作为可调用对象时，
    # 其行为和类型签名与 Apply 实例相同。
    __call__: Apply 

class Foo:
    __add__ = Op(operator.add)
    __mul__ = Op(operator.mul)

# 实例化类并测试
foo = Foo()

# 使用 reveal_type 验证 Pyright 的类型推断
# (在 Pyright Playground 或集成环境中运行)
# reveal_type(foo.__add__(2))    # 期望: str
# reveal_type(foo.__mul__("2"))  # 期望: int
# reveal_type(foo + 1)           # 期望: str
# reveal_type(foo + "2")         # 期望: int