如何对Maturin创建的Python类进行Pickle/Deepcopy操作？

要解决这个问题，核心是给你的#[pyclass]实现Python的pickle协议——因为deepcopy底层依赖pickle的序列化/反序列化流程，而PyO3默认不会为自定义类自动实现该协议。以下是具体方案：

1. 直接实现__getstate__和__setstate__方法 #

这是最直接的方案，在Rust的#[pymethods]块中添加这两个方法，让pickle能正确保存和恢复对象状态：

示例代码 #

use pyo3::prelude::*;

#[pyclass]
struct MyRustClass {
    value: i32,
    // 其他自定义字段（比如字符串、Vec等可序列化类型）
}

#[pymethods]
impl MyRustClass {
    #[new]
    fn new(value: i32) -> Self {
        MyRustClass { value }
    }

    // 导出__getstate__：返回对象的可序列化状态（用元组/字典都可以）
    fn __getstate__(&self) -> PyResult<(i32,)> {
        Ok((self.value,))
    }

    // 导出__setstate__：从恢复的状态中重建对象
    fn __setstate__(&mut self, state: (i32,)) -> PyResult<()> {
        self.value = state.0;
        Ok(())
    }

    // 对于字段简单的类，也可以用PyO3宏自动生成这两个方法：
    // #[pyo3(getstate, setstate)]
}

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

完成后，Python中即可正常使用copy.deepcopy()：

import copy
from my_rust_module import MyRustClass

obj = MyRustClass(42)
obj_copy = copy.deepcopy(obj)

2. 处理不可直接序列化的Rust资源 #

如果你的类包含原始指针、非Send/Sync类型或其他无法直接序列化的资源，需要在__getstate__中把这些资源转换成可序列化的形式，或者在__setstate__中创建新的实例来拷贝资源：

• 比如对于一个内部持有文件句柄的类，可以在__getstate__中保存文件路径，__setstate__中重新打开文件；
• 对于自定义的Rust结构体，要确保它能被正确拷贝（实现Clone trait），并在状态中传递必要的信息来重建。

3. 替代方案：自定义拷贝方法 #

如果实现pickle协议成本过高，可以在Rust类中直接实现一个copy方法，返回新的类实例，然后在Python中用这个方法代替deepcopy：

Rust端代码 #

#[pymethods]
impl MyRustClass {
    // ... 其他方法 ...

    fn copy(&self) -> Self {
        MyRustClass {
            value: self.value,
            // 其他字段按需要拷贝
        }
    }
}

Python端调用 #

obj = MyRustClass(42)
obj_copy = obj.copy()

是否需要避免deepcopy？ #

如果你的类涉及极复杂的不可拷贝资源（比如绑定到硬件的句柄、独占锁等），且实现pickle或自定义拷贝的成本极高，那确实应该在Python层避免使用deepcopy，改用手动管理对象拷贝的逻辑。

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