push string to attribute instead of printing

Author: William Grant

typed_python/compiler/native_ast_to_llvm.py

@@ -1498,6 +1498,7 @@ def __init__(self):
         self._inlineRequests = []
 
         self._printAllNativeCalls = os.getenv("TP_COMPILER_LOG_NATIVE_CALLS")
+        self._native_ast_str = ""
         self.verbose = False
 
     def markExternal(self, functionNameToType):
@@ -1575,30 +1576,28 @@ def add_functions(self, names_to_definitions):
             )
             func_type = llvmlite.ir.FunctionType(
                 type_to_llvm_type(function.output_type),
-                [type_to_llvm_type(x[1]) for x in function.args]
+                [type_to_llvm_type(x[1]) for x in function.args],
+            )
+            self._functions_by_name[name] = llvmlite.ir.Function(
+                module, func_type, name
             )
-            self._functions_by_name[name] = llvmlite.ir.Function(module, func_type, name)
 
-            self._functions_by_name[name].linkage = 'external'
+            self._functions_by_name[name].linkage = "external"
             self._function_definitions[name] = function
 
-        if self.verbose:
-            for name in names_to_definitions:
-                definition = names_to_definitions[name]
-                func = self._functions_by_name[name]
+            self._native_ast_str += "\n"
+            self._native_ast_str += "*************\n"
+            self._native_ast_str += "def %s(%s): #->%s\n" % (
+                name,
+                ",".join(["%s=%s" % (k, str(t)) for k, t in function.args]),
+                str(function.output_type),
+            )
+            self._native_ast_str += native_ast.indent(str(function.body.body)) + "\n"
+            self._native_ast_str += "*************\n"
+            self._native_ast_str += "\n"
 
-                print()
-                print("*************")
-                print(
-                    "def %s(%s): #->%s" % (
-                        name,
-                        ",".join(["%s=%s" % (k, str(t)) for k, t in definition.args]),
-                        str(definition.output_type)
-                    )
-                )
-                print(native_ast.indent(str(definition.body.body)))
-                print("*************")
-                print()
+        if self.verbose:
+            print(self._native_ast_str)
 
         globalDefinitions = {}
         globalDefinitionsLlvmValues = {}