blob: 21ce11c9b6fbaf2765a6593b2a997a973641aa14 [file] [log] [blame]
Jeremy Hylton5e7cb242001-02-02 18:24:26 +00001#include "Python.h"
2
3#include "compile.h"
4#include "symtable.h"
5
6static PyObject *
7symtable_symtable(PyObject *self, PyObject *args)
8{
9 struct symtable *st;
10 PyObject *t;
11
12 char *str;
13 char *filename;
14 char *startstr;
15 int start;
16
17 if (!PyArg_ParseTuple(args, "sss:symtable", &str, &filename,
18 &startstr))
19 return NULL;
20 if (strcmp(startstr, "exec") == 0)
21 start = Py_file_input;
22 else if (strcmp(startstr, "eval") == 0)
23 start = Py_eval_input;
24 else if (strcmp(startstr, "single") == 0)
25 start = Py_single_input;
26 else {
27 PyErr_SetString(PyExc_ValueError,
28 "symtable() arg 3 must be 'exec' or 'eval' or 'single'");
29 return NULL;
30 }
31 st = Py_SymtableString(str, filename, start);
32 if (st == NULL)
33 return NULL;
Jeremy Hyltoncb17ae82001-02-09 22:22:18 +000034 t = Py_BuildValue("O", st->st_symbols);
Jeremy Hylton5e7cb242001-02-02 18:24:26 +000035 PySymtable_Free(st);
36 return t;
37}
38
39static PyMethodDef symtable_methods[] = {
40 {"symtable", symtable_symtable, METH_VARARGS,
41 "Return symbol and scope dictionaries used internally by compiler."},
42 {NULL, NULL} /* sentinel */
43};
44
45DL_EXPORT(void)
46init_symtable(void)
47{
48 PyObject *m;
49
50 m = Py_InitModule("_symtable", symtable_methods);
51 PyModule_AddIntConstant(m, "USE", USE);
52 PyModule_AddIntConstant(m, "DEF_GLOBAL", DEF_GLOBAL);
53 PyModule_AddIntConstant(m, "DEF_LOCAL", DEF_LOCAL);
54 PyModule_AddIntConstant(m, "DEF_PARAM", DEF_PARAM);
55 PyModule_AddIntConstant(m, "DEF_STAR", DEF_STAR);
56 PyModule_AddIntConstant(m, "DEF_DOUBLESTAR", DEF_DOUBLESTAR);
57 PyModule_AddIntConstant(m, "DEF_INTUPLE", DEF_INTUPLE);
58 PyModule_AddIntConstant(m, "DEF_FREE", DEF_FREE);
59 PyModule_AddIntConstant(m, "DEF_FREE_GLOBAL", DEF_FREE_GLOBAL);
60 PyModule_AddIntConstant(m, "DEF_FREE_CLASS", DEF_FREE_CLASS);
61 PyModule_AddIntConstant(m, "DEF_IMPORT", DEF_IMPORT);
Jeremy Hylton8bf395f2001-03-22 23:10:44 +000062 PyModule_AddIntConstant(m, "DEF_BOUND", DEF_BOUND);
Jeremy Hylton5e7cb242001-02-02 18:24:26 +000063
64 PyModule_AddIntConstant(m, "TYPE_FUNCTION", TYPE_FUNCTION);
65 PyModule_AddIntConstant(m, "TYPE_CLASS", TYPE_CLASS);
66 PyModule_AddIntConstant(m, "TYPE_MODULE", TYPE_MODULE);
67
68 PyModule_AddIntConstant(m, "LOCAL", LOCAL);
69 PyModule_AddIntConstant(m, "GLOBAL_EXPLICIT", GLOBAL_EXPLICIT);
70 PyModule_AddIntConstant(m, "GLOBAL_IMPLICIT", GLOBAL_IMPLICIT);
71 PyModule_AddIntConstant(m, "FREE", FREE);
72 PyModule_AddIntConstant(m, "CELL", CELL);
73}