Add minimal interface to symtable: _symtable module.

1 files changed, 72 insertions, 0 deletions

diff --git a/Modules/symtablemodule.c b/Modules/symtablemodule.c
new file mode 100644
index 0000000..ccf4aba
--- /dev/null
+++ b/Modules/symtablemodule.c
@@ -0,0 +1,72 @@
+#include "Python.h"
+
+#include "compile.h"
+#include "symtable.h"
+
+static PyObject *
+symtable_symtable(PyObject *self, PyObject *args)
+{
+	struct symtable *st;
+	PyObject *t;
+
+	char *str;
+	char *filename;
+	char *startstr;
+	int start;
+
+	if (!PyArg_ParseTuple(args, "sss:symtable", &str, &filename, 
+			      &startstr))
+		return NULL;
+	if (strcmp(startstr, "exec") == 0)
+		start = Py_file_input;
+	else if (strcmp(startstr, "eval") == 0)
+		start = Py_eval_input;
+	else if (strcmp(startstr, "single") == 0)
+		start = Py_single_input;
+	else {
+		PyErr_SetString(PyExc_ValueError,
+		   "symtable() arg 3 must be 'exec' or 'eval' or 'single'");
+		return NULL;
+	}
+	st = Py_SymtableString(str, filename, start);
+	if (st == NULL)
+		return NULL;
+	t = Py_BuildValue("OO", st->st_symbols, st->st_scopes);
+	PySymtable_Free(st);
+	return t;
+}
+
+static PyMethodDef symtable_methods[] = {
+	{"symtable",	symtable_symtable,	METH_VARARGS,
+	 "Return symbol and scope dictionaries used internally by compiler."},
+	{NULL,		NULL}		/* sentinel */
+};
+
+DL_EXPORT(void)
+init_symtable(void)
+{
+	PyObject *m;
+
+	m = Py_InitModule("_symtable", symtable_methods);
+	PyModule_AddIntConstant(m, "USE", USE);
+	PyModule_AddIntConstant(m, "DEF_GLOBAL", DEF_GLOBAL);
+	PyModule_AddIntConstant(m, "DEF_LOCAL", DEF_LOCAL);
+	PyModule_AddIntConstant(m, "DEF_PARAM", DEF_PARAM);
+	PyModule_AddIntConstant(m, "DEF_STAR", DEF_STAR);
+	PyModule_AddIntConstant(m, "DEF_DOUBLESTAR", DEF_DOUBLESTAR);
+	PyModule_AddIntConstant(m, "DEF_INTUPLE", DEF_INTUPLE);
+	PyModule_AddIntConstant(m, "DEF_FREE", DEF_FREE);
+	PyModule_AddIntConstant(m, "DEF_FREE_GLOBAL", DEF_FREE_GLOBAL);
+	PyModule_AddIntConstant(m, "DEF_FREE_CLASS", DEF_FREE_CLASS);
+	PyModule_AddIntConstant(m, "DEF_IMPORT", DEF_IMPORT);
+
+	PyModule_AddIntConstant(m, "TYPE_FUNCTION", TYPE_FUNCTION);
+	PyModule_AddIntConstant(m, "TYPE_CLASS", TYPE_CLASS);
+	PyModule_AddIntConstant(m, "TYPE_MODULE", TYPE_MODULE);
+
+	PyModule_AddIntConstant(m, "LOCAL", LOCAL);
+	PyModule_AddIntConstant(m, "GLOBAL_EXPLICIT", GLOBAL_EXPLICIT);
+	PyModule_AddIntConstant(m, "GLOBAL_IMPLICIT", GLOBAL_IMPLICIT);
+	PyModule_AddIntConstant(m, "FREE", FREE);
+	PyModule_AddIntConstant(m, "CELL", CELL);
+}