better method of getting clr object

mcneel · May 29, 2024 · 69ec514 · 69ec514
1 parent 6e5dfbb
commit 69ec514
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/runtime/McNeel.PythonEngine.cs b/src/runtime/McNeel.PythonEngine.cs
@@ -273,10 +273,11 @@ public void SetStdIO(Stream stdout)
                 using var sys = Runtime.PyImport_ImportModule("sys");
                 using (PyObject sysObj = sys.MoveToPyObject())
                 {
-                    if (sysObj.GetAttr("stdout")
-                              .AsManagedObject(typeof(RhinoCodePythonEngineIO)) is RhinoCodePythonEngineIO currentStdout)
+                    if (sysObj.GetAttr("stdout") is PyObject stdoutObj
+                            && ManagedType.GetManagedObject(stdoutObj) is CLRObject clrObj
+                            && clrObj.inst is RhinoCodePythonEngineIO exstStdout)
                     {
-                        currentStdout.Dispose();
+                        exstStdout.Dispose();
                     }
 
                     using var stdio = PyObject.FromManagedObject(new RhinoCodePythonEngineIO(stdout));