2 /* Module object implementation */
5 #include "structmember.h"
12 static PyMemberDef module_members
[] = {
13 {"__dict__", T_OBJECT
, offsetof(PyModuleObject
, md_dict
), READONLY
},
18 PyModule_New(const char *name
)
22 m
= PyObject_GC_New(PyModuleObject
, &PyModule_Type
);
25 nameobj
= PyString_FromString(name
);
26 m
->md_dict
= PyDict_New();
27 if (m
->md_dict
== NULL
|| nameobj
== NULL
)
29 if (PyDict_SetItemString(m
->md_dict
, "__name__", nameobj
) != 0)
31 if (PyDict_SetItemString(m
->md_dict
, "__doc__", Py_None
) != 0)
33 if (PyDict_SetItemString(m
->md_dict
, "__package__", Py_None
) != 0)
46 PyModule_GetDict(PyObject
*m
)
49 if (!PyModule_Check(m
)) {
50 PyErr_BadInternalCall();
53 d
= ((PyModuleObject
*)m
) -> md_dict
;
55 ((PyModuleObject
*)m
) -> md_dict
= d
= PyDict_New();
60 PyModule_GetName(PyObject
*m
)
64 if (!PyModule_Check(m
)) {
68 d
= ((PyModuleObject
*)m
)->md_dict
;
70 (nameobj
= PyDict_GetItemString(d
, "__name__")) == NULL
||
71 !PyString_Check(nameobj
))
73 PyErr_SetString(PyExc_SystemError
, "nameless module");
76 return PyString_AsString(nameobj
);
80 PyModule_GetFilename(PyObject
*m
)
84 if (!PyModule_Check(m
)) {
88 d
= ((PyModuleObject
*)m
)->md_dict
;
90 (fileobj
= PyDict_GetItemString(d
, "__file__")) == NULL
||
91 !PyString_Check(fileobj
))
93 PyErr_SetString(PyExc_SystemError
, "module filename missing");
96 return PyString_AsString(fileobj
);
100 _PyModule_Clear(PyObject
*m
)
102 /* To make the execution order of destructors for global
103 objects a bit more predictable, we first zap all objects
104 whose name starts with a single underscore, before we clear
105 the entire dictionary. We zap them by replacing them with
106 None, rather than deleting them from the dictionary, to
107 avoid rehashing the dictionary (to some extent). */
110 PyObject
*key
, *value
;
113 d
= ((PyModuleObject
*)m
)->md_dict
;
117 /* First, clear only names starting with a single underscore */
119 while (PyDict_Next(d
, &pos
, &key
, &value
)) {
120 if (value
!= Py_None
&& PyString_Check(key
)) {
121 char *s
= PyString_AsString(key
);
122 if (s
[0] == '_' && s
[1] != '_') {
123 if (Py_VerboseFlag
> 1)
124 PySys_WriteStderr("# clear[1] %s\n", s
);
125 PyDict_SetItem(d
, key
, Py_None
);
130 /* Next, clear all names except for __builtins__ */
132 while (PyDict_Next(d
, &pos
, &key
, &value
)) {
133 if (value
!= Py_None
&& PyString_Check(key
)) {
134 char *s
= PyString_AsString(key
);
135 if (s
[0] != '_' || strcmp(s
, "__builtins__") != 0) {
136 if (Py_VerboseFlag
> 1)
137 PySys_WriteStderr("# clear[2] %s\n", s
);
138 PyDict_SetItem(d
, key
, Py_None
);
143 /* Note: we leave __builtins__ in place, so that destructors
144 of non-global objects defined in this module can still use
145 builtins, in particularly 'None'. */
152 module_init(PyModuleObject
*m
, PyObject
*args
, PyObject
*kwds
)
154 static char *kwlist
[] = {"name", "doc", NULL
};
155 PyObject
*dict
, *name
= Py_None
, *doc
= Py_None
;
156 if (!PyArg_ParseTupleAndKeywords(args
, kwds
, "S|O:module.__init__",
157 kwlist
, &name
, &doc
))
166 if (PyDict_SetItemString(dict
, "__name__", name
) < 0)
168 if (PyDict_SetItemString(dict
, "__doc__", doc
) < 0)
174 module_dealloc(PyModuleObject
*m
)
176 PyObject_GC_UnTrack(m
);
177 if (m
->md_dict
!= NULL
) {
178 _PyModule_Clear((PyObject
*)m
);
179 Py_DECREF(m
->md_dict
);
181 Py_TYPE(m
)->tp_free((PyObject
*)m
);
185 module_repr(PyModuleObject
*m
)
190 name
= PyModule_GetName((PyObject
*)m
);
195 filename
= PyModule_GetFilename((PyObject
*)m
);
196 if (filename
== NULL
) {
198 return PyString_FromFormat("<module '%s' (built-in)>", name
);
200 return PyString_FromFormat("<module '%s' from '%s'>", name
, filename
);
203 /* We only need a traverse function, no clear function: If the module
204 is in a cycle, md_dict will be cleared as well, which will break
207 module_traverse(PyModuleObject
*m
, visitproc visit
, void *arg
)
209 Py_VISIT(m
->md_dict
);
213 PyDoc_STRVAR(module_doc
,
214 "module(name[, doc])\n\
216 Create a module object.\n\
217 The name must be a string; the optional doc argument can have any type.");
219 PyTypeObject PyModule_Type
= {
220 PyVarObject_HEAD_INIT(&PyType_Type
, 0)
221 "module", /* tp_name */
222 sizeof(PyModuleObject
), /* tp_size */
224 (destructor
)module_dealloc
, /* tp_dealloc */
229 (reprfunc
)module_repr
, /* tp_repr */
230 0, /* tp_as_number */
231 0, /* tp_as_sequence */
232 0, /* tp_as_mapping */
236 PyObject_GenericGetAttr
, /* tp_getattro */
237 PyObject_GenericSetAttr
, /* tp_setattro */
238 0, /* tp_as_buffer */
239 Py_TPFLAGS_DEFAULT
| Py_TPFLAGS_HAVE_GC
|
240 Py_TPFLAGS_BASETYPE
, /* tp_flags */
241 module_doc
, /* tp_doc */
242 (traverseproc
)module_traverse
, /* tp_traverse */
244 0, /* tp_richcompare */
245 0, /* tp_weaklistoffset */
249 module_members
, /* tp_members */
253 0, /* tp_descr_get */
254 0, /* tp_descr_set */
255 offsetof(PyModuleObject
, md_dict
), /* tp_dictoffset */
256 (initproc
)module_init
, /* tp_init */
257 PyType_GenericAlloc
, /* tp_alloc */
258 PyType_GenericNew
, /* tp_new */
259 PyObject_GC_Del
, /* tp_free */