+++ /dev/null
-\r
-/* Module object implementation */\r
-\r
-#include "Python.h"\r
-#include "structmember.h"\r
-\r
-typedef struct {\r
- PyObject_HEAD\r
- PyObject *md_dict;\r
-} PyModuleObject;\r
-\r
-static PyMemberDef module_members[] = {\r
- {"__dict__", T_OBJECT, offsetof(PyModuleObject, md_dict), READONLY},\r
- {0}\r
-};\r
-\r
-PyObject *\r
-PyModule_New(const char *name)\r
-{\r
- PyModuleObject *m;\r
- PyObject *nameobj;\r
- m = PyObject_GC_New(PyModuleObject, &PyModule_Type);\r
- if (m == NULL)\r
- return NULL;\r
- nameobj = PyString_FromString(name);\r
- m->md_dict = PyDict_New();\r
- if (m->md_dict == NULL || nameobj == NULL)\r
- goto fail;\r
- if (PyDict_SetItemString(m->md_dict, "__name__", nameobj) != 0)\r
- goto fail;\r
- if (PyDict_SetItemString(m->md_dict, "__doc__", Py_None) != 0)\r
- goto fail;\r
- if (PyDict_SetItemString(m->md_dict, "__package__", Py_None) != 0)\r
- goto fail;\r
- Py_DECREF(nameobj);\r
- PyObject_GC_Track(m);\r
- return (PyObject *)m;\r
-\r
- fail:\r
- Py_XDECREF(nameobj);\r
- Py_DECREF(m);\r
- return NULL;\r
-}\r
-\r
-PyObject *\r
-PyModule_GetDict(PyObject *m)\r
-{\r
- PyObject *d;\r
- if (!PyModule_Check(m)) {\r
- PyErr_BadInternalCall();\r
- return NULL;\r
- }\r
- d = ((PyModuleObject *)m) -> md_dict;\r
- if (d == NULL)\r
- ((PyModuleObject *)m) -> md_dict = d = PyDict_New();\r
- return d;\r
-}\r
-\r
-char *\r
-PyModule_GetName(PyObject *m)\r
-{\r
- PyObject *d;\r
- PyObject *nameobj;\r
- if (!PyModule_Check(m)) {\r
- PyErr_BadArgument();\r
- return NULL;\r
- }\r
- d = ((PyModuleObject *)m)->md_dict;\r
- if (d == NULL ||\r
- (nameobj = PyDict_GetItemString(d, "__name__")) == NULL ||\r
- !PyString_Check(nameobj))\r
- {\r
- PyErr_SetString(PyExc_SystemError, "nameless module");\r
- return NULL;\r
- }\r
- return PyString_AsString(nameobj);\r
-}\r
-\r
-char *\r
-PyModule_GetFilename(PyObject *m)\r
-{\r
- PyObject *d;\r
- PyObject *fileobj;\r
- if (!PyModule_Check(m)) {\r
- PyErr_BadArgument();\r
- return NULL;\r
- }\r
- d = ((PyModuleObject *)m)->md_dict;\r
- if (d == NULL ||\r
- (fileobj = PyDict_GetItemString(d, "__file__")) == NULL ||\r
- !PyString_Check(fileobj))\r
- {\r
- PyErr_SetString(PyExc_SystemError, "module filename missing");\r
- return NULL;\r
- }\r
- return PyString_AsString(fileobj);\r
-}\r
-\r
-void\r
-_PyModule_Clear(PyObject *m)\r
-{\r
- /* To make the execution order of destructors for global\r
- objects a bit more predictable, we first zap all objects\r
- whose name starts with a single underscore, before we clear\r
- the entire dictionary. We zap them by replacing them with\r
- None, rather than deleting them from the dictionary, to\r
- avoid rehashing the dictionary (to some extent). */\r
-\r
- Py_ssize_t pos;\r
- PyObject *key, *value;\r
- PyObject *d;\r
-\r
- d = ((PyModuleObject *)m)->md_dict;\r
- if (d == NULL)\r
- return;\r
-\r
- /* First, clear only names starting with a single underscore */\r
- pos = 0;\r
- while (PyDict_Next(d, &pos, &key, &value)) {\r
- if (value != Py_None && PyString_Check(key)) {\r
- char *s = PyString_AsString(key);\r
- if (s[0] == '_' && s[1] != '_') {\r
- if (Py_VerboseFlag > 1)\r
- PySys_WriteStderr("# clear[1] %s\n", s);\r
- if (PyDict_SetItem(d, key, Py_None) != 0)\r
- PyErr_Clear();\r
- }\r
- }\r
- }\r
-\r
- /* Next, clear all names except for __builtins__ */\r
- pos = 0;\r
- while (PyDict_Next(d, &pos, &key, &value)) {\r
- if (value != Py_None && PyString_Check(key)) {\r
- char *s = PyString_AsString(key);\r
- if (s[0] != '_' || strcmp(s, "__builtins__") != 0) {\r
- if (Py_VerboseFlag > 1)\r
- PySys_WriteStderr("# clear[2] %s\n", s);\r
- if (PyDict_SetItem(d, key, Py_None) != 0)\r
- PyErr_Clear();\r
- }\r
- }\r
- }\r
-\r
- /* Note: we leave __builtins__ in place, so that destructors\r
- of non-global objects defined in this module can still use\r
- builtins, in particularly 'None'. */\r
-\r
-}\r
-\r
-/* Methods */\r
-\r
-static int\r
-module_init(PyModuleObject *m, PyObject *args, PyObject *kwds)\r
-{\r
- static char *kwlist[] = {"name", "doc", NULL};\r
- PyObject *dict, *name = Py_None, *doc = Py_None;\r
- if (!PyArg_ParseTupleAndKeywords(args, kwds, "S|O:module.__init__",\r
- kwlist, &name, &doc))\r
- return -1;\r
- dict = m->md_dict;\r
- if (dict == NULL) {\r
- dict = PyDict_New();\r
- if (dict == NULL)\r
- return -1;\r
- m->md_dict = dict;\r
- }\r
- if (PyDict_SetItemString(dict, "__name__", name) < 0)\r
- return -1;\r
- if (PyDict_SetItemString(dict, "__doc__", doc) < 0)\r
- return -1;\r
- return 0;\r
-}\r
-\r
-static void\r
-module_dealloc(PyModuleObject *m)\r
-{\r
- PyObject_GC_UnTrack(m);\r
- if (m->md_dict != NULL) {\r
- _PyModule_Clear((PyObject *)m);\r
- Py_DECREF(m->md_dict);\r
- }\r
- Py_TYPE(m)->tp_free((PyObject *)m);\r
-}\r
-\r
-static PyObject *\r
-module_repr(PyModuleObject *m)\r
-{\r
- char *name;\r
- char *filename;\r
-\r
- name = PyModule_GetName((PyObject *)m);\r
- if (name == NULL) {\r
- PyErr_Clear();\r
- name = "?";\r
- }\r
- filename = PyModule_GetFilename((PyObject *)m);\r
- if (filename == NULL) {\r
- PyErr_Clear();\r
- return PyString_FromFormat("<module '%s' (built-in)>", name);\r
- }\r
- return PyString_FromFormat("<module '%s' from '%s'>", name, filename);\r
-}\r
-\r
-/* We only need a traverse function, no clear function: If the module\r
- is in a cycle, md_dict will be cleared as well, which will break\r
- the cycle. */\r
-static int\r
-module_traverse(PyModuleObject *m, visitproc visit, void *arg)\r
-{\r
- Py_VISIT(m->md_dict);\r
- return 0;\r
-}\r
-\r
-PyDoc_STRVAR(module_doc,\r
-"module(name[, doc])\n\\r
-\n\\r
-Create a module object.\n\\r
-The name must be a string; the optional doc argument can have any type.");\r
-\r
-PyTypeObject PyModule_Type = {\r
- PyVarObject_HEAD_INIT(&PyType_Type, 0)\r
- "module", /* tp_name */\r
- sizeof(PyModuleObject), /* tp_size */\r
- 0, /* tp_itemsize */\r
- (destructor)module_dealloc, /* tp_dealloc */\r
- 0, /* tp_print */\r
- 0, /* tp_getattr */\r
- 0, /* tp_setattr */\r
- 0, /* tp_compare */\r
- (reprfunc)module_repr, /* tp_repr */\r
- 0, /* tp_as_number */\r
- 0, /* tp_as_sequence */\r
- 0, /* tp_as_mapping */\r
- 0, /* tp_hash */\r
- 0, /* tp_call */\r
- 0, /* tp_str */\r
- PyObject_GenericGetAttr, /* tp_getattro */\r
- PyObject_GenericSetAttr, /* tp_setattro */\r
- 0, /* tp_as_buffer */\r
- Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |\r
- Py_TPFLAGS_BASETYPE, /* tp_flags */\r
- module_doc, /* tp_doc */\r
- (traverseproc)module_traverse, /* tp_traverse */\r
- 0, /* tp_clear */\r
- 0, /* tp_richcompare */\r
- 0, /* tp_weaklistoffset */\r
- 0, /* tp_iter */\r
- 0, /* tp_iternext */\r
- 0, /* tp_methods */\r
- module_members, /* tp_members */\r
- 0, /* tp_getset */\r
- 0, /* tp_base */\r
- 0, /* tp_dict */\r
- 0, /* tp_descr_get */\r
- 0, /* tp_descr_set */\r
- offsetof(PyModuleObject, md_dict), /* tp_dictoffset */\r
- (initproc)module_init, /* tp_init */\r
- PyType_GenericAlloc, /* tp_alloc */\r
- PyType_GenericNew, /* tp_new */\r
- PyObject_GC_Del, /* tp_free */\r
-};\r