mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 16:48:30 +00:00
0229390ad1
- Adds sep, end, file, and flush parameters to match Python's built-in print function signature.
91 lines
2.4 KiB
Python
91 lines
2.4 KiB
Python
"""Utility for colored console output."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING, Final, Literal, NamedTuple
|
|
|
|
if TYPE_CHECKING:
|
|
from _typeshed import SupportsWrite
|
|
|
|
PrinterColor = Literal[
|
|
"purple",
|
|
"bold_purple",
|
|
"green",
|
|
"bold_green",
|
|
"cyan",
|
|
"bold_cyan",
|
|
"magenta",
|
|
"bold_magenta",
|
|
"yellow",
|
|
"bold_yellow",
|
|
"red",
|
|
"blue",
|
|
"bold_blue",
|
|
]
|
|
|
|
_COLOR_CODES: Final[dict[PrinterColor, str]] = {
|
|
"purple": "\033[95m",
|
|
"bold_purple": "\033[1m\033[95m",
|
|
"red": "\033[91m",
|
|
"bold_green": "\033[1m\033[92m",
|
|
"green": "\033[32m",
|
|
"blue": "\033[94m",
|
|
"bold_blue": "\033[1m\033[94m",
|
|
"yellow": "\033[93m",
|
|
"bold_yellow": "\033[1m\033[93m",
|
|
"cyan": "\033[96m",
|
|
"bold_cyan": "\033[1m\033[96m",
|
|
"magenta": "\033[35m",
|
|
"bold_magenta": "\033[1m\033[35m",
|
|
}
|
|
|
|
RESET: Final[str] = "\033[0m"
|
|
|
|
|
|
class ColoredText(NamedTuple):
|
|
"""Represents text with an optional color for console output.
|
|
|
|
Attributes:
|
|
text: The text content to be printed.
|
|
color: Optional color for the text, specified as a PrinterColor.
|
|
"""
|
|
|
|
text: str
|
|
color: PrinterColor | None
|
|
|
|
|
|
class Printer:
|
|
"""Handles colored console output formatting."""
|
|
|
|
@staticmethod
|
|
def print(
|
|
content: str | list[ColoredText],
|
|
color: PrinterColor | None = None,
|
|
sep: str | None = " ",
|
|
end: str | None = "\n",
|
|
file: SupportsWrite[str] | None = None,
|
|
flush: Literal[False] = False,
|
|
) -> None:
|
|
"""Prints content to the console with optional color formatting.
|
|
|
|
Args:
|
|
content: Either a string or a list of ColoredText objects for multicolor output.
|
|
color: Optional color for the text when content is a string. Ignored when content is a list.
|
|
sep: Separator to use between the text and color.
|
|
end: String appended after the last value.
|
|
file: A file-like object (stream); defaults to the current sys.stdout.
|
|
flush: Whether to forcibly flush the stream.
|
|
"""
|
|
if isinstance(content, str):
|
|
content = [ColoredText(content, color)]
|
|
print(
|
|
"".join(
|
|
f"{_COLOR_CODES[c.color] if c.color else ''}{c.text}{RESET}"
|
|
for c in content
|
|
),
|
|
sep=sep,
|
|
end=end,
|
|
file=file,
|
|
flush=flush,
|
|
)
|