Python C API：如何为C扩展定义的Python类型实现copy与deepcopy方法？

嘿，这个问题问到点子上了！确实，Python的C API里没有直接对应__copy__和__deepcopy__的tp_slot槽位，但咱们可以通过给自定义类型注册特殊方法的方式，让copy模块能识别并调用它们。下面我给你一步步拆解具体实现：

一、实现__copy__（浅拷贝）方法 #

浅拷贝的核心是创建一个新实例，然后复制原对象的顶层属性（可变属性会直接引用，不递归拷贝）。

1. 编写__copy__的C实现函数 #

函数要符合Python C API的方法签名，用METH_NOARGS标记（因为不需要额外参数）：

static PyObject* MyType_copy(PyObject* self) {
    // 创建当前类型的新实例（调用tp_new）
    PyObject* new_obj = PyObject_New(MyObject, &MyType);
    if (new_obj == NULL) {
        return NULL;
    }

    // 浅拷贝属性：这里假设我们的类型有一个"value"属性
    // 如果是不可变类型，直接赋值；可变类型则共享引用
    MyObject* self_obj = (MyObject*)self;
    MyObject* new_obj_ptr = (MyObject*)new_obj;
    new_obj_ptr->value = Py_IncRef(self_obj->value); // 增加引用计数，共享对象

    return new_obj;
}

2. 将方法注册到类型的方法列表 #

在你的PyTypeObject结构体的tp_methods数组里，添加__copy__方法：

static PyMethodDef MyType_methods[] = {
    {"__copy__", (PyCFunction)MyType_copy, METH_NOARGS, "Return a shallow copy of the object"},
    // 其他方法...
    {NULL} // 数组结束标记
};

二、实现__deepcopy__（深拷贝）方法 #

深拷贝需要递归拷贝所有属性，还要处理循环引用，所以必须接收memo字典参数（用来记录已经拷贝过的对象，避免无限递归）。

1. 编写__deepcopy__的C实现函数 #

函数用METH_VARARGS标记，因为需要接收memo参数：

static PyObject* MyType_deepcopy(PyObject* self, PyObject* args) {
    PyObject* memo;
    // 解析参数：memo是必须的字典
    if (!PyArg_ParseTuple(args, "O", &memo)) {
        return NULL;
    }
    // 检查memo是否是字典类型
    if (!PyDict_Check(memo)) {
        PyErr_SetString(PyExc_TypeError, "memo must be a dictionary");
        return NULL;
    }

    // 先检查memo里是否已经有当前对象的拷贝，避免循环引用
    PyObject* existing_copy = PyDict_GetItem(memo, self);
    if (existing_copy != NULL) {
        Py_IncRef(existing_copy);
        return existing_copy;
    }

    // 创建新实例
    PyObject* new_obj = PyObject_New(MyObject, &MyType);
    if (new_obj == NULL) {
        return NULL;
    }

    // 先把新实例加入memo，防止拷贝属性时遇到循环引用
    if (PyDict_SetItem(memo, self, new_obj) == -1) {
        Py_DECREF(new_obj);
        return NULL;
    }

    // 深拷贝属性：使用PyDeepCopy函数递归拷贝每个属性
    MyObject* self_obj = (MyObject*)self;
    MyObject* new_obj_ptr = (MyObject*)new_obj;
    new_obj_ptr->value = PyDeepCopy(self_obj->value);
    if (new_obj_ptr->value == NULL) {
        Py_DECREF(new_obj);
        return NULL;
    }

    return new_obj;
}

2. 注册__deepcopy__方法 #

同样把它加到tp_methods数组里：

static PyMethodDef MyType_methods[] = {
    {"__copy__", (PyCFunction)MyType_copy, METH_NOARGS, "Return a shallow copy of the object"},
    {"__deepcopy__", (PyCFunction)MyType_deepcopy, METH_VARARGS, "Return a deep copy of the object"},
    // 其他方法...
    {NULL}
};

三、关键注意事项 #

• 循环引用处理：深拷贝时一定要先把新实例加入memo字典再拷贝属性，否则如果对象引用自身，会导致无限递归。
• 引用计数：所有Python对象的操作都要正确管理引用计数，避免内存泄漏或野指针。
• 方法名准确性：必须严格使用__copy__和__deepcopy__作为方法名，这样copy.copy()和copy.deepcopy()会自动调用它们。

这样实现后，你的C扩展类型就能像原生Python类型一样，支持copy.copy()和copy.deepcopy()操作啦！

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