如何在PyO3中获取Packet内Header成员的引用以修改其值？

问题根源在于PyO3默认导出结构体字段时，会返回字段的拷贝/克隆（受限于Rust的所有权规则），所以你修改的只是拷贝后的对象，而非原Packet实例内的Header成员。要实现Python的引用语义，需要通过自定义getter方法返回原成员的引用，而非拷贝。

具体实现步骤 #

1. 定义结构体时不使用默认字段导出：不给Packet的h成员加#[pyo3(get, set)]，改用自定义方法。
2. 用PyRef/PyRefMut返回引用：PyO3提供PyRef（不可变引用）和PyRefMut（可变引用）类型，用于在Python中暴露Rust结构体的引用。

完整Rust代码示例 #

use pyo3::prelude::*;

#[pyclass]
#[derive(Debug, Clone)]
struct Header {
    #[pyo3(get, set)]
    a: i32,
    #[pyo3(get, set)]
    b: String,
}

#[pyclass]
struct Packet {
    h: Header,
}

#[pymethods]
impl Packet {
    #[new]
    fn new(a: i32, b: &str) -> Self {
        Packet {
            h: Header { a, b: b.to_string() },
        }
    }

    // 返回不可变引用
    fn get_h(&self) -> PyRef<'_, Header> {
        PyRef::from(self, &self.h)
    }

    // 返回可变引用，用于修改原成员
    fn get_h_mut(&mut self) -> PyRefMut<'_, Header> {
        PyRefMut::from(self, &mut self.h)
    }
}

#[pymodule]
fn rust_ext(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
    m.add_class::<Header>()?;
    m.add_class::<Packet>()?;
    Ok(())
}

Python端测试 #

from rust_ext import Packet

# 创建Packet实例
p = Packet(10, "test")

# 通过可变引用修改原Header的a值
h_mut = p.get_h_mut()
h_mut.a = 20
print(p.get_h().a)  # 输出20，原实例已修改

# 直接通过引用链修改
p.get_h_mut().b = "updated"
print(p.get_h().b)  # 输出updated

# 将引用赋值给变量后修改，同样生效
h_ref = p.get_h_mut()
h_ref.a = 30
print(p.get_h().a)  # 输出30

关键说明 #

• PyRef::from(self, &self.h)：将Packet实例中的h成员包装为Python可访问的不可变引用，确保Python端拿到的是原对象的视图。
• PyRefMut::from(self, &mut self.h)：返回可变引用，允许Python端直接修改原Packet内的Header成员，符合Python的引用语义。
• 如果你只需要读取Header，用get_h；需要修改则用get_h_mut，避免不必要的可变权限。

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