55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
import io
|
|
from typing import IO, TYPE_CHECKING, Any, List
|
|
|
|
from .ansi import AnsiDecoder
|
|
from .text import Text
|
|
|
|
if TYPE_CHECKING:
|
|
from .console import Console
|
|
|
|
|
|
class FileProxy(io.TextIOBase):
|
|
"""Wraps a file (e.g. sys.stdout) and redirects writes to a console."""
|
|
|
|
def __init__(self, console: "Console", file: IO[str]) -> None:
|
|
self.__console = console
|
|
self.__file = file
|
|
self.__buffer: List[str] = []
|
|
self.__ansi_decoder = AnsiDecoder()
|
|
|
|
@property
|
|
def rich_proxied_file(self) -> IO[str]:
|
|
"""Get proxied file."""
|
|
return self.__file
|
|
|
|
def __getattr__(self, name: str) -> Any:
|
|
return getattr(self.__file, name)
|
|
|
|
def write(self, text: str) -> int:
|
|
if not isinstance(text, str):
|
|
raise TypeError(f"write() argument must be str, not {type(text).__name__}")
|
|
buffer = self.__buffer
|
|
lines: List[str] = []
|
|
while text:
|
|
line, new_line, text = text.partition("\n")
|
|
if new_line:
|
|
lines.append("".join(buffer) + line)
|
|
del buffer[:]
|
|
else:
|
|
buffer.append(line)
|
|
break
|
|
if lines:
|
|
console = self.__console
|
|
with console:
|
|
output = Text("\n").join(
|
|
self.__ansi_decoder.decode_line(line) for line in lines
|
|
)
|
|
console.print(output)
|
|
return len(text)
|
|
|
|
def flush(self) -> None:
|
|
output = "".join(self.__buffer)
|
|
if output:
|
|
self.__console.print(output)
|
|
del self.__buffer[:]
|