42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from rich.cells import cell_len
|
||
|
|
from rich.console import Console, RenderableType
|
||
|
|
from rich.protocol import rich_cast
|
||
|
|
|
||
|
|
|
||
|
|
def measure(
|
||
|
|
console: Console,
|
||
|
|
renderable: RenderableType,
|
||
|
|
default: int,
|
||
|
|
*,
|
||
|
|
container_width: int | None = None,
|
||
|
|
) -> int:
|
||
|
|
"""Measure a rich renderable.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
console: A console object.
|
||
|
|
renderable: Rich renderable.
|
||
|
|
default: Default width to use if renderable does not expose dimensions.
|
||
|
|
container_width: Width of container or None to use console width.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
Width in cells
|
||
|
|
"""
|
||
|
|
if isinstance(renderable, str):
|
||
|
|
return cell_len(renderable)
|
||
|
|
|
||
|
|
width = default
|
||
|
|
renderable = rich_cast(renderable)
|
||
|
|
get_console_width = getattr(renderable, "__rich_measure__", None)
|
||
|
|
if get_console_width is not None:
|
||
|
|
options = (
|
||
|
|
console.options
|
||
|
|
if container_width is None
|
||
|
|
else console.options.update_width(container_width)
|
||
|
|
)
|
||
|
|
render_width = get_console_width(console, options).maximum
|
||
|
|
width = max(0, render_width)
|
||
|
|
|
||
|
|
return width
|