diff options
Diffstat (limited to 'Modules')
| -rw-r--r-- | Modules/arraymodule.c | 10 | ||||
| -rw-r--r-- | Modules/cmathmodule.c | 12 |
2 files changed, 11 insertions, 11 deletions
diff --git a/Modules/arraymodule.c b/Modules/arraymodule.c index 7372a48..46de178 100644 --- a/Modules/arraymodule.c +++ b/Modules/arraymodule.c @@ -1731,12 +1731,14 @@ static PyMethodDef a_methods[] = { DL_EXPORT(void) initarray(void) { - PyObject *m, *d; + PyObject *m; Arraytype.ob_type = &PyType_Type; m = Py_InitModule3("array", a_methods, module_doc); - d = PyModule_GetDict(m); - PyDict_SetItemString(d, "ArrayType", (PyObject *)&Arraytype); - PyDict_SetItemString(d, "array", (PyObject *)&Arraytype); + + Py_INCREF((PyObject *)&Arraytype); + PyModule_AddObject(m, "ArrayType", (PyObject *)&Arraytype); + Py_INCREF((PyObject *)&Arraytype); + PyModule_AddObject(m, "array", (PyObject *)&Arraytype); /* No need to check the error here, the caller will do that */ } diff --git a/Modules/cmathmodule.c b/Modules/cmathmodule.c index 6e79680..8e74e51 100644 --- a/Modules/cmathmodule.c +++ b/Modules/cmathmodule.c @@ -394,13 +394,11 @@ static PyMethodDef cmath_methods[] = { DL_EXPORT(void) initcmath(void) { - PyObject *m, *d, *v; + PyObject *m; m = Py_InitModule3("cmath", cmath_methods, module_doc); - d = PyModule_GetDict(m); - PyDict_SetItemString(d, "pi", - v = PyFloat_FromDouble(atan(1.0) * 4.0)); - Py_DECREF(v); - PyDict_SetItemString(d, "e", v = PyFloat_FromDouble(exp(1.0))); - Py_DECREF(v); + + PyModule_AddObject(m, "pi", + PyFloat_FromDouble(atan(1.0) * 4.0)); + PyModule_AddObject(m, "e", PyFloat_FromDouble(exp(1.0))); } |