| 1 | import compiler
|
| 2 | import dis
|
| 3 | import types
|
| 4 |
|
| 5 | def extract_code_objects(co):
|
| 6 | l = [co]
|
| 7 | for const in co.co_consts:
|
| 8 | if type(const) == types.CodeType:
|
| 9 | l.append(const)
|
| 10 | return l
|
| 11 |
|
| 12 | def compare(a, b):
|
| 13 | if not (a.co_name == "?" or a.co_name.startswith('<lambda')):
|
| 14 | assert a.co_name == b.co_name, (a, b)
|
| 15 | if a.co_stacksize != b.co_stacksize:
|
| 16 | print("stack mismatch %s: %d vs. %d" % (a.co_name,
|
| 17 | a.co_stacksize,
|
| 18 | b.co_stacksize))
|
| 19 | if a.co_stacksize > b.co_stacksize:
|
| 20 | print("good code")
|
| 21 | dis.dis(a)
|
| 22 | print("bad code")
|
| 23 | dis.dis(b)
|
| 24 | assert 0
|
| 25 |
|
| 26 | def main(files):
|
| 27 | for file in files:
|
| 28 | print(file)
|
| 29 | buf = open(file).read()
|
| 30 | try:
|
| 31 | co1 = compile(buf, file, "exec")
|
| 32 | except SyntaxError:
|
| 33 | print("skipped")
|
| 34 | continue
|
| 35 | co2 = compiler.compile(buf, file, "exec")
|
| 36 | co1l = extract_code_objects(co1)
|
| 37 | co2l = extract_code_objects(co2)
|
| 38 | for a, b in zip(co1l, co2l):
|
| 39 | compare(a, b)
|
| 40 |
|
| 41 | if __name__ == "__main__":
|
| 42 | import sys
|
| 43 | main(sys.argv[1:])
|