modify-capturing-tee

Author: Darshan808

IPython/core/interactiveshell.py

@@ -4026,10 +4026,8 @@ def __init__(self, shell: InteractiveShell, channel="stdout"):
 
         Parameters
         ----------
-        outputs : dict
-            Dictionary to store captured outputs, with execution count as key.
-        execution_count : int
-            The current cell execution number.
+        shell : InteractiveShell
+            The shell instance
         channel : str, optional
             Output channel to capture (default: 'stdout').
         """
@@ -4039,6 +4037,9 @@ def __init__(self, shell: InteractiveShell, channel="stdout"):
         setattr(sys, channel, self)  # Redirect stdout to this instance
         self._closed = False
 
+        # Store original methods and attributes for delegation
+        self._original_attrs = dir(self.ostream)
+
     def write(self, data):
         """Write data to both the original stdout and the capture dictionary."""
         self.ostream.write(data)  # Display in notebook
@@ -4077,6 +4078,14 @@ def close(self):
         setattr(sys, self.channel, self.ostream)
         self._closed = True
 
+    def __getattr__(self, name):
+        """Delegate any other attribute access to the original stream."""
+        if name in self._original_attrs:
+            return getattr(self.ostream, name)
+        raise AttributeError(
+            f"'{self.__class__.__name__}' object has no attribute '{name}'"
+        )
+
 
 class InteractiveShellABC(metaclass=abc.ABCMeta):
     """An abstract base class for InteractiveShell."""