--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/python-2.5.2/win32/Lib/symtable.py Fri Apr 03 17:19:34 2009 +0100
@@ -0,0 +1,252 @@
+"""Interface to the compiler's internal symbol tables"""
+
+import _symtable
+from _symtable import USE, DEF_GLOBAL, DEF_LOCAL, DEF_PARAM, \
+ DEF_STAR, DEF_DOUBLESTAR, DEF_INTUPLE, DEF_FREE, \
+ DEF_FREE_GLOBAL, DEF_FREE_CLASS, DEF_IMPORT, DEF_BOUND, \
+ OPT_IMPORT_STAR, OPT_EXEC, OPT_BARE_EXEC
+
+import weakref
+
+__all__ = ["symtable", "SymbolTable", "newSymbolTable", "Class",
+ "Function", "Symbol"]
+
+def symtable(code, filename, compile_type):
+ raw = _symtable.symtable(code, filename, compile_type)
+ for top in raw.itervalues():
+ if top.name == 'top':
+ break
+ return newSymbolTable(top, filename)
+
+class SymbolTableFactory:
+ def __init__(self):
+ self.__memo = weakref.WeakValueDictionary()
+
+ def new(self, table, filename):
+ if table.type == _symtable.TYPE_FUNCTION:
+ return Function(table, filename)
+ if table.type == _symtable.TYPE_CLASS:
+ return Class(table, filename)
+ return SymbolTable(table, filename)
+
+ def __call__(self, table, filename):
+ key = table, filename
+ obj = self.__memo.get(key, None)
+ if obj is None:
+ obj = self.__memo[key] = self.new(table, filename)
+ return obj
+
+newSymbolTable = SymbolTableFactory()
+
+def is_free(flags):
+ if (flags & (USE | DEF_FREE)) \
+ and (flags & (DEF_LOCAL | DEF_PARAM | DEF_GLOBAL)):
+ return True
+ if flags & DEF_FREE_CLASS:
+ return True
+ return False
+
+class SymbolTable:
+ def __init__(self, raw_table, filename):
+ self._table = raw_table
+ self._filename = filename
+ self._symbols = {}
+
+ def __repr__(self):
+ if self.__class__ == SymbolTable:
+ kind = ""
+ else:
+ kind = "%s " % self.__class__.__name__
+
+ if self._table.name == "global":
+ return "<%sSymbolTable for module %s>" % (kind, self._filename)
+ else:
+ return "<%sSymbolTable for %s in %s>" % (kind, self._table.name,
+ self._filename)
+
+ def get_type(self):
+ if self._table.type == _symtable.TYPE_MODULE:
+ return "module"
+ if self._table.type == _symtable.TYPE_FUNCTION:
+ return "function"
+ if self._table.type == _symtable.TYPE_CLASS:
+ return "class"
+ assert self._table.type in (1, 2, 3), \
+ "unexpected type: %s" % self._table.type
+
+ def get_id(self):
+ return self._table.id
+
+ def get_name(self):
+ return self._table.name
+
+ def get_lineno(self):
+ return self._table.lineno
+
+ def is_optimized(self):
+ return bool(self._table.type == _symtable.TYPE_FUNCTION
+ and not self._table.optimized)
+
+ def is_nested(self):
+ return bool(self._table.nested)
+
+ def has_children(self):
+ return bool(self._table.children)
+
+ def has_exec(self):
+ """Return true if the scope uses exec"""
+ return bool(self._table.optimized & (OPT_EXEC | OPT_BARE_EXEC))
+
+ def has_import_star(self):
+ """Return true if the scope uses import *"""
+ return bool(self._table.optimized & OPT_IMPORT_STAR)
+
+ def get_identifiers(self):
+ return self._table.symbols.keys()
+
+ def lookup(self, name):
+ sym = self._symbols.get(name)
+ if sym is None:
+ flags = self._table.symbols[name]
+ namespaces = self.__check_children(name)
+ sym = self._symbols[name] = Symbol(name, flags, namespaces)
+ return sym
+
+ def get_symbols(self):
+ return [self.lookup(ident) for ident in self.get_identifiers()]
+
+ def __check_children(self, name):
+ return [newSymbolTable(st, self._filename)
+ for st in self._table.children
+ if st.name == name]
+
+ def get_children(self):
+ return [newSymbolTable(st, self._filename)
+ for st in self._table.children]
+
+class Function(SymbolTable):
+
+ # Default values for instance variables
+ __params = None
+ __locals = None
+ __frees = None
+ __globals = None
+
+ def __idents_matching(self, test_func):
+ return tuple([ident for ident in self.get_identifiers()
+ if test_func(self._table.symbols[ident])])
+
+ def get_parameters(self):
+ if self.__params is None:
+ self.__params = self.__idents_matching(lambda x:x & DEF_PARAM)
+ return self.__params
+
+ def get_locals(self):
+ if self.__locals is None:
+ self.__locals = self.__idents_matching(lambda x:x & DEF_BOUND)
+ return self.__locals
+
+ def get_globals(self):
+ if self.__globals is None:
+ glob = DEF_GLOBAL | DEF_FREE_GLOBAL
+ self.__globals = self.__idents_matching(lambda x:x & glob)
+ return self.__globals
+
+ def get_frees(self):
+ if self.__frees is None:
+ self.__frees = self.__idents_matching(is_free)
+ return self.__frees
+
+class Class(SymbolTable):
+
+ __methods = None
+
+ def get_methods(self):
+ if self.__methods is None:
+ d = {}
+ for st in self._table.children:
+ d[st.name] = 1
+ self.__methods = tuple(d)
+ return self.__methods
+
+class Symbol:
+ def __init__(self, name, flags, namespaces=None):
+ self.__name = name
+ self.__flags = flags
+ self.__namespaces = namespaces or ()
+
+ def __repr__(self):
+ return "<symbol '%s'>" % self.__name
+
+ def get_name(self):
+ return self.__name
+
+ def is_referenced(self):
+ return bool(self.__flags & _symtable.USE)
+
+ def is_parameter(self):
+ return bool(self.__flags & DEF_PARAM)
+
+ def is_global(self):
+ return bool((self.__flags & DEF_GLOBAL)
+ or (self.__flags & DEF_FREE_GLOBAL))
+
+ def is_vararg(self):
+ return bool(self.__flags & DEF_STAR)
+
+ def is_keywordarg(self):
+ return bool(self.__flags & DEF_DOUBLESTAR)
+
+ def is_local(self):
+ return bool(self.__flags & DEF_BOUND)
+
+ def is_free(self):
+ if (self.__flags & (USE | DEF_FREE)) \
+ and (self.__flags & (DEF_LOCAL | DEF_PARAM | DEF_GLOBAL)):
+ return True
+ if self.__flags & DEF_FREE_CLASS:
+ return True
+ return False
+
+ def is_imported(self):
+ return bool(self.__flags & DEF_IMPORT)
+
+ def is_assigned(self):
+ return bool(self.__flags & DEF_LOCAL)
+
+ def is_in_tuple(self):
+ return bool(self.__flags & DEF_INTUPLE)
+
+ def is_namespace(self):
+ """Returns true if name binding introduces new namespace.
+
+ If the name is used as the target of a function or class
+ statement, this will be true.
+
+ Note that a single name can be bound to multiple objects. If
+ is_namespace() is true, the name may also be bound to other
+ objects, like an int or list, that does not introduce a new
+ namespace.
+ """
+ return bool(self.__namespaces)
+
+ def get_namespaces(self):
+ """Return a list of namespaces bound to this name"""
+ return self.__namespaces
+
+ def get_namespace(self):
+ """Returns the single namespace bound to this name.
+
+ Raises ValueError if the name is bound to multiple namespaces.
+ """
+ if len(self.__namespaces) != 1:
+ raise ValueError, "name is bound to multiple namespaces"
+ return self.__namespaces[0]
+
+if __name__ == "__main__":
+ import os, sys
+ src = open(sys.argv[0]).read()
+ mod = symtable(src, os.path.split(sys.argv[0])[1], "exec")
+ for ident in mod.get_identifiers():
+ info = mod.lookup(ident)
+ print info, info.is_local(), info.is_namespace()