forked from varia/varia.website
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
|
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
|