Staging
v0.5.1
https://github.com/python/cpython
Raw File
Tip revision: 4ce90b3a994e3eef18e43ac5a1f6c9acd0b16b8c authored by cvs2svn on 30 May 2003, 20:57:36 UTC
This commit was manufactured by cvs2svn to create tag 'r223'.
Tip revision: 4ce90b3
demo.py
#! /usr/bin/env python

"""Print names of all methods defined in module

This script demonstrates use of the visitor interface of the compiler
package.
"""

import compiler

class MethodFinder:
    """Print the names of all the methods

    Each visit method takes two arguments, the node and its current
    scope.  The scope is the name of the current class or None.
    """
    
    def visitClass(self, node, scope=None):
        self.visit(node.code, node.name)

    def visitFunction(self, node, scope=None):
        if scope is not None:
            print "%s.%s" % (scope, node.name)
        self.visit(node.code, None)
            
def main(files):
    mf = MethodFinder()
    for file in files:
        f = open(file)
        buf = f.read()
        f.close()
        ast = compiler.parse(buf)
        compiler.walk(ast, mf)

if __name__ == "__main__":
    import sys
    
    main(sys.argv[1:])
back to top