Get filenos of original stdout and stderr in console

Textualize · Mar 16, 2022 · 98969ab · 98969ab
1 parent 1d61867
commit 98969ab
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/rich/console.py b/rich/console.py
@@ -2,7 +2,6 @@
 import io
 import os
 import platform
-import pty
 import sys
 import threading
 from abc import ABC, abstractmethod
@@ -83,6 +82,15 @@ class NoChange:
 
 NO_CHANGE = NoChange()
 
+try:
+    _STDOUT_FILENO = sys.__stdout__.fileno()
+except Exception:
+    _STDOUT_FILENO = 1
+
+try:
+    _STDERR_FILENO = sys.__stderr__.fileno()
+except Exception:
+    _STDERR_FILENO = 2
 
 CONSOLE_HTML_FORMAT = """\
 <!DOCTYPE html>
@@ -1937,8 +1945,8 @@ def _check_buffer(self) -> None:
                         except (ValueError, io.UnsupportedOperation):
                             file_no = -1
 
-                        stdout_num = pty.STDOUT_FILENO
-                        stderr_num = pty.STDERR_FILENO
+                        stdout_num = _STDOUT_FILENO
+                        stderr_num = _STDERR_FILENO
                         is_std_stream = file_no in (stdout_num, stderr_num)
                         legacy_windows_std = self.legacy_windows and is_std_stream
                         if legacy_windows_std: