Modules/symtablemodule.c

   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_FREE", DEF_FREE);
  63         PyModule_AddIntConstant(m, "DEF_FREE_CLASS", DEF_FREE_CLASS);
  64         PyModule_AddIntConstant(m, "DEF_IMPORT", DEF_IMPORT);
  65         PyModule_AddIntConstant(m, "DEF_BOUND", DEF_BOUND);
  66
  67         PyModule_AddIntConstant(m, "TYPE_FUNCTION", FunctionBlock);
  68         PyModule_AddIntConstant(m, "TYPE_CLASS", ClassBlock);
  69         PyModule_AddIntConstant(m, "TYPE_MODULE", ModuleBlock);
  70
  71         PyModule_AddIntConstant(m, "OPT_IMPORT_STAR", OPT_IMPORT_STAR);
  72         PyModule_AddIntConstant(m, "OPT_EXEC", OPT_EXEC);
  73         PyModule_AddIntConstant(m, "OPT_BARE_EXEC", OPT_BARE_EXEC);
  74
  75         PyModule_AddIntConstant(m, "LOCAL", LOCAL);
  76         PyModule_AddIntConstant(m, "GLOBAL_EXPLICIT", GLOBAL_EXPLICIT);
  77         PyModule_AddIntConstant(m, "GLOBAL_IMPLICIT", GLOBAL_IMPLICIT);
  78         PyModule_AddIntConstant(m, "FREE", FREE);
  79         PyModule_AddIntConstant(m, "CELL", CELL);
  80
  81         PyModule_AddIntConstant(m, "SCOPE_OFF", SCOPE_OFF);
  82         PyModule_AddIntConstant(m, "SCOPE_MASK", SCOPE_MASK);
  83 }