Reintroduce Python2 support in generate_opcode_h.py

vstinner · vstinner · commit 213cc388c79c · 2016-11-28T18:13:52.000+01:00
Issue #28821. Add also a message to show that the command did something :-)
diff --git a/Tools/scripts/generate_opcode_h.py b/Tools/scripts/generate_opcode_h.py
@@ -1,6 +1,8 @@
 # This script generates the opcode.h header file.
 
 import sys
+import tokenize
+
 header = """/* Auto-generated by Tools/scripts/generate_opcode_h.py */
 #ifndef Py_OPCODE_H
 #define Py_OPCODE_H
@@ -32,12 +34,14 @@
 #endif /* !Py_OPCODE_H */
 """
 
-import tokenize
-
 
 def main(opcode_py, outfile='Include/opcode.h'):
     opcode = {}
-    with tokenize.open(opcode_py) as fp:
+    if hasattr(tokenize, 'open'):
+        fp = tokenize.open(opcode_py)   # Python 3.2+
+    else:
+        fp = open(opcode_py)            # Python 2.7
+    with fp:
         code = fp.read()
     exec(code, opcode)
     opmap = opcode['opmap']
@@ -51,6 +55,8 @@ def main(opcode_py, outfile='Include/opcode.h'):
                             ('HAVE_ARGUMENT', opcode['HAVE_ARGUMENT']))
         fobj.write(footer)
 
+    print("%s regenerated from %s" % (outfile, opcode_py))
+
 
 if __name__ == '__main__':
     main(sys.argv[1], sys.argv[2])
