OILS / opy / tools / demo.py View on Github | oilshell.org

38 lines, 19 significant
1#!/usr/bin/env python2
2
3"""Print names of all methods defined in module
4
5This script demonstrates use of the visitor interface of the compiler
6package.
7"""
8
9import compiler
10
11class MethodFinder:
12 """Print the names of all the methods
13
14 Each visit method takes two arguments, the node and its current
15 scope. The scope is the name of the current class or None.
16 """
17
18 def visitClass(self, node, scope=None):
19 self.visit(node.code, node.name)
20
21 def visitFunction(self, node, scope=None):
22 if scope is not None:
23 print("%s.%s" % (scope, node.name))
24 self.visit(node.code, None)
25
26def main(files):
27 mf = MethodFinder()
28 for file in files:
29 f = open(file)
30 buf = f.read()
31 f.close()
32 ast = compiler.parse(buf)
33 compiler.walk(ast, mf)
34
35if __name__ == "__main__":
36 import sys
37
38 main(sys.argv[1:])