set __file__, __doc__ and __class__ on ModuleObject to make the managed

tonyroberts · tonyroberts · commit f0d0aeb81030 · 2014-09-15T13:07:06.000+01:00
modules behave better in ipython (%autoreload expects __file__ to be set
and the ipython help ? expects __class__ to be set).
diff --git a/pythonnet/src/runtime/assemblymanager.cs b/pythonnet/src/runtime/assemblymanager.cs
@@ -314,6 +314,16 @@ public static bool IsValidNamespace(string name) {
             return namespaces.ContainsKey(name);
         }
 
+        //===================================================================
+        // Returns list of assemblies that declare types in a given namespace
+        //===================================================================
+
+        public static IEnumerable<Assembly> GetAssemblies(string nsname) {
+            if (!namespaces.ContainsKey(nsname))
+                return new List<Assembly>();
+
+            return namespaces[nsname].Keys;
+        }
 
         //===================================================================
         // Returns the current list of valid names for the input namespace.
diff --git a/pythonnet/src/runtime/moduleobject.cs b/pythonnet/src/runtime/moduleobject.cs
@@ -37,12 +37,27 @@ public ModuleObject(string name) : base() {
             cache = new Dictionary<string, ManagedType>();
             _namespace = name;
 
+            // Use the filename from any of the assemblies just so there's something for
+            // anything that expects __file__ to be set.
+            string filename = "unknown";
+            string docstring = "Namespace containing types from the following assemblies:\n\n";
+            foreach (Assembly a in AssemblyManager.GetAssemblies(name)) {
+                filename = a.Location;
+                docstring += "- " + a.FullName + "\n";
+            }
+
             dict = Runtime.PyDict_New();
             IntPtr pyname = Runtime.PyString_FromString(moduleName);
+            IntPtr pyfilename = Runtime.PyString_FromString(filename);
+            IntPtr pydocstring = Runtime.PyString_FromString(docstring);
+            IntPtr pycls = TypeManager.GetTypeHandle(this.GetType());
             Runtime.PyDict_SetItemString(dict, "__name__", pyname);
-            Runtime.PyDict_SetItemString(dict, "__file__", Runtime.PyNone);
-            Runtime.PyDict_SetItemString(dict, "__doc__", Runtime.PyNone);
+            Runtime.PyDict_SetItemString(dict, "__file__", pyfilename);
+            Runtime.PyDict_SetItemString(dict, "__doc__", pydocstring);
+            Runtime.PyDict_SetItemString(dict, "__class__", pycls);
             Runtime.Decref(pyname);
+            Runtime.Decref(pyfilename);
+            Runtime.Decref(pydocstring);
 
             Marshal.WriteIntPtr(this.pyHandle, ObjectOffset.DictOffset(this.pyHandle), dict);
 
