Jeremy Hylton | b769e80 | 2000-10-09 14:35:24 +0000 | [diff] [blame] | 1 | #! /usr/bin/env python |
| 2 | |
| 3 | import marshal |
Jeremy Hylton | ce0c19c | 2001-09-17 18:08:20 +0000 | [diff] [blame] | 4 | import os |
Jeremy Hylton | b769e80 | 2000-10-09 14:35:24 +0000 | [diff] [blame] | 5 | import dis |
| 6 | import types |
| 7 | |
| 8 | def dump(obj): |
| 9 | print obj |
| 10 | for attr in dir(obj): |
Jeremy Hylton | ce0c19c | 2001-09-17 18:08:20 +0000 | [diff] [blame] | 11 | if attr.startswith('co_'): |
| 12 | val = getattr(obj, attr) |
| 13 | print "\t", attr, repr(val) |
Jeremy Hylton | b769e80 | 2000-10-09 14:35:24 +0000 | [diff] [blame] | 14 | |
| 15 | def loadCode(path): |
| 16 | f = open(path) |
| 17 | f.read(8) |
| 18 | co = marshal.load(f) |
| 19 | f.close() |
| 20 | return co |
| 21 | |
| 22 | def walk(co, match=None): |
| 23 | if match is None or co.co_name == match: |
| 24 | dump(co) |
| 25 | print |
| 26 | dis.dis(co) |
| 27 | for obj in co.co_consts: |
| 28 | if type(obj) == types.CodeType: |
| 29 | walk(obj, match) |
| 30 | |
| 31 | def main(filename, codename=None): |
| 32 | co = loadCode(filename) |
| 33 | walk(co, codename) |
| 34 | |
| 35 | if __name__ == "__main__": |
| 36 | import sys |
| 37 | if len(sys.argv) == 3: |
| 38 | filename, codename = sys.argv[1:] |
| 39 | else: |
| 40 | filename = sys.argv[1] |
| 41 | codename = None |
Jeremy Hylton | ce0c19c | 2001-09-17 18:08:20 +0000 | [diff] [blame] | 42 | if filename.endswith('.py') and os.path.exists(filename+"c"): |
| 43 | filename += "c" |
Jeremy Hylton | b769e80 | 2000-10-09 14:35:24 +0000 | [diff] [blame] | 44 | main(filename, codename) |