1 #include "Python.h"

     2 

     3 #include "code.h"

     4 #include "compile.h"

     5 #include "Python-ast.h"

     6 #include "symtable.h"

     7 

     8 static PyObject *

     9 symtable_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 

    43 static 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 

    50 PyMODINIT_FUNC

    51 init_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 }