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

annotate Lib/symtable.py @ 25838:8540887bcec6 2.2

Backport of 2.183: Fix memory leak in add_subclass() found by NealN with valgrind.

author	Guido van Rossum <guido@python.org>
date	Fri, 18 Oct 2002 13:42:21 +0000
parents	1fbcf549c1c4
children	a9dd5e6d6805

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 bool(x):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	39 """Helper to force boolean result to 1 or 0"""
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	40 if x:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	41 return 1
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	42 return 0
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	43
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	44 def is_free(flags):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	45 if (flags & (USE \| DEF_FREE)) \
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	46 and (flags & (DEF_LOCAL \| DEF_PARAM \| DEF_GLOBAL)):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	47 return 1
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	48 if flags & DEF_FREE_CLASS:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	49 return 1
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	50 return 0
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 class SymbolTable:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	53 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	54 self._table = raw_table
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	55 self._filename = filename
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	56 self._symbols = {}
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	57
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	58 def __repr__(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	59 if self.__class__ == SymbolTable:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	60 kind = ""
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	61 else:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	62 kind = "%s " % self.__class__.__name__
17245 0377f927cafa Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 17192 diff changeset	63
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	64 if self._table.name == "global":
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	65 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	66 else:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	67 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	68 self._filename)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	69
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	70 def get_type(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	71 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	72 return "module"
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	73 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	74 return "function"
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	75 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	76 return "class"
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	77 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	78 "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	79
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	80 def get_id(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	81 return self._table.id
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 get_name(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	84 return self._table.name
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	85
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	86 def get_lineno(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	87 return self._table.lineno
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	88
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	89 def is_optimized(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	90 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	91 and not self._table.optimized)
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 is_nested(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	94 return bool(self._table.nested)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	95
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	96 def has_children(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	97 return bool(self._table.children)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	98
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	99 def has_exec(self):
17552 05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	100 """Return true if the scope uses exec"""
05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	101 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	102
05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	103 def has_import_star(self):
05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	104 """Return true if the scope uses import *"""
05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	105 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	106
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	107 def get_identifiers(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	108 return self._table.symbols.keys()
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	109
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	110 def lookup(self, name):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	111 sym = self._symbols.get(name)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	112 if sym is None:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	113 flags = self._table.symbols[name]
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	114 namespaces = self.__check_children(name)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	115 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	116 return sym
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	117
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	118 def get_symbols(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	119 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	120
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	121 def __check_children(self, name):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	122 return [newSymbolTable(st, self._filename)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	123 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	124 if st.name == name]
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	125
17192 1153796d21c2 flesh out __all__ Jeremy Hylton <jeremy@alum.mit.edu> parents: 17161 diff changeset	126 def get_children(self):
1153796d21c2 flesh out __all__ Jeremy Hylton <jeremy@alum.mit.edu> parents: 17161 diff changeset	127 return [newSymbolTable(st, self._filename)
1153796d21c2 flesh out __all__ Jeremy Hylton <jeremy@alum.mit.edu> parents: 17161 diff changeset	128 for st in self._table.children]
1153796d21c2 flesh out __all__ Jeremy Hylton <jeremy@alum.mit.edu> parents: 17161 diff changeset	129
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	130 class Function(SymbolTable):
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 # Default values for instance variables
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	133 __params = None
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	134 __locals = None
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	135 __frees = None
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	136 __globals = None
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	137
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	138 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	139 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	140 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	141
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	142 def get_parameters(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	143 if self.__params is None:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	144 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	145 return self.__params
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	146
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	147 def get_locals(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	148 if self.__locals is None:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	149 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	150 return self.__locals
17245 0377f927cafa Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 17192 diff changeset	151
17161 e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	152 def get_globals(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	153 if self.__globals is None:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	154 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	155 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	156 return self.__globals
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	157
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	158 def get_frees(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	159 if self.__frees is None:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	160 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	161 return self.__frees
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	162
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	163 class Class(SymbolTable):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	164
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	165 __methods = None
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	166
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	167 def get_methods(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	168 if self.__methods is None:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	169 d = {}
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	170 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	171 d[st.name] = 1
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	172 self.__methods = tuple(d.keys())
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	173 return self.__methods
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 class Symbol:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	176 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	177 self.__name = name
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	178 self.__flags = flags
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	179 self.__namespaces = namespaces or ()
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 __repr__(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	182 return "<symbol '%s'>" % self.__name
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 get_name(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	185 return self.__name
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_referenced(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	188 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	189
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	190 def is_parameter(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	191 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	192
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	193 def is_global(self):
17245 0377f927cafa Whitespace normalization. Tim Peters <tim.peters@gmail.com> parents: 17192 diff changeset	194 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	195 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	196
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	197 def is_vararg(self):
17552 05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	198 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	199
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	200 def is_keywordarg(self):
17552 05436a74012d Fix three PyChecker-detected gotchas. Jeremy Hylton <jeremy@alum.mit.edu> parents: 17245 diff changeset	201 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	202
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	203 def is_local(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	204 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	205
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	206 def is_free(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	207 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	208 and (self.__flags & (DEF_LOCAL \| DEF_PARAM \| DEF_GLOBAL)):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	209 return 1
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	210 if self.__flags & DEF_FREE_CLASS:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	211 return 1
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	212 return 0
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_imported(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_IMPORT)
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_assigned(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	218 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	219
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	220 def is_in_tuple(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	221 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	222
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	223 def is_namespace(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	224 """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	225
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	226 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	227 statement, this will be true.
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	228
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	229 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	230 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	231 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	232 namespace.
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 return bool(self.__namespaces)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	235
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	236 def get_namespaces(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	237 """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	238 return self.__namespaces
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	239
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	240 def get_namespace(self):
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	241 """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	242
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	243 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	244 """
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	245 if len(self.__namespaces) != 1:
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	246 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	247 return self.__namespaces[0]
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	248
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	249 if __name__ == "__main__":
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	250 import os, sys
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	251 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	252 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	253 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	254 info = mod.lookup(ident)
e3b5d03e3d7d First cut at a high-level symbol table interface Jeremy Hylton <jeremy@alum.mit.edu> parents: diff changeset	255 print info, info.is_local(), info.is_namespace()

Mercurial > cpython-lru_cache-weakref

annotate Lib/symtable.py @ 25838:8540887bcec6 2.2