PyO3开发Rust-Python库时__radd__方法的TypeError问题

我之前在使用PyO3旧版本时也踩过这个坑！问题出在你创建TypeError异常的方式上——在PyO3 0.11.1这个版本里，创建异常必须显式传递Python<'_>实例，否则异常对象无法正确初始化，导致在操作符调用（如other + myitem）时抛出无信息的TypeError。

问题根源 #

你的__add__和__radd__方法中，直接调用TypeError::py_err("Not implemented for this type!")没有传递Python实例。直接调用方法时这个问题可能被掩盖，但当Python通过操作符触发__radd__时，错误传播机制会暴露这个未正确初始化的异常，导致空错误信息。

而TestClass2能用特定类型正常工作，是因为PyO3内部帮你处理了参数转换时的异常创建，自动传递了Python实例，所以错误信息是正常的。

解决方案 #

修改异常创建逻辑，从&PyAny参数获取Python实例（通过.py()方法），再传递给TypeError::py_err：

#[pyproto]
impl PyNumberProtocol for TestClass1 {
    fn __add__(lhs: TestClass1, rhs: &PyAny) -> PyResult<TestClass1> {
        let py = rhs.py(); // 获取当前Python上下文实例
        let pynum_result: Result<i64, _> = rhs.extract();
        if let Ok(pynum) = pynum_result {
            Ok(TestClass1 { value: lhs.value + pynum })
        } else {
            Err(TypeError::py_err(py, "Not implemented for this type!")) // 传递实例创建正确的异常
        }
    }

    fn __radd__(self, other: &PyAny) -> PyResult<TestClass1> {
        let py = other.py(); // 获取当前Python上下文实例
        let pynum_result: Result<i64, _> = other.extract();
        if let Ok(pynum) = pynum_result {
            Ok(TestClass1 { value: self.value + pynum })
        } else {
            Err(TypeError::py_err(py, "Not implemented for this type!")) // 传递实例创建正确的异常
        }
    }
}

测试验证 #

修改后重新编译运行你的Python测试代码：

• (3 + tc1).value会正常输出13
• 3.0 + tc1会抛出带有明确信息的TypeError: Not implemented for this type!
• 直接调用tc1.__radd__(3)依然正常工作

额外提示 #

如果后续环境允许，建议升级PyO3到0.12及以上版本——新版本简化了异常创建API，无需显式传递Python实例，直接写TypeError::py_err("...")即可正常工作，能避免这类版本相关的坑。

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