cpython-lru_cache-weakref: Lib/symtable.py annotate

annotate Lib/symtable.py @ 28646:193b8aec00ce legacy-trunk

Changes from Jonathan Riehl to allow his pgen extension (PEP 269) to work. This includes some more code that used to be part of pgen in the main parser; I'm okay with that. I'll see if the Windows build needs work next.

author	Guido van Rossum <guido@python.org>
date	Thu, 17 Apr 2003 14:55:42 +0000
parents	dbee2f628a90
children	5c8dc5d218d8

rev	line source
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	1 """Interface to the compiler's internal symbol tables"""
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	2
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	3 import _symtable
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	4 from _symtable import USE, DEF_GLOBAL, DEF_LOCAL, DEF_PARAM, \
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	5 DEF_STAR, DEF_DOUBLESTAR, DEF_INTUPLE, DEF_FREE, \
17552 05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	6 DEF_FREE_GLOBAL, DEF_FREE_CLASS, DEF_IMPORT, DEF_BOUND, \
05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	7 OPT_IMPORT_STAR, OPT_EXEC, OPT_BARE_EXEC
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	8
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	9 import weakref
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	10
17192 1153796d21c2 flesh out __all__ Jeremy Hylton <jeremy@alum.mit.edu> parents: 17161 diff changeset	11 __all__ = ["symtable", "SymbolTable", "newSymbolTable", "Class",
1153796d21c2 flesh out __all__ Jeremy Hylton <jeremy@alum.mit.edu> parents: 17161 diff changeset	12 "Function", "Symbol"]
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	13
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	14 def symtable(code, filename, compile_type):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	15 raw = _symtable.symtable(code, filename, compile_type)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	16 return newSymbolTable(raw[0], filename)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	17
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	18 class SymbolTableFactory:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	19 def __init__(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	20 self.__memo = weakref.WeakValueDictionary()
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	21
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	22 def new(self, table, filename):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	23 if table.type == _symtable.TYPE_FUNCTION:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	24 return Function(table, filename)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	25 if table.type == _symtable.TYPE_CLASS:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	26 return Class(table, filename)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	27 return SymbolTable(table, filename)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	28
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	29 def __call__(self, table, filename):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	30 key = table, filename
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	31 obj = self.__memo.get(key, None)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	32 if obj is None:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	33 obj = self.__memo[key] = self.new(table, filename)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	34 return obj
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	35
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	36 newSymbolTable = SymbolTableFactory()
17245 0377f927cafa Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 17192 diff changeset	37
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	38 def is_free(flags):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	39 if (flags & (USE \| DEF_FREE)) \
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	40 and (flags & (DEF_LOCAL \| DEF_PARAM \| DEF_GLOBAL)):
22830 a9dd5e6d6805 Convert a pile of obvious "yes/no" functions to return bool. Tim Peters <tim.peters@gmail.com> parents: 18524 diff changeset	41 return True
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	42 if flags & DEF_FREE_CLASS:
22830 a9dd5e6d6805 Convert a pile of obvious "yes/no" functions to return bool. Tim Peters <tim.peters@gmail.com> parents: 18524 diff changeset	43 return True
a9dd5e6d6805 Convert a pile of obvious "yes/no" functions to return bool. Tim Peters <tim.peters@gmail.com> parents: 18524 diff changeset	44 return False
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	45
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	46 class SymbolTable:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	47 def __init__(self, raw_table, filename):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	48 self._table = raw_table
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	49 self._filename = filename
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	50 self._symbols = {}
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	51
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	52 def __repr__(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	53 if self.__class__ == SymbolTable:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	54 kind = ""
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	55 else:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	56 kind = "%s " % self.__class__.__name__
17245 0377f927cafa Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 17192 diff changeset	57
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	58 if self._table.name == "global":
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	59 return "<%sSymbolTable for module %s>" % (kind, self._filename)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	60 else:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	61 return "<%sSymbolTable for %s in %s>" % (kind, self._table.name,
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	62 self._filename)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	63
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	64 def get_type(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	65 if self._table.type == _symtable.TYPE_MODULE:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	66 return "module"
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	67 if self._table.type == _symtable.TYPE_FUNCTION:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	68 return "function"
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	69 if self._table.type == _symtable.TYPE_CLASS:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	70 return "class"
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	71 assert self._table.type in (1, 2, 3), \
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	72 "unexpected type: %s" % self._table.type
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	73
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	74 def get_id(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	75 return self._table.id
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	76
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	77 def get_name(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	78 return self._table.name
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	79
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	80 def get_lineno(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	81 return self._table.lineno
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	82
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	83 def is_optimized(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	84 return bool(self._table.type == _symtable.TYPE_FUNCTION
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	85 and not self._table.optimized)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	86
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	87 def is_nested(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	88 return bool(self._table.nested)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	89
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	90 def has_children(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	91 return bool(self._table.children)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	92
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	93 def has_exec(self):
17552 05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	94 """Return true if the scope uses exec"""
05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	95 return bool(self._table.optimized & (OPT_EXEC \| OPT_BARE_EXEC))
05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	96
05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	97 def has_import_star(self):
05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	98 """Return true if the scope uses import *"""
05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	99 return bool(self._table.optimized & OPT_IMPORT_STAR)
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	100
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	101 def get_identifiers(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	102 return self._table.symbols.keys()
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	103
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	104 def lookup(self, name):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	105 sym = self._symbols.get(name)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	106 if sym is None:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	107 flags = self._table.symbols[name]
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	108 namespaces = self.__check_children(name)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	109 sym = self._symbols[name] = Symbol(name, flags, namespaces)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	110 return sym
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	111
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	112 def get_symbols(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	113 return [self.lookup(ident) for ident in self.get_identifiers()]
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	114
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	115 def __check_children(self, name):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	116 return [newSymbolTable(st, self._filename)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	117 for st in self._table.children
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	118 if st.name == name]
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	119
17192 1153796d21c2 flesh out __all__ Jeremy Hylton <jeremy@alum.mit.edu> parents: 17161 diff changeset	120 def get_children(self):
1153796d21c2 flesh out __all__ Jeremy Hylton <jeremy@alum.mit.edu> parents: 17161 diff changeset	121 return [newSymbolTable(st, self._filename)
1153796d21c2 flesh out __all__ Jeremy Hylton <jeremy@alum.mit.edu> parents: 17161 diff changeset	122 for st in self._table.children]
1153796d21c2 flesh out __all__ Jeremy Hylton <jeremy@alum.mit.edu> parents: 17161 diff changeset	123
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	124 class Function(SymbolTable):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	125
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	126 # Default values for instance variables
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	127 __params = None
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	128 __locals = None
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	129 __frees = None
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	130 __globals = None
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	131
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	132 def __idents_matching(self, test_func):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	133 return tuple([ident for ident in self.get_identifiers()
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	134 if test_func(self._table.symbols[ident])])
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	135
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	136 def get_parameters(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	137 if self.__params is None:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	138 self.__params = self.__idents_matching(lambda x:x & DEF_PARAM)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	139 return self.__params
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	140
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	141 def get_locals(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	142 if self.__locals is None:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	143 self.__locals = self.__idents_matching(lambda x:x & DEF_BOUND)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	144 return self.__locals
17245 0377f927cafa Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 17192 diff changeset	145
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	146 def get_globals(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	147 if self.__globals is None:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	148 glob = DEF_GLOBAL \| DEF_FREE_GLOBAL
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	149 self.__globals = self.__idents_matching(lambda x:x & glob)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	150 return self.__globals
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	151
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	152 def get_frees(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	153 if self.__frees is None:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	154 self.__frees = self.__idents_matching(is_free)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	155 return self.__frees
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	156
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	157 class Class(SymbolTable):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	158
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	159 __methods = None
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	160
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	161 def get_methods(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	162 if self.__methods is None:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	163 d = {}
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	164 for st in self._table.children:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	165 d[st.name] = 1
23634 dbee2f628a90 Replaced .keys() with dictionary iterators Raymond Hettinger <python@rcn.com> parents: 22830 diff changeset	166 self.__methods = tuple(d)
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	167 return self.__methods
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	168
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	169 class Symbol:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	170 def __init__(self, name, flags, namespaces=None):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	171 self.__name = name
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	172 self.__flags = flags
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	173 self.__namespaces = namespaces or ()
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	174
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	175 def __repr__(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	176 return "<symbol '%s'>" % self.__name
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	177
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	178 def get_name(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	179 return self.__name
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	180
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	181 def is_referenced(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	182 return bool(self.__flags & _symtable.USE)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	183
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	184 def is_parameter(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	185 return bool(self.__flags & DEF_PARAM)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	186
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	187 def is_global(self):
17245 0377f927cafa Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 17192 diff changeset	188 return bool((self.__flags & DEF_GLOBAL)
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	189 or (self.__flags & DEF_FREE_GLOBAL))
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	190
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	191 def is_vararg(self):
17552 05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	192 return bool(self.__flags & DEF_STAR)
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	193
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	194 def is_keywordarg(self):
17552 05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	195 return bool(self.__flags & DEF_DOUBLESTAR)
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	196
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	197 def is_local(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	198 return bool(self.__flags & DEF_BOUND)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	199
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	200 def is_free(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	201 if (self.__flags & (USE \| DEF_FREE)) \
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	202 and (self.__flags & (DEF_LOCAL \| DEF_PARAM \| DEF_GLOBAL)):
22830 a9dd5e6d6805 Convert a pile of obvious "yes/no" functions to return bool. Tim Peters <tim.peters@gmail.com> parents: 18524 diff changeset	203 return True
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	204 if self.__flags & DEF_FREE_CLASS:
22830 a9dd5e6d6805 Convert a pile of obvious "yes/no" functions to return bool. Tim Peters <tim.peters@gmail.com> parents: 18524 diff changeset	205 return True
a9dd5e6d6805 Convert a pile of obvious "yes/no" functions to return bool. Tim Peters <tim.peters@gmail.com> parents: 18524 diff changeset	206 return False
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	207
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	208 def is_imported(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	209 return bool(self.__flags & DEF_IMPORT)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	210
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	211 def is_assigned(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	212 return bool(self.__flags & DEF_LOCAL)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	213
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	214 def is_in_tuple(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	215 return bool(self.__flags & DEF_INTUPLE)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	216
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	217 def is_namespace(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	218 """Returns true if name binding introduces new namespace.
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	219
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	220 If the name is used as the target of a function or class
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	221 statement, this will be true.
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	222
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	223 Note that a single name can be bound to multiple objects. If
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	224 is_namespace() is true, the name may also be bound to other
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	225 objects, like an int or list, that does not introduce a new
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	226 namespace.
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	227 """
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	228 return bool(self.__namespaces)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	229
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	230 def get_namespaces(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	231 """Return a list of namespaces bound to this name"""
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	232 return self.__namespaces
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	233
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	234 def get_namespace(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	235 """Returns the single namespace bound to this name.
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	236
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	237 Raises ValueError if the name is bound to multiple namespaces.
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	238 """
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	239 if len(self.__namespaces) != 1:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	240 raise ValueError, "name is bound to multiple namespaces"
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	241 return self.__namespaces[0]
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	242
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	243 if __name__ == "__main__":
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	244 import os, sys
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	245 src = open(sys.argv[0]).read()
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	246 mod = symtable(src, os.path.split(sys.argv[0])[1], "exec")
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	247 for ident in mod.get_identifiers():
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	248 info = mod.lookup(ident)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	249 print info, info.is_local(), info.is_namespace()

Mercurial > cpython-lru_cache-weakref

annotate Lib/symtable.py @ 28646:193b8aec00ce legacy-trunk