summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--Include/cpython/dictobject.h3
-rw-r--r--Lib/test/test_capi.py14
-rw-r--r--Misc/NEWS.d/next/C API/2022-07-25-15-54-27.gh-issue-92678.ziZpxz.rst3
-rw-r--r--Modules/_testcapimodule.c8
-rw-r--r--Objects/dictobject.c29
5 files changed, 57 insertions, 0 deletions
diff --git a/Include/cpython/dictobject.h b/Include/cpython/dictobject.h
index 565ad791..c2e4a46 100644
--- a/Include/cpython/dictobject.h
+++ b/Include/cpython/dictobject.h
@@ -83,3 +83,6 @@ typedef struct {
PyAPI_FUNC(PyObject *) _PyDictView_New(PyObject *, PyTypeObject *);
PyAPI_FUNC(PyObject *) _PyDictView_Intersect(PyObject* self, PyObject *other);
+
+PyAPI_FUNC(int) _PyObject_VisitManagedDict(PyObject *self, visitproc visit, void *arg);
+PyAPI_FUNC(void) _PyObject_ClearManagedDict(PyObject *self);
diff --git a/Lib/test/test_capi.py b/Lib/test/test_capi.py
index 7e571ab..a88a17d 100644
--- a/Lib/test/test_capi.py
+++ b/Lib/test/test_capi.py
@@ -722,6 +722,20 @@ class CAPITest(unittest.TestCase):
with self.subTest(name=name):
self.assertTrue(hasattr(ctypes.pythonapi, name))
+ def test_clear_managed_dict(self):
+
+ class C:
+ def __init__(self):
+ self.a = 1
+
+ c = C()
+ _testcapi.clear_managed_dict(c)
+ self.assertEqual(c.__dict__, {})
+ c = C()
+ self.assertEqual(c.__dict__, {'a':1})
+ _testcapi.clear_managed_dict(c)
+ self.assertEqual(c.__dict__, {})
+
class TestPendingCalls(unittest.TestCase):
diff --git a/Misc/NEWS.d/next/C API/2022-07-25-15-54-27.gh-issue-92678.ziZpxz.rst b/Misc/NEWS.d/next/C API/2022-07-25-15-54-27.gh-issue-92678.ziZpxz.rst
new file mode 100644
index 0000000..52473c9
--- /dev/null
+++ b/Misc/NEWS.d/next/C API/2022-07-25-15-54-27.gh-issue-92678.ziZpxz.rst
@@ -0,0 +1,3 @@
+Adds unstable C-API functions ``_PyObject_VisitManagedDict`` and
+``_PyObject_ClearManagedDict`` to allow C extensions to allow the VM to
+manage their object's dictionaries.
diff --git a/Modules/_testcapimodule.c b/Modules/_testcapimodule.c
index 02635c4..b9f75d1 100644
--- a/Modules/_testcapimodule.c
+++ b/Modules/_testcapimodule.c
@@ -6009,6 +6009,13 @@ settrace_to_record(PyObject *self, PyObject *list)
Py_RETURN_NONE;
}
+static PyObject *
+clear_managed_dict(PyObject *self, PyObject *obj)
+{
+ _PyObject_ClearManagedDict(obj);
+ Py_RETURN_NONE;
+}
+
static PyObject *
test_macros(PyObject *self, PyObject *Py_UNUSED(args))
@@ -6347,6 +6354,7 @@ static PyMethodDef TestMethods[] = {
{"test_code_api", test_code_api, METH_NOARGS, NULL},
{"settrace_to_record", settrace_to_record, METH_O, NULL},
{"test_macros", test_macros, METH_NOARGS, NULL},
+ {"clear_managed_dict", clear_managed_dict, METH_O, NULL},
{NULL, NULL} /* sentinel */
};
diff --git a/Objects/dictobject.c b/Objects/dictobject.c
index ebbd22e..25e191f 100644
--- a/Objects/dictobject.c
+++ b/Objects/dictobject.c
@@ -5583,6 +5583,35 @@ _PyObject_FreeInstanceAttributes(PyObject *self)
free_values(*values_ptr);
}
+int
+_PyObject_VisitManagedDict(PyObject *self, visitproc visit, void *arg)
+{
+ PyTypeObject *tp = Py_TYPE(self);
+ if((tp->tp_flags & Py_TPFLAGS_MANAGED_DICT) == 0) {
+ return 0;
+ }
+ assert(tp->tp_dictoffset);
+ int err = _PyObject_VisitInstanceAttributes(self, visit, arg);
+ if (err) {
+ return err;
+ }
+ Py_VISIT(*_PyObject_ManagedDictPointer(self));
+ return 0;
+}
+
+
+void
+_PyObject_ClearManagedDict(PyObject *self)
+{
+ PyTypeObject *tp = Py_TYPE(self);
+ if((tp->tp_flags & Py_TPFLAGS_MANAGED_DICT) == 0) {
+ return;
+ }
+ _PyObject_FreeInstanceAttributes(self);
+ *_PyObject_ValuesPointer(self) = NULL;
+ Py_CLEAR(*_PyObject_ManagedDictPointer(self));
+}
+
PyObject *
PyObject_GenericGetDict(PyObject *obj, void *context)
{