1.1 --- a/test.py Sat Dec 02 00:14:33 2006 +0100
1.2 +++ b/test.py Sat Dec 02 00:17:54 2006 +0100
1.3 @@ -6,21 +6,7 @@
1.4 import viewer
1.5 from annotate import AnnotationError, annotate_all as aa
1.6
1.7 -builtins = simplify.simplify(os.path.join("lib", "builtins.py"), 1)
1.8 -module = simplify.simplify(sys.argv[1])
1.9 -fixnames.fix(builtins)
1.10 -fixnames.fix(module, builtins)
1.11 -
1.12 -stream = sys.stdout
1.13 -v = viewer.Viewer(stream)
1.14 -
1.15 -if "-a" in sys.argv:
1.16 - try:
1.17 - aa([module], builtins)
1.18 - except AnnotationError, exc:
1.19 - v.report(exc)
1.20 -
1.21 -if "-d" in sys.argv:
1.22 +def makedoc(viewer, module, builtins):
1.23 try:
1.24 dirname = "%s-docs" % module.name
1.25 if not os.path.exists(dirname):
1.26 @@ -30,4 +16,23 @@
1.27 except viewer.ViewerError, exc:
1.28 raise
1.29
1.30 +if __name__ == "__main__":
1.31 +
1.32 + builtins = simplify.simplify(os.path.join("lib", "builtins.py"), 1)
1.33 + module = simplify.simplify(sys.argv[1])
1.34 + fixnames.fix(builtins)
1.35 + fixnames.fix(module, builtins)
1.36 +
1.37 + stream = sys.stdout
1.38 + v = viewer.Viewer(stream)
1.39 +
1.40 + if "-a" in sys.argv:
1.41 + try:
1.42 + aa([module], builtins)
1.43 + except AnnotationError, exc:
1.44 + v.report(exc)
1.45 +
1.46 + if "-d" in sys.argv:
1.47 + makedoc(viewer, module, builtins)
1.48 +
1.49 # vim: tabstop=4 expandtab shiftwidth=4