1#include "Python.h"
2#include "pycore_symtable.h" // struct symtable
3
4#include "clinic/symtablemodule.c.h"
5/*[clinic input]
6module _symtable
7[clinic start generated code]*/
8/*[clinic end generated code: output=da39a3ee5e6b4b0d input=f4685845a7100605]*/
9
10
11/*[clinic input]
12_symtable.symtable
13
14 source: object
15 filename: object(converter='PyUnicode_FSDecoder')
16 startstr: str
17 /
18
19Return symbol and scope dictionaries used internally by compiler.
20[clinic start generated code]*/
21
22static PyObject *
23_symtable_symtable_impl(PyObject *module, PyObject *source,
24 PyObject *filename, const char *startstr)
25/*[clinic end generated code: output=59eb0d5fc7285ac4 input=9dd8a50c0c36a4d7]*/
26{
27 struct symtable *st;
28 PyObject *t;
29 int start;
30 PyCompilerFlags cf = _PyCompilerFlags_INIT;
31 PyObject *source_copy = NULL;
32
33 cf.cf_flags = PyCF_SOURCE_IS_UTF8;
34
35 const char *str = _Py_SourceAsString(source, "symtable", "string or bytes", &cf, &source_copy);
36 if (str == NULL) {
37 return NULL;
38 }
39
40 if (strcmp(startstr, "exec") == 0)
41 start = Py_file_input;
42 else if (strcmp(startstr, "eval") == 0)
43 start = Py_eval_input;
44 else if (strcmp(startstr, "single") == 0)
45 start = Py_single_input;
46 else {
47 PyErr_SetString(PyExc_ValueError,
48 "symtable() arg 3 must be 'exec' or 'eval' or 'single'");
49 Py_DECREF(filename);
50 Py_XDECREF(source_copy);
51 return NULL;
52 }
53 st = _Py_SymtableStringObjectFlags(str, filename, start, &cf);
54 Py_DECREF(filename);
55 Py_XDECREF(source_copy);
56 if (st == NULL) {
57 return NULL;
58 }
59 t = (PyObject *)st->st_top;
60 Py_INCREF(t);
61 _PySymtable_Free(st);
62 return t;
63}
64
65static PyMethodDef symtable_methods[] = {
66 _SYMTABLE_SYMTABLE_METHODDEF
67 {NULL, NULL} /* sentinel */
68};
69
70static int
71symtable_init_stentry_type(PyObject *m)
72{
73 return PyType_Ready(&PySTEntry_Type);
74}
75
76static int
77symtable_init_constants(PyObject *m)
78{
79 if (PyModule_AddIntMacro(m, USE) < 0) return -1;
80 if (PyModule_AddIntMacro(m, DEF_GLOBAL) < 0) return -1;
81 if (PyModule_AddIntMacro(m, DEF_NONLOCAL) < 0) return -1;
82 if (PyModule_AddIntMacro(m, DEF_LOCAL) < 0) return -1;
83 if (PyModule_AddIntMacro(m, DEF_PARAM) < 0) return -1;
84 if (PyModule_AddIntMacro(m, DEF_FREE) < 0) return -1;
85 if (PyModule_AddIntMacro(m, DEF_FREE_CLASS) < 0) return -1;
86 if (PyModule_AddIntMacro(m, DEF_IMPORT) < 0) return -1;
87 if (PyModule_AddIntMacro(m, DEF_BOUND) < 0) return -1;
88 if (PyModule_AddIntMacro(m, DEF_ANNOT) < 0) return -1;
89
90 if (PyModule_AddIntConstant(m, "TYPE_FUNCTION", FunctionBlock) < 0)
91 return -1;
92 if (PyModule_AddIntConstant(m, "TYPE_CLASS", ClassBlock) < 0) return -1;
93 if (PyModule_AddIntConstant(m, "TYPE_MODULE", ModuleBlock) < 0)
94 return -1;
95
96 if (PyModule_AddIntMacro(m, LOCAL) < 0) return -1;
97 if (PyModule_AddIntMacro(m, GLOBAL_EXPLICIT) < 0) return -1;
98 if (PyModule_AddIntMacro(m, GLOBAL_IMPLICIT) < 0) return -1;
99 if (PyModule_AddIntMacro(m, FREE) < 0) return -1;
100 if (PyModule_AddIntMacro(m, CELL) < 0) return -1;
101
102 if (PyModule_AddIntConstant(m, "SCOPE_OFF", SCOPE_OFFSET) < 0) return -1;
103 if (PyModule_AddIntMacro(m, SCOPE_MASK) < 0) return -1;
104
105 return 0;
106}
107
108static PyModuleDef_Slot symtable_slots[] = {
109 {Py_mod_exec, symtable_init_stentry_type},
110 {Py_mod_exec, symtable_init_constants},
111 {0, NULL}
112};
113
114static struct PyModuleDef symtablemodule = {
115 PyModuleDef_HEAD_INIT,
116 .m_name = "_symtable",
117 .m_size = 0,
118 .m_methods = symtable_methods,
119 .m_slots = symtable_slots,
120};
121
122PyMODINIT_FUNC
123PyInit__symtable(void)
124{
125 return PyModuleDef_Init(&symtablemodule);
126}
127