/Modules/symtablemodule.c

 1#include "Python.h"
 2
 3#include "code.h"
 4#include "compile.h"
 5#include "Python-ast.h"
 6#include "symtable.h"
 7
 8static PyObject *
 9symtable_symtable(PyObject *self, PyObject *args)
10{
11	struct symtable *st;
12	PyObject *t;
13
14	char *str;
15	char *filename;
16	char *startstr;
17	int start;
18
19	if (!PyArg_ParseTuple(args, "sss:symtable", &str, &filename, 
20			      &startstr))
21		return NULL;
22	if (strcmp(startstr, "exec") == 0)
23		start = Py_file_input;
24	else if (strcmp(startstr, "eval") == 0)
25		start = Py_eval_input;
26	else if (strcmp(startstr, "single") == 0)
27		start = Py_single_input;
28	else {
29		PyErr_SetString(PyExc_ValueError,
30		   "symtable() arg 3 must be 'exec' or 'eval' or 'single'");
31		return NULL;
32	}
33	st = Py_SymtableString(str, filename, start);
34	if (st == NULL)
35		return NULL;
36	t = st->st_symbols;
37	Py_INCREF(t);
38	PyMem_Free((void *)st->st_future);
39	PySymtable_Free(st);
40	return t;
41}
42
43static PyMethodDef symtable_methods[] = {
44	{"symtable",	symtable_symtable,	METH_VARARGS,
45	 PyDoc_STR("Return symbol and scope dictionaries"
46	 	   " used internally by compiler.")},
47	{NULL,		NULL}		/* sentinel */
48};
49
50PyMODINIT_FUNC
51init_symtable(void)
52{
53	PyObject *m;
54
55	m = Py_InitModule("_symtable", symtable_methods);
56	if (m == NULL)
57		return;
58	PyModule_AddIntConstant(m, "USE", USE);
59	PyModule_AddIntConstant(m, "DEF_GLOBAL", DEF_GLOBAL);
60	PyModule_AddIntConstant(m, "DEF_LOCAL", DEF_LOCAL);
61	PyModule_AddIntConstant(m, "DEF_PARAM", DEF_PARAM);
62	PyModule_AddIntConstant(m, "DEF_STAR", DEF_STAR);
63	PyModule_AddIntConstant(m, "DEF_DOUBLESTAR", DEF_DOUBLESTAR);
64	PyModule_AddIntConstant(m, "DEF_INTUPLE", DEF_INTUPLE);
65	PyModule_AddIntConstant(m, "DEF_FREE", DEF_FREE);
66	PyModule_AddIntConstant(m, "DEF_FREE_GLOBAL", DEF_FREE_GLOBAL);
67	PyModule_AddIntConstant(m, "DEF_FREE_CLASS", DEF_FREE_CLASS);
68	PyModule_AddIntConstant(m, "DEF_IMPORT", DEF_IMPORT);
69	PyModule_AddIntConstant(m, "DEF_BOUND", DEF_BOUND);
70
71	PyModule_AddIntConstant(m, "TYPE_FUNCTION", FunctionBlock);
72	PyModule_AddIntConstant(m, "TYPE_CLASS", ClassBlock);
73	PyModule_AddIntConstant(m, "TYPE_MODULE", ModuleBlock);
74
75	PyModule_AddIntConstant(m, "OPT_IMPORT_STAR", OPT_IMPORT_STAR);
76	PyModule_AddIntConstant(m, "OPT_EXEC", OPT_EXEC);
77	PyModule_AddIntConstant(m, "OPT_BARE_EXEC", OPT_BARE_EXEC);
78
79	PyModule_AddIntConstant(m, "LOCAL", LOCAL);
80	PyModule_AddIntConstant(m, "GLOBAL_EXPLICIT", GLOBAL_EXPLICIT);
81	PyModule_AddIntConstant(m, "GLOBAL_IMPLICIT", GLOBAL_IMPLICIT);
82	PyModule_AddIntConstant(m, "FREE", FREE);
83	PyModule_AddIntConstant(m, "CELL", CELL);
84
85	PyModule_AddIntConstant(m, "SCOPE_OFF", SCOPE_OFF);
86	PyModule_AddIntConstant(m, "SCOPE_MASK", SCOPE_MASK);
87}