| from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple |
|
|
| from rich import panel |
| from rich.console import ( |
| Console as RichConsole, |
| Theme, |
| ) |
| from rich.text import Text |
|
|
| from openbb_cli.config.menu_text import RICH_TAGS |
|
|
| if TYPE_CHECKING: |
| from openbb_cli.models.settings import Settings |
|
|
|
|
| class Console: |
| """Create a rich console to wrap the console print with a Panel.""" |
|
|
| def __init__( |
| self, |
| settings: "Settings", |
| style: Optional[Dict[str, Any]] = None, |
| ): |
| """Initialize the ConsoleAndPanel class.""" |
| self._console = RichConsole( |
| theme=Theme(style), |
| highlight=False, |
| soft_wrap=True, |
| ) |
| self._settings = settings |
| self.menu_text = "" |
| self.menu_path = "" |
|
|
| @staticmethod |
| def _filter_rich_tags(text): |
| """Filter out rich tags from text.""" |
| for val in RICH_TAGS: |
| text = text.replace(val, "") |
|
|
| return text |
|
|
| @staticmethod |
| def _blend_text( |
| message: str, color1: Tuple[int, int, int], color2: Tuple[int, int, int] |
| ) -> Text: |
| """Blend text from one color to another.""" |
| text = Text(message) |
| r1, g1, b1 = color1 |
| r2, g2, b2 = color2 |
| dr = r2 - r1 |
| dg = g2 - g1 |
| db = b2 - b1 |
| size = len(text) + 5 |
| for index in range(size): |
| blend = index / size |
| color = f"#{int(r1 + dr * blend):02X}{int(g1 + dg * blend):02X}{int(b1 + db * blend):02X}" |
| text.stylize(color, index, index + 1) |
| return text |
|
|
| def print(self, *args, **kwargs): |
| """Print the text to the console.""" |
| if kwargs and "text" in list(kwargs) and "menu" in list(kwargs): |
| if not self._settings.TEST_MODE: |
| if self._settings.ENABLE_RICH_PANEL: |
| if self._settings.SHOW_VERSION: |
| version = self._settings.VERSION |
| version = f"[param]OpenBB Platform CLI v{version}[/param] (https://openbb.co)" |
| else: |
| version = ( |
| "[param]OpenBB Platform CLI[/param] (https://openbb.co)" |
| ) |
| self._console.print( |
| panel.Panel( |
| "\n" + kwargs["text"], |
| title=kwargs["menu"], |
| subtitle_align="right", |
| subtitle=version, |
| ) |
| ) |
|
|
| else: |
| self._console.print(kwargs["text"]) |
| else: |
| print(self._filter_rich_tags(kwargs["text"])) |
| elif not self._settings.TEST_MODE: |
| self._console.print(*args, **kwargs) |
| else: |
| print(*args, **kwargs) |
|
|
| def input(self, *args, **kwargs): |
| """Get input from the user.""" |
| self.print(*args, **kwargs, end="") |
| return input() |
|
|
