Add examples of prompt_toolkit

cli/in_put.py

@@ -0,0 +1,97 @@
+import time
+
+from prompt_toolkit.formatted_text import HTML, merge_formatted_text
+from prompt_toolkit.key_binding import KeyBindings
+from prompt_toolkit.key_binding.key_processor import KeyPressEvent as E
+from prompt_toolkit.keys import Keys
+from prompt_toolkit.shortcuts import PromptSession
+from prompt_toolkit.shortcuts.progress_bar import formatters
+from prompt_toolkit.styles import Style
+
+style = Style.from_dict(
+    {
+        "title": "#D08770 underline",
+        "label": "#D8DEE9 bold",
+        "percentage": "#D08770",
+        "bar-a": "bg:#D08770 #D08770",
+        "bar-b": "bg:#D08770 #2E3440",
+        "bar-c": "#D8DEE9",
+        "current": "#D8DEE9",
+        "total": "#D08770",
+        "time-elapsed": "#D8DEE9",
+        "time-left": "#D08770",
+    }
+)
+
+
+custom_formatters = [
+    formatters.Label(suffix=": "),
+    formatters.Bar(start="|", end="|", sym_a="\u2588", sym_b="\u2588", sym_c="\u2591"),
+    formatters.Text(" "),
+    formatters.Progress(),
+    formatters.Text(" "),
+    formatters.Percentage(),
+    formatters.Text(" [elapsed: "),
+    formatters.TimeElapsed(),
+    formatters.Text(" left: "),
+    formatters.TimeLeft(),
+    formatters.Text("]"),
+]
+
+
+def get_toolbar():
+    return f"time: {time.ctime():<30}"
+
+
+def create_confirm_session(
+    message: str, suffix: str = " (y/n) "
+) -> PromptSession[bool]:
+    """
+    Create a `PromptSession` object for the 'confirm' function.
+    """
+    bindings = KeyBindings()
+
+    @bindings.add("y")
+    @bindings.add("Y")
+    def yes(event: E) -> None:
+        session.default_buffer.text = "y"
+        event.app.exit(result=True)
+
+    @bindings.add("n")
+    @bindings.add("N")
+    def no(event: E) -> None:
+        session.default_buffer.text = "n"
+        event.app.exit(result=False)
+
+    @bindings.add("enter")
+    def enter(event: E) -> None:
+        "Accept the current value."
+        session.default_buffer.text = "y"
+        event.app.exit(result=True)
+
+    @bindings.add(Keys.Any)
+    def _(event: E) -> None:
+        "Disallow inserting other text."
+        pass
+
+    complete_message = merge_formatted_text([message, suffix])
+    session: PromptSession[bool] = PromptSession(
+        complete_message, key_bindings=bindings
+    )
+    return session
+
+
+def confirm(message: str = "Confirm?", suffix: str = " (y/n) ") -> bool:
+    """
+    Display a confirmation prompt that returns True/False.
+    """
+    session = create_confirm_session(message, suffix)
+    return session.prompt()
+
+
+def confirm_async(message: str = "Confirm?", suffix: str = " (y/n) ") -> bool:
+    """
+    Display a confirmation prompt that returns True/False.
+    """
+    session = create_confirm_session(message, suffix)
+    return session.prompt_async()