From 6f4635fe20457a7c513050bb117c2f0511cd4e44 Mon Sep 17 00:00:00 2001 From: Victor Stinner Date: Thu, 19 Nov 2020 00:19:06 +0100 Subject: [PATCH] bpo-1635741: Port _warnings to the multi-phase init (GH-23379) Port the _warnings extension module to the multi-phase initialization API (PEP 489). --- ...2020-11-18-23-46-31.bpo-1635741.GVOQ-m.rst | 2 + Python/_warnings.c | 76 ++++++++----------- 2 files changed, 35 insertions(+), 43 deletions(-) create mode 100644 Misc/NEWS.d/next/Core and Builtins/2020-11-18-23-46-31.bpo-1635741.GVOQ-m.rst diff --git a/Misc/NEWS.d/next/Core and Builtins/2020-11-18-23-46-31.bpo-1635741.GVOQ-m.rst b/Misc/NEWS.d/next/Core and Builtins/2020-11-18-23-46-31.bpo-1635741.GVOQ-m.rst new file mode 100644 index 00000000000..ae9cc0bc5dd --- /dev/null +++ b/Misc/NEWS.d/next/Core and Builtins/2020-11-18-23-46-31.bpo-1635741.GVOQ-m.rst @@ -0,0 +1,2 @@ +Port the ``_warnings`` extension module to the multi-phase initialization +API (:pep:`489`). Patch by Victor Stinner. diff --git a/Python/_warnings.c b/Python/_warnings.c index 313420c6314..021400f5580 100644 --- a/Python/_warnings.c +++ b/Python/_warnings.c @@ -24,9 +24,6 @@ _Py_IDENTIFIER(ignore); typedef struct _warnings_runtime_state WarningsState; -/* Forward declaration of the _warnings module definition. */ -static struct PyModuleDef warningsmodule; - _Py_IDENTIFIER(__name__); /* Given a module object, get its per-module state. */ @@ -1353,52 +1350,45 @@ static PyMethodDef warnings_functions[] = { }; -static struct PyModuleDef warningsmodule = { - PyModuleDef_HEAD_INIT, - MODULE_NAME, /* m_name */ - warnings__doc__, /* m_doc */ - 0, /* m_size */ - warnings_functions, /* m_methods */ - NULL, /* m_reload */ - NULL, /* m_traverse */ - NULL, /* m_clear */ - NULL /* m_free */ +static int +warnings_module_exec(PyObject *module) +{ + WarningsState *st = warnings_get_state(); + if (st == NULL) { + return -1; + } + if (PyModule_AddObjectRef(module, "filters", st->filters) < 0) { + return -1; + } + if (PyModule_AddObjectRef(module, "_onceregistry", st->once_registry) < 0) { + return -1; + } + if (PyModule_AddObjectRef(module, "_defaultaction", st->default_action) < 0) { + return -1; + } + return 0; +} + + +static PyModuleDef_Slot warnings_slots[] = { + {Py_mod_exec, warnings_module_exec}, + {0, NULL} +}; + +static struct PyModuleDef warnings_module = { + PyModuleDef_HEAD_INIT, + .m_name = MODULE_NAME, + .m_doc = warnings__doc__, + .m_size = 0, + .m_methods = warnings_functions, + .m_slots = warnings_slots, }; PyMODINIT_FUNC _PyWarnings_Init(void) { - PyObject *m; - - m = PyModule_Create(&warningsmodule); - if (m == NULL) { - return NULL; - } - - WarningsState *st = warnings_get_state(); - if (st == NULL) { - goto error; - } - - if (PyModule_AddObjectRef(m, "filters", st->filters) < 0) { - goto error; - } - if (PyModule_AddObjectRef(m, "_onceregistry", st->once_registry) < 0) { - goto error; - } - if (PyModule_AddObjectRef(m, "_defaultaction", st->default_action) < 0) { - goto error; - } - - return m; - -error: - if (st != NULL) { - warnings_clear_state(st); - } - Py_DECREF(m); - return NULL; + return PyModuleDef_Init(&warnings_module); } // We need this to ensure that warnings still work until late in finalization.