Added another default.

README.md

@@ -62,7 +62,7 @@ In addition, this metric supports three additional arguments, specifying which i
 
 As the new arguments are optional, this could be used as a drop-in replacement for `code_eval`.
 
-Additionally, this metric sets several different `globals` if they are not provided as additional globals. The full list of globals set is: `__metaclass__, __name__, _getiter_, _iter_unpack_sequence_, _getitem_, getattr, _write_, _inplacevar_`. See the code for additional details.
+Additionally, this metric sets several different `globals` if they are not provided as additional globals. The full list of globals set is: `__metaclass__, __name__, _getiter_, _iter_unpack_sequence_, _getitem_, getattr, _write_, _inplacevar_, _print_`. See the code for additional details.
 
 ### Output Values
 

restrictedpython_code_eval.py

@@ -286,6 +286,31 @@ def _default_write_(obj):
     return obj
 
 
+class DefaultPrinter:
+    def __init__(self, _getattr_=None, *args, **kwargs):
+        self._getattr_ = _getattr_
+        self.txt = []
+
+        self.args = args
+        self.kwargs = kwargs
+
+    def write(self, text):
+        self.txt.append(text)
+        print(text)
+
+    def __call__(self):
+        return ''.join(self.txt)
+
+    def _call_print(self, *objects, **kwargs):
+        if kwargs.get('file', None) is None:
+            kwargs['file'] = self
+        else:
+            self._getattr_(kwargs['file'], 'write')  # type: ignore
+
+        print(*objects, **kwargs)
+
+
+
 def _unsafe_execute(check_program, result, timeout, 
                     use_safe_builtins: bool = True, use_limited_builtins: bool = True, use_utility_builtins: bool = True,
                     additional_globals: Optional[Dict[str, Any]] = None, additional_locals: Optional[Dict[str, Any]] = None,
@@ -347,6 +372,9 @@ def _unsafe_execute(check_program, result, timeout,
             if '_inplacevar_' not in exec_globals:
                 exec_globals['_inplacevar_'] = protected_inplacevar  # type: ignore
 
+            if '_print_' not in exec_globals:
+                exec_globals['_print_'] = DefaultPrinter  # type: ignore
+
             with swallow_io():
                 with time_limit(timeout):
                     byte_code = compile_restricted(check_program, filename="<model output>", mode="exec")