Use Python 3.7 generator when running on Python 3.7

DinoV · DinoV · commit 3f7be6bba681 · 2021-01-28T11:14:38.000-08:00
Summary: This is the start of actually being able to really use the Python 3.7 compiler - we'll now switch the default compilation mode based upon the version in sys.version_info.  We also now need to define the SETUP_ANNOTATIONS opcode to be able to import the compiler on 3.7 similar to how we define LOAD_METHOD/CALL_METHOD on 3.6

Test Plan: ./python -m test.test_compiler
diff --git a/compiler/pycodegen.py b/compiler/pycodegen.py
@@ -59,7 +59,7 @@ def compile(source, filename, mode, flags=None, dont_inherit=None):
     if flags is not None or dont_inherit is not None:
         raise RuntimeError("not implemented yet")
 
-    return make_compiler(source, filename, mode, CodeGenerator).getCode()
+    return make_compiler(source, filename, mode, get_default_generator()).getCode()
 
 
 def make_compiler(source, filename, mode, generator=None):
@@ -73,7 +73,7 @@ def make_compiler(source, filename, mode, generator=None):
         tree = ast.parse(source, filename, mode)
 
     if generator is None:
-        generator = CodeGenerator
+        generator = get_default_generator()
 
     return generator.make_code_gen("<module>", tree, filename)
 
@@ -1975,7 +1975,11 @@ def _visitAnnotation(self, node):
         else:
             self.visit(node)
 
+def get_default_generator():
+    if sys.version_info >= (3, 7):
+        return Python37CodeGenerator
 
+    return CodeGenerator
 
 def get_docstring(node):
     if node.body and isinstance(node.body[0], ast.Expr) \
