Set stdio to None if feature is disabled

danielstuart14 · danielstuart14 · commit e8d847459c5b · 2024-10-13T23:30:03.000-03:00
More in line with what CPython does
diff --git a/vm/src/vm/mod.rs b/vm/src/vm/mod.rs
@@ -19,8 +19,8 @@ use crate::{
         code::PyCode,
         pystr::AsPyStr,
         tuple::{PyTuple, PyTupleTyped},
-        PyBaseExceptionRef, PyDictRef, PyInt, PyList, PyModule, PyStr, PyStrInterned, PyStrRef,
-        PyTypeRef,
+        PyBaseExceptionRef, PyDictRef, PyInt, PyList, PyModule, PyNone, PyStr, PyStrInterned,
+        PyStrRef, PyTypeRef,
     },
     codecs::CodecsRegistry,
     common::{hash::HashSecret, lock::PyMutex, rc::PyRc},
@@ -329,6 +329,16 @@ impl VirtualMachine {
                     set_stdio("stderr", 2, "w")?;
                 }
 
+                #[cfg(not(feature = "stdio"))]
+                {
+                    self.sys_module
+                        .set_attr("stdin", PyNone.into_pyobject(self), self)?;
+                    self.sys_module
+                        .set_attr("stdout", PyNone.into_pyobject(self), self)?;
+                    self.sys_module
+                        .set_attr("stderr", PyNone.into_pyobject(self), self)?;
+                }
+
                 let io_open = io.get_attr("open", self)?;
                 self.builtins.set_attr("open", io_open, self)?;
             }
