cpython-lru_cache-weakref: Python/Python-ast.c comparison

Make AST nodes pickleable.

author	Georg Brandl <georg@python.org>
date	Sun, 30 Mar 2008 20:20:39 +0000
parents	59d42e6d1885
children	4fe192d48960

comparison

equal deleted inserted replaced

-:59d42e6d1885
+:be18df634405
 cleanup:
 Py_XDECREF(fields);
 return res;
 }
+/* Pickling support */
+static PyObject *
+ast_type_reduce(PyObject *self, PyObject *unused)
+{
+PyObject *res;
+PyObject *dict = PyObject_GetAttrString(self, "__dict__");
+if (dict == NULL) {
+if (PyErr_ExceptionMatches(PyExc_AttributeError))
+PyErr_Clear();
+else
+return NULL;
+}
+if (dict) {
+res = Py_BuildValue("O()O", Py_TYPE(self), dict);
+Py_DECREF(dict);
+return res;
+}
+return Py_BuildValue("O()", Py_TYPE(self));
+}
+static PyMethodDef ast_type_methods[] = {
+{"__reduce__", ast_type_reduce, METH_NOARGS, NULL},
+{NULL}
+};
 static PyTypeObject AST_type = {
 PyVarObject_HEAD_INIT(&PyType_Type, 0)
-"AST",
+"_ast.AST",
 sizeof(PyObject),
 0,
 0,                       /* tp_dealloc */
 0,                       /* tp_print */
 0,                       /* tp_getattr */
 0,                       /* tp_clear */
 0,                       /* tp_richcompare */
 0,                       /* tp_weaklistoffset */
 0,                       /* tp_iter */
 0,                       /* tp_iternext */
-0,                       /* tp_methods */
+ast_type_methods,        /* tp_methods */
 0,                       /* tp_members */
 0,                       /* tp_getset */
 0,                       /* tp_base */
 0,                       /* tp_dict */
 0,                       /* tp_descr_get */

Mercurial > cpython-lru_cache-weakref