From 6c40eb7f421804eba3c24a9336c4a9e59dba636c Mon Sep 17 00:00:00 2001
From: Antoine Pitrou <solipsis@pitrou.net>
Date: Wed, 18 Jan 2012 20:16:09 +0100
Subject: Fix the builtin module initialization code to store the init function
 for future reinitialization.

---
 Python/import.c | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'Python/import.c')

diff --git a/Python/import.c b/Python/import.c
index e721498aaa3..ee3f9b0682f 100644
--- a/Python/import.c
+++ b/Python/import.c
@@ -2169,6 +2169,7 @@ init_builtin(char *name)
 
     for (p = PyImport_Inittab; p->name != NULL; p++) {
         PyObject *mod;
+        PyModuleDef *def;
         if (strcmp(name, p->name) == 0) {
             if (p->initfunc == NULL) {
                 PyErr_Format(PyExc_ImportError,
@@ -2181,6 +2182,9 @@ init_builtin(char *name)
             mod = (*p->initfunc)();
             if (mod == 0)
                 return -1;
+            /* Remember pointer to module init function. */
+            def = PyModule_GetDef(mod);
+            def->m_base.m_init = p->initfunc;
             if (_PyImport_FixupBuiltin(mod, name) < 0)
                 return -1;
             /* FixupExtension has put the module into sys.modules,
-- 
cgit v1.2.3