Import all assemblies for Unity

All assemblies in the `UnityEngine` and `UnityEditor` namespaces are now
imported automatically when using `UnityPython.CreateEngine`.

This fixes problems where not all types were available to import inside
of a Python script, such as Rigidbody.

Author: exodrifter

CHANGELOG.md

@@ -6,6 +6,10 @@ and this project adheres to [Semantic Versioning](http://semver.org/).
 
 ## [Unreleased]
 
+### Fixed
+* Fix problem where not all types in the `UnityEngine` namespace could be
+  imported
+
 
 ## [0.3.1]
 

Scripts/UnityPython.cs

@@ -1,34 +1,53 @@
 using IronPython.Hosting;
 using Microsoft.Scripting.Hosting;
+using System;
 using System.Collections.Generic;
+using System.Linq;
+using System.Reflection;
 
-public sealed class UnityPython
+/// <summary>
+/// Convenience class for creating a Python engine integrated with Unity.
+///
+/// All scripts executed by the ScriptEngine created by this class will:
+/// * 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
+/// </summary>
+public static class UnityPython
 {
-	public static ScriptEngine CreateEngine()
+	public static ScriptEngine CreateEngine(IDictionary<string, object> options = null)
 	{
-		return CreateEngine(null);
-	}
+		var engine = Python.CreateEngine(options);
 
-	public static ScriptEngine CreateEngine(IDictionary<string, object> options)
-	{
-		ScriptEngine engine;
-		if (options == null)
+		// Load assemblies for the `UnityEngine*` namespaces
+		foreach (var assembly in GetAssembliesInNamespace("UnityEngine"))
 		{
-			engine = Python.CreateEngine();
+			engine.Runtime.LoadAssembly(assembly);
 		}
-		else
-		{
-			engine = Python.CreateEngine(options);
-		}
-
-		var unityEngine = typeof(UnityEngine.GameObject).Assembly;
-		engine.Runtime.LoadAssembly(unityEngine);
 
 #if UNITY_EDITOR
-		var unityEditor = typeof(UnityEditor.Editor).Assembly;
-		engine.Runtime.LoadAssembly(unityEditor);
+		// Load assemblies for the `UnityEditor*` namespaces
+		foreach (var assembly in GetAssembliesInNamespace("UnityEditor"))
+		{
+			engine.Runtime.LoadAssembly(assembly);
+		}
 #endif
 
 		return engine;
 	}
+
+	/// <summary>
+	/// Get a list of all loaded assemblies in the current AppDomain for a
+	/// namespace beginning with the specified string.
+	/// </summary>
+	/// <param name="prefix">The beginning of the namespace.</param>
+	/// <returns>All matching assemblies.</returns>
+	private static IEnumerable<Assembly> GetAssembliesInNamespace(string prefix)
+	{
+		return AppDomain.CurrentDomain.GetAssemblies()
+			.SelectMany(t => t.GetTypes())
+			.Where(t => t.Namespace != null && t.Namespace.StartsWith(prefix))
+			.Select(t => t.Assembly)
+			.Distinct();
+	}
 }