| 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 = (PyObject *)st->st_top;
|
|---|
| 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 | if (PyType_Ready(&PySTEntry_Type) < 0)
|
|---|
| 56 | return;
|
|---|
| 57 |
|
|---|
| 58 | m = Py_InitModule("_symtable", symtable_methods);
|
|---|
| 59 | if (m == NULL)
|
|---|
| 60 | return;
|
|---|
| 61 | PyModule_AddIntConstant(m, "USE", USE);
|
|---|
| 62 | PyModule_AddIntConstant(m, "DEF_GLOBAL", DEF_GLOBAL);
|
|---|
| 63 | PyModule_AddIntConstant(m, "DEF_LOCAL", DEF_LOCAL);
|
|---|
| 64 | PyModule_AddIntConstant(m, "DEF_PARAM", DEF_PARAM);
|
|---|
| 65 | PyModule_AddIntConstant(m, "DEF_FREE", DEF_FREE);
|
|---|
| 66 | PyModule_AddIntConstant(m, "DEF_FREE_CLASS", DEF_FREE_CLASS);
|
|---|
| 67 | PyModule_AddIntConstant(m, "DEF_IMPORT", DEF_IMPORT);
|
|---|
| 68 | PyModule_AddIntConstant(m, "DEF_BOUND", DEF_BOUND);
|
|---|
| 69 |
|
|---|
| 70 | PyModule_AddIntConstant(m, "TYPE_FUNCTION", FunctionBlock);
|
|---|
| 71 | PyModule_AddIntConstant(m, "TYPE_CLASS", ClassBlock);
|
|---|
| 72 | PyModule_AddIntConstant(m, "TYPE_MODULE", ModuleBlock);
|
|---|
| 73 |
|
|---|
| 74 | PyModule_AddIntConstant(m, "OPT_IMPORT_STAR", OPT_IMPORT_STAR);
|
|---|
| 75 | PyModule_AddIntConstant(m, "OPT_EXEC", OPT_EXEC);
|
|---|
| 76 | PyModule_AddIntConstant(m, "OPT_BARE_EXEC", OPT_BARE_EXEC);
|
|---|
| 77 |
|
|---|
| 78 | PyModule_AddIntConstant(m, "LOCAL", LOCAL);
|
|---|
| 79 | PyModule_AddIntConstant(m, "GLOBAL_EXPLICIT", GLOBAL_EXPLICIT);
|
|---|
| 80 | PyModule_AddIntConstant(m, "GLOBAL_IMPLICIT", GLOBAL_IMPLICIT);
|
|---|
| 81 | PyModule_AddIntConstant(m, "FREE", FREE);
|
|---|
| 82 | PyModule_AddIntConstant(m, "CELL", CELL);
|
|---|
| 83 |
|
|---|
| 84 | PyModule_AddIntConstant(m, "SCOPE_OFF", SCOPE_OFF);
|
|---|
| 85 | PyModule_AddIntConstant(m, "SCOPE_MASK", SCOPE_MASK);
|
|---|
| 86 | }
|
|---|
