Redirect IronPython output to Unity console

Author: exodrifter

CHANGELOG.md

@@ -6,6 +6,9 @@ and this project adheres to [Semantic Versioning](http://semver.org/).
 
 ## [Unreleased]
 
+### Added
+* IronPython output is now redirected to the Unity console
+
 ### Fixed
 * Fix problem where not all types in the `UnityEngine` namespace could be
   imported

Scripts/DebugLogWriter.cs

@@ -0,0 +1,39 @@
+using System;
+using System.IO;
+
+/// <summary>
+/// Used to redirect IronPython output to the Unity debug console.
+/// </summary>
+public class UnityLogWriter : StreamWriter
+{
+	private readonly Action<string> logger;
+	private string buffer = "";
+
+	public UnityLogWriter(Action<string> logger, Stream s) : base(s)
+	{
+		if (logger == null)
+		{
+			throw new ArgumentNullException("logger");
+		}
+		this.logger = logger;
+	}
+
+	public override void Write(string value)
+	{
+		base.Write(value);
+		buffer += value;
+
+		var lines = buffer.Split('\n');
+		for (var i = 0; i < lines.Length; ++i)
+		{
+			if (i == lines.Length - 1)
+			{
+				buffer = lines[i];
+			}
+			else
+			{
+				logger(lines[i]);
+			}
+		}
+	}
+}

Scripts/DebugLogWriter.cs.meta

@@ -2,13 +2,16 @@
 using Microsoft.Scripting.Hosting;
 using System;
 using System.Collections.Generic;
+using System.IO;
 using System.Linq;
 using System.Reflection;
+using UnityEngine;
 
 /// <summary>
 /// Convenience class for creating a Python engine integrated with Unity.
 ///
 /// All scripts executed by the ScriptEngine created by this class will:
+/// * Redirect output to Unity's console
 /// * Be able to import any class in a `UnityEngine*` namespace
 /// * Be able to import any class in a `UnityEditor*` namespace, if the script
 ///   is running in the UnityEditor
@@ -19,6 +22,16 @@ public static ScriptEngine CreateEngine(IDictionary<string, object> options = nu
 	{
 		var engine = Python.CreateEngine(options);
 
+		// Redirect IronPython IO
+		var infoStream = new MemoryStream();
+		var infoWriter = new UnityLogWriter(Debug.Log, infoStream);
+		engine.Runtime.IO.SetOutput(infoStream, infoWriter);
+
+		var errorStream = new MemoryStream();
+		var errorWriter = new UnityLogWriter(Debug.Log, errorStream);
+		engine.Runtime.IO.SetErrorOutput(errorStream, errorWriter);
+
+
 		// Load assemblies for the `UnityEngine*` namespaces
 		foreach (var assembly in GetAssembliesInNamespace("UnityEngine"))
 		{