python-uncompyle6/test/add-test.py

#!/usr/bin/env python
""" Trivial helper program to bytecompile and run an uncompile
"""
import os, sys, py_compile
from xdis.version_info import version_tuple_to_str, PYTHON_VERSION_TRIPLE

assert (2 <= len(sys.argv) <= 4)
version = sys.version[0:3]
vers = sys.version_info[:2]
if sys.argv[1] in ("--run", "-r"):
    suffix = "_run"
    py_source = sys.argv[2:]
    i = 2
else:
    suffix = ""
    py_source = sys.argv[1:]
    i = 1
try:
    optimize = int(sys.argv[-1])
    py_source = sys.argv[i:-1]
except:
    optimize = 2

for path in py_source:
    short = os.path.basename(path)
    if hasattr(sys, "pypy_version_info"):
        version = version_tuple_to_str(end=2, delimiter="")
        bytecode = "bytecode_pypy%s%s/%spy%s.pyc" % (version, suffix, short, version)
    else:
        version = version_tuple_to_str(end=2)
        bytecode = "bytecode_%s%s/%s.pyc" % (version, suffix, short)

    print("byte-compiling %s to %s" % (path, bytecode))
    py_compile.compile(path, bytecode, optimize=optimize)
    if vers > (3, 1):
        py_compile.compile(path, bytecode, optimize=optimize)
    else:
        py_compile.compile(path, bytecode)
    if PYTHON_VERSION_TRIPLE >= (2, 6):
        os.system("../bin/uncompyle6 -a -t %s" % bytecode)