Python定义ProgressData类传入Rust PyO3模块遇转换错误求解决方案

我在Python中定义了如下类：

class ProgressData():
    def __init__(self):
        self.start_time_ms = 0
        self.last_progress_percent = 0.0

实例化后将其传入Rust模块中签名如下的函数：

#[pyfunction]
fn index_documents(py: Python, progress_data: ProgressData) -> PyResult<()> {
    ...
}

我在Rust模块中定义了“等效”的struct并导出为Python类：

#[derive(Clone)]
#[pyclass]
pub struct ProgressData {
    start_time_ms: usize,
    last_progress_percent: f64,
}

#[pymodule]
#[pyo3(name = "populate_index")]
fn my_extension(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
    m.add_function(wrap_pyfunction!(index_documents, m)?)?;
    m.add_class::<HandlingFramework>()?;
    m.add_class::<TextDocument>()?;
    m.add_class::<ProgressData>()?;
    Ok(())
}

但运行时报错：

argument 'progress_data': 'ProgressData' object cannot be converted to 'ProgressData'

此前通过Rust定义HandlingFramework和TextDocument并导出到Python，属性和方法都能正常使用。现咨询以下问题：

• 是否支持反向操作（Python定义类供Rust使用）？
• 是否只能通过Rust模块中的工厂函数生成ProgressData对象，让Python填充数据后再传回Rust函数？
• 能否在Python中为ProgressData定义方法，让Rust将其视为等效的Rust struct+impl？

报错原因 #

Rust中用#[pyclass]导出的ProgressData和Python自定义的ProgressData是完全独立的类型。PyO3只会将Rust导出的类实例转换为对应的Rust struct，Python自定义类的实例无法直接映射，这就是报错的核心原因。

问题解答 #

1. 是否支持Python定义类供Rust直接使用？
   不支持直接将Python自定义类映射为Rust struct。有两种可行替代方案：

   • 方案一：使用Rust导出的ProgressData类
     给Rust的ProgressData添加#[pymethods]实现初始化方法，方便Python实例化：

     #[derive(Clone)]
     #[pyclass]
     pub struct ProgressData {
         start_time_ms: usize,
         last_progress_percent: f64,
     }
     
     #[pymethods]
     impl ProgressData {
         #[new]
         fn new() -> Self {
             ProgressData {
                 start_time_ms: 0,
                 last_progress_percent: 0.0,
             }
         }
     }
     

     之后Python中直接导入该类：from populate_index import ProgressData，实例化后填充数据再传入Rust函数即可。
   • 方案二：通过PyAny手动提取属性
     如果一定要用Python自定义类，Rust函数参数需改为&PyAny，手动提取并转换属性：

     #[pyfunction]
     fn index_documents(py: Python<'_>, progress_data: &PyAny) -> PyResult<()> {
         let start_time_ms = progress_data.getattr("start_time_ms")?.extract::<usize>()?;
         let last_progress_percent = progress_data.getattr("last_progress_percent")?.extract::<f64>()?;
         // 后续业务逻辑
         Ok(())
     }
     

     这种方式无需在Rust中定义ProgressData的#[pyclass]，但需要手动处理属性提取和类型转换的错误。

2. 能否在Python中定义方法让Rust视为Rust struct的impl？
   不行。Rust的impl是编译期确定的静态方法，Python定义的方法属于对象的动态方法，Rust无法直接将其当作原生Rust方法调用。如果需要给ProgressData添加方法，有两种选择：

   • 在Rust的#[pymethods]块中实现方法，导出到Python使用；
   • 若方法逻辑适合在Python端实现，可在Python类中定义，但Rust只能通过PyAny的call_method等API动态调用这些方法，无法像调用Rust原生方法那样直接使用。

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