PyO3中如何让函数接收HashMap<String,自定义PyClass>参数？

云上老周

2026-6-28

问题原因及解决方法

问题原因

PyO3为HashMap<K, V>实现FromPyObject trait的前提是：

K 同时实现 FromPyObject、Eq 和 Hash
V 实现 FromPyObject

你代码中的String满足K的所有要求，但Bar作为被#[pyclass]标记的结构体，本身并没有直接实现FromPyObject——因为在Python环境中，Bar的实例是被PyO3包装在PyCell<Bar>中的，Rust无法直接将Python对象转换为原始的Bar结构体，必须通过PyO3提供的句柄类型来完成转换。

解决方法

有两种常用的正确写法，根据你的需求选择：

方法一：使用`&PyCell<Bar>`接收引用

如果只需要访问Bar的实例而不需要获取所有权，推荐使用&PyCell<Bar>，它实现了FromPyObject，同时可以安全地获取Bar的可变或不可变引用：

use std::collections::HashMap;
use pyo3::prelude::*;

#[pyclass]
struct Bar;

#[pyfunction]
fn example(foo: HashMap<String, &PyCell<Bar>>) {
    // 示例：获取不可变引用
    for (key, cell) in foo {
        let bar = cell.borrow();
        // 使用bar
    }
}

方法二：使用`Py<Bar>`接收所有权句柄

如果需要持有Bar实例的所有权，可以使用Py<Bar>，它是PyO3提供的拥有所有权的Python对象句柄，同样实现了FromPyObject：

use std::collections::HashMap;
use pyo3::prelude::*;

#[pyclass]
struct Bar;

#[pyfunction]
fn example(foo: HashMap<String, Py<Bar>>) {
    // 示例：将Py<Bar>转换为Bar实例
    for (key, py_bar) in foo {
        let bar: Bar = py_bar.extract().unwrap();
        // 使用bar
    }
}