| from typing import Optional, TYPE_CHECKING | |
| from .segment import Segment | |
| from .style import StyleType | |
| from ._loop import loop_last | |
| if TYPE_CHECKING: | |
| from .console import ( | |
| Console, | |
| ConsoleOptions, | |
| RenderResult, | |
| RenderableType, | |
| Group, | |
| ) | |
| class Screen: | |
| """A renderable that fills the terminal screen and crops excess. | |
| Args: | |
| renderable (RenderableType): Child renderable. | |
| style (StyleType, optional): Optional background style. Defaults to None. | |
| """ | |
| renderable: "RenderableType" | |
| def __init__( | |
| self, | |
| *renderables: "RenderableType", | |
| style: Optional[StyleType] = None, | |
| application_mode: bool = False, | |
| ) -> None: | |
| from rich.console import Group | |
| self.renderable = Group(*renderables) | |
| self.style = style | |
| self.application_mode = application_mode | |
| def __rich_console__( | |
| self, console: "Console", options: "ConsoleOptions" | |
| ) -> "RenderResult": | |
| width, height = options.size | |
| style = console.get_style(self.style) if self.style else None | |
| render_options = options.update(width=width, height=height) | |
| lines = console.render_lines( | |
| self.renderable or "", render_options, style=style, pad=True | |
| ) | |
| lines = Segment.set_shape(lines, width, height, style=style) | |
| new_line = Segment("\n\r") if self.application_mode else Segment.line() | |
| for last, line in loop_last(lines): | |
| yield from line | |
| if not last: | |
| yield new_line | |