introduce test theme

This commit is contained in:
pedrocarlo 2025-04-04 00:56:50 -03:00
parent 321def3c30
commit 3cd2017df4
2 changed files with 36 additions and 3 deletions

View file

@ -4,7 +4,14 @@ from rich.theme import Theme
from rich.style import Style
custom_theme = Theme({"info": "bold blue", "error": "bold red", "debug": "bold blue"})
custom_theme = Theme(
{
"info": "bold blue",
"error": "bold red",
"debug": "bold blue",
"test": "bold green",
}
)
console = Console(theme=custom_theme, force_terminal=True)
@ -87,3 +94,29 @@ def debug(
log_locals=log_locals,
_stack_offset=_stack_offset + 1,
)
def test(
*objects: Any,
sep: str = " ",
end: str = "\n",
style: Optional[Union[str, Style]] = None,
justify: Optional[JustifyMethod] = None,
emoji: Optional[bool] = None,
markup: Optional[bool] = None,
highlight: Optional[bool] = None,
log_locals: bool = False,
_stack_offset: int = 1,
):
console.log(
"[test]TEST[/test]",
*objects,
sep=sep,
end=end,
style=style,
justify=justify,
emoji=emoji,
markup=markup,
highlight=highlight,
log_locals=log_locals,
_stack_offset=_stack_offset + 1,
)