second commit

env/lib/python3.11/site-packages/rich_toolkit/toolkit.py

@@ -0,0 +1,111 @@
+from typing import Any, Dict, List, Union
+
+from rich.console import Console, RenderableType
+from rich.theme import Theme
+
+from .styles.base import BaseStyle
+from .input import Input
+from .menu import Menu, Option, ReturnValue
+from .progress import Progress
+
+
+class RichToolkitTheme:
+    def __init__(self, style: BaseStyle, theme: Dict[str, str]) -> None:
+        self.style = style
+        self.rich_theme = Theme(theme)
+
+
+class RichToolkit:
+    def __init__(
+        self,
+        theme: RichToolkitTheme,
+        handle_keyboard_interrupts: bool = True,
+    ) -> None:
+        self.console = Console(theme=theme.rich_theme)
+        self.theme = theme
+        self.handle_keyboard_interrupts = handle_keyboard_interrupts
+
+    def __enter__(self):
+        self.console.print()
+        return self
+
+    def __exit__(
+        self, exc_type: Any, exc_value: Any, traceback: Any
+    ) -> Union[bool, None]:
+        if self.handle_keyboard_interrupts and exc_type is KeyboardInterrupt:
+            # we want to handle keyboard interrupts gracefully, instead of showing a traceback
+            # or any other error message
+            return True
+
+        self.console.print()
+
+    def print_title(self, title: str, **metadata: Any) -> None:
+        self.console.print(
+            self.theme.style.with_decoration(title, title=True, **metadata)
+        )
+
+    def print(self, *renderables: RenderableType, **metadata: Any) -> None:
+        self.console.print(self.theme.style.with_decoration(*renderables, **metadata))
+
+    def print_as_string(self, *renderables: RenderableType, **metadata: Any) -> str:
+        with self.console.capture() as capture:
+            self.print(*renderables, **metadata)
+
+        return capture.get().rstrip()
+
+    def print_line(self) -> None:
+        self.console.print(self.theme.style.empty_line())
+
+    def confirm(self, title: str, **metadata: Any) -> bool:
+        return self.ask(
+            title=title,
+            options=[
+                Option({"value": True, "name": "Yes"}),
+                Option({"value": False, "name": "No"}),
+            ],
+            inline=True,
+            **metadata,
+        )
+
+    def ask(
+        self,
+        title: str,
+        options: List[Option[ReturnValue]],
+        inline: bool = False,
+        allow_filtering: bool = False,
+        **metadata: Any,
+    ) -> ReturnValue:
+        return Menu(
+            title=title,
+            options=options,
+            console=self.console,
+            style=self.theme.style,
+            inline=inline,
+            allow_filtering=allow_filtering,
+            cursor_offset=self.theme.style.cursor_offset,
+            **metadata,
+        ).ask()
+
+    def input(
+        self, title: str, default: str = "", password: bool = False, **metadata: Any
+    ) -> str:
+        return Input(
+            console=self.console,
+            style=self.theme.style,
+            title=title,
+            default=default,
+            cursor_offset=self.theme.style.cursor_offset,
+            password=password,
+            **metadata,
+        ).ask()
+
+    def progress(
+        self, title: str, transient: bool = False, transient_on_error: bool = False
+    ) -> Progress:
+        return Progress(
+            title=title,
+            console=self.console,
+            style=self.theme.style,
+            transient=transient,
+            transient_on_error=transient_on_error,
+        )