Gitiles
Code Review Sign In
gerrit-public.fairphone.software / platform / external / python / cpython3 / 2021100d6daec7ed40f9a47c97184d3fe708ed8d / . / Modules / symtablemodule.c
blob: f66b5989622abafdd40db446997610b987491600 [file] [log] [blame]
Jeremy Hylton5e7cb242001-02-02 18:24:26 +0000[diff] [blame]1#include "Python.h"
2
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000[diff] [blame]3#include "code.h"
Jeremy Hylton5e7cb242001-02-02 18:24:26 +0000[diff] [blame]4#include "compile.h"
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000[diff] [blame]5#include "Python-ast.h"
Jeremy Hylton5e7cb242001-02-02 18:24:26 +0000[diff] [blame]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;
Nick Coghlan650f0d02007-04-15 12:05:43 +0000[diff] [blame]36 t = st->st_blocks;
Raymond Hettinger8ae46892003-10-12 19:09:37 +0000[diff] [blame]37 Py_INCREF(t);
Jeremy Hylton2554dd92001-12-06 14:34:58 +0000[diff] [blame]38 PyMem_Free((void *)st->st_future);
Jeremy Hylton5e7cb242001-02-02 18:24:26 +0000[diff] [blame]39 PySymtable_Free(st);
40 return t;
41}
42
43static PyMethodDef symtable_methods[] = {
44 {"symtable", symtable_symtable, METH_VARARGS,
Neal Norwitz200788c2002-08-13 22:20:41 +0000[diff] [blame]45 PyDoc_STR("Return symbol and scope dictionaries"
46 " used internally by compiler.")},
Jeremy Hylton5e7cb242001-02-02 18:24:26 +0000[diff] [blame]47 {NULL, NULL} /* sentinel */
48};
49
Martin v. Löwis1a214512008-06-11 05:26:20 +0000[diff] [blame]50static struct PyModuleDef symtablemodule = {
51 PyModuleDef_HEAD_INIT,
52 "_symtable",
53 NULL,
54 -1,
55 symtable_methods,
56 NULL,
57 NULL,
58 NULL,
59 NULL
60};
61
Mark Hammondfe51c6d2002-08-02 02:27:13 +0000[diff] [blame]62PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +0000[diff] [blame]63PyInit__symtable(void)
Jeremy Hylton5e7cb242001-02-02 18:24:26 +0000[diff] [blame]64{
65 PyObject *m;
66
Martin v. Löwis1a214512008-06-11 05:26:20 +0000[diff] [blame]67 m = PyModule_Create(&symtablemodule);
Neal Norwitz1ac754f2006-01-19 06:09:39 +0000[diff] [blame]68 if (m == NULL)
Martin v. Löwis1a214512008-06-11 05:26:20 +0000[diff] [blame]69 return NULL;
Jeremy Hylton5e7cb242001-02-02 18:24:26 +0000[diff] [blame]70 PyModule_AddIntConstant(m, "USE", USE);
71 PyModule_AddIntConstant(m, "DEF_GLOBAL", DEF_GLOBAL);
72 PyModule_AddIntConstant(m, "DEF_LOCAL", DEF_LOCAL);
73 PyModule_AddIntConstant(m, "DEF_PARAM", DEF_PARAM);
Jeremy Hylton5e7cb242001-02-02 18:24:26 +0000[diff] [blame]74 PyModule_AddIntConstant(m, "DEF_FREE", DEF_FREE);
Jeremy Hylton5e7cb242001-02-02 18:24:26 +0000[diff] [blame]75 PyModule_AddIntConstant(m, "DEF_FREE_CLASS", DEF_FREE_CLASS);
76 PyModule_AddIntConstant(m, "DEF_IMPORT", DEF_IMPORT);
Jeremy Hylton8bf395f2001-03-22 23:10:44 +0000[diff] [blame]77 PyModule_AddIntConstant(m, "DEF_BOUND", DEF_BOUND);
Jeremy Hylton5e7cb242001-02-02 18:24:26 +0000[diff] [blame]78
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000[diff] [blame]79 PyModule_AddIntConstant(m, "TYPE_FUNCTION", FunctionBlock);
80 PyModule_AddIntConstant(m, "TYPE_CLASS", ClassBlock);
81 PyModule_AddIntConstant(m, "TYPE_MODULE", ModuleBlock);
Jeremy Hylton5e7cb242001-02-02 18:24:26 +0000[diff] [blame]82
Jeremy Hylton43454762001-04-16 18:42:13 +0000[diff] [blame]83 PyModule_AddIntConstant(m, "OPT_IMPORT_STAR", OPT_IMPORT_STAR);
Georg Brandl7cae87c2006-09-06 06:51:57 +0000[diff] [blame]84 PyModule_AddIntConstant(m, "OPT_TOPLEVEL", OPT_TOPLEVEL);
Jeremy Hylton43454762001-04-16 18:42:13 +0000[diff] [blame]85
Jeremy Hylton5e7cb242001-02-02 18:24:26 +0000[diff] [blame]86 PyModule_AddIntConstant(m, "LOCAL", LOCAL);
87 PyModule_AddIntConstant(m, "GLOBAL_EXPLICIT", GLOBAL_EXPLICIT);
88 PyModule_AddIntConstant(m, "GLOBAL_IMPLICIT", GLOBAL_IMPLICIT);
89 PyModule_AddIntConstant(m, "FREE", FREE);
90 PyModule_AddIntConstant(m, "CELL", CELL);
Benjamin Peterson55e00f22008-08-17 18:02:44 +0000[diff] [blame]91
92 PyModule_AddIntConstant(m, "SCOPE_OFF", SCOPE_OFFSET);
93 PyModule_AddIntConstant(m, "SCOPE_MASK", SCOPE_MASK);
94
Martin v. Löwis1a214512008-06-11 05:26:20 +0000[diff] [blame]95 if (PyErr_Occurred()) {
96 Py_DECREF(m);
97 m = 0;
98 }
99 return m;
Jeremy Hylton5e7cb242001-02-02 18:24:26 +0000[diff] [blame]100}