#!/usr/bin/env python
""" Trivial helper program to bytecompile and run an uncompile
"""
import os, sys, py_compile
assert len(sys.argv) >= 2
version = sys.version[0:3]
if sys.argv[1] == '--run':
    suffix = '_run'
    py_source = sys.argv[2:]
else:
    suffix = ''
    py_source = sys.argv[1:]

for path in py_source:
    short = os.path.basename(path)
    if hasattr(sys, 'pypy_version_info'):
        cfile =  "bytecode_pypy%s%s/%s" % (version, suffix, short) + 'c'
    else:
        cfile =  "bytecode_%s%s/%s" % (version, suffix, short) + 'c'
    print("byte-compiling %s to %s" % (path, cfile))
    py_compile.compile(path, cfile)
    if isinstance(version, str) or version >= (2, 6, 0):
        os.system("../bin/uncompyle6 -a -t %s" % cfile)