From 05e4a296ecc127641160a04f39cc02c0f66a8c27 Mon Sep 17 00:00:00 2001 From: Dong-hee Na Date: Mon, 23 Mar 2020 01:17:34 +0900 Subject: bpo-40024: Add PyModule_AddType() helper function (GH-19088) --- Python/modsupport.c | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'Python/modsupport.c') diff --git a/Python/modsupport.c b/Python/modsupport.c index 7271af3ac5..a8e78c3cec 100644 --- a/Python/modsupport.c +++ b/Python/modsupport.c @@ -678,3 +678,22 @@ PyModule_AddStringConstant(PyObject *m, const char *name, const char *value) Py_DECREF(o); return -1; } + +int +PyModule_AddType(PyObject *module, PyTypeObject *type) +{ + if (PyType_Ready(type) < 0) { + return -1; + } + + const char *name = _PyType_Name(type); + assert(name != NULL); + + Py_INCREF(type); + if (PyModule_AddObject(module, name, (PyObject *)type) < 0) { + Py_DECREF(type); + return -1; + } + + return 0; +} -- cgit v1.2.1