[Python-checkins] cpython: Modernize Python/makeopcodetargets.py

[victor.stinner]

https://hg.python.org/cpython/rev/f682b620c64d
changeset:   100743:f682b620c64d
user:        Victor Stinner <victor.stinner at gmail.com>
date:        Fri Mar 25 11:54:47 2016 +0100
summary:
  Modernize Python/makeopcodetargets.py

* Simply use "import opcode" to import the opcode module instead of tricks
  using the imp module
* Use context manager for the output file
* Move code into a new main() function
* Replace assert with a regular if to check the number of arguments
* Import modules at top level

files:
  Python/makeopcodetargets.py |  32 +++++++++---------------
  1 files changed, 12 insertions(+), 20 deletions(-)


diff --git a/Python/makeopcodetargets.py b/Python/makeopcodetargets.py
--- a/Python/makeopcodetargets.py
+++ b/Python/makeopcodetargets.py
@@ -3,24 +3,14 @@
 (for compilers supporting computed gotos or "labels-as-values", such as gcc).
 """
 
-# This code should stay compatible with Python 2.3, at least while
-# some of the buildbots have Python 2.3 as their system Python.
+import opcode
+import os
+import sys
 
-import imp
-import os
-
-
-def find_module(modname):
-    """Finds and returns a module in the local dist/checkout.
-    """
-    modpath = os.path.join(
-        os.path.dirname(os.path.dirname(__file__)), "Lib")
-    return imp.load_module(modname, *imp.find_module(modname, [modpath]))
 
 def write_contents(f):
     """Write C code contents to the target file object.
     """
-    opcode = find_module("opcode")
     targets = ['_unknown_opcode'] * 256
     for opname, op in opcode.opmap.items():
         targets[op] = "TARGET_%s" % opname
@@ -29,15 +19,17 @@
     f.write("\n};\n")
 
 
-if __name__ == "__main__":
-    import sys
-    assert len(sys.argv) < 3, "Too many arguments"
+def main():
+    if len(sys.argv) >= 3:
+        sys.exit("Too many arguments")
     if len(sys.argv) == 2:
         target = sys.argv[1]
     else:
         target = "Python/opcode_targets.h"
-    f = open(target, "w")
-    try:
+    with open(target, "w") as f:
         write_contents(f)
-    finally:
-        f.close()
+    print("Jump table written into %s" % target)
+
+
+if __name__ == "__main__":
+    main()

-- 
Repository URL: https://hg.python.org/cpython