3.8 SETUP_EXCEPT removal workaround; reinstate option -c | --compile

uncompyle6/main.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2018 Rocky Bernstein <rocky@gnu.org>
+# Copyright (C) 2018-2019 Rocky Bernstein <rocky@gnu.org>
 #
 #  This program is free software: you can redistribute it and/or modify
 #  it under the terms of the GNU General Public License as published by
@@ -13,7 +13,7 @@
 #  You should have received a copy of the GNU General Public License
 #  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 from __future__ import print_function
-import datetime, os, subprocess, sys, tempfile
+import datetime, py_compile, os, subprocess, sys, tempfile
 
 from uncompyle6 import verify, IS_PYPY, PYTHON_VERSION
 from xdis.code import iscode
@@ -119,6 +119,22 @@ def decompile(
         # deparsing failed
         raise pysource.SourceWalkerError(str(e))
 
+def compile_file(source_path):
+    if source_path.endswith('.py'):
+        basename = source_path[:-3]
+    else:
+        basename = source_path
+
+    if hasattr(sys, 'pypy_version_info'):
+        bytecode_path = "%s-pypy%s.pyc" % (basename, PYTHON_VERSION)
+    else:
+        bytecode_path = "%s-%s.pyc" % (basename, PYTHON_VERSION)
+
+    print("compiling %s to %s" % (source_path, bytecode_path))
+    py_compile.compile(source_path, bytecode_path, 'exec')
+    return bytecode_path
+
+
 def decompile_file(filename, outstream=None, showasm=None, showast=False,
                    showgrammar=False, mapstream=None, do_fragments=False):
     """
@@ -150,7 +166,7 @@ def decompile_file(filename, outstream=None, showasm=None, showast=False,
 
 
 # FIXME: combine into an options parameter
-def main(in_base, out_base, files, codes, outfile=None,
+def main(in_base, out_base, compiled_files, source_files, outfile=None,
          showasm=None, showast=False, do_verify=False,
          showgrammar=False, raise_on_error=False,
          do_linemaps=False, do_fragments=False):
@@ -160,8 +176,6 @@ def main(in_base, out_base, files, codes, outfile=None,
     files	list of filenames to be uncompyled (relative to in_base)
     outfile	write output to this filename (overwrites out_base)
 
-    Note: `codes` is not use. Historical compatability?
-
     For redirecting output to
     - <filename>		outfile=<filename> (out_base is ignored)
     - files below out_base	out_base=...
@@ -171,7 +185,10 @@ def main(in_base, out_base, files, codes, outfile=None,
     current_outfile = outfile
     linemap_stream = None
 
-    for filename in files:
+    for source_path in source_files:
+        compiled_files.append(compile_file(source_path))
+
+    for filename in compiled_files:
         infile = os.path.join(in_base, filename)
         # print("XXX", infile)
         if not os.path.exists(infile):