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