Uploaded Test files
This commit is contained in:
parent
f584ad9d97
commit
2e81cb7d99
16627 changed files with 2065359 additions and 102444 deletions
324
venv/Lib/site-packages/prompt_toolkit/shortcuts/dialogs.py
Normal file
324
venv/Lib/site-packages/prompt_toolkit/shortcuts/dialogs.py
Normal file
|
|
@ -0,0 +1,324 @@
|
|||
import functools
|
||||
from asyncio import get_event_loop
|
||||
from typing import Any, Callable, List, Optional, Tuple, TypeVar
|
||||
|
||||
from prompt_toolkit.application import Application
|
||||
from prompt_toolkit.application.current import get_app
|
||||
from prompt_toolkit.buffer import Buffer
|
||||
from prompt_toolkit.completion import Completer
|
||||
from prompt_toolkit.eventloop import run_in_executor_with_context
|
||||
from prompt_toolkit.filters import FilterOrBool
|
||||
from prompt_toolkit.formatted_text import AnyFormattedText
|
||||
from prompt_toolkit.key_binding.bindings.focus import focus_next, focus_previous
|
||||
from prompt_toolkit.key_binding.defaults import load_key_bindings
|
||||
from prompt_toolkit.key_binding.key_bindings import KeyBindings, merge_key_bindings
|
||||
from prompt_toolkit.layout import Layout
|
||||
from prompt_toolkit.layout.containers import AnyContainer, HSplit
|
||||
from prompt_toolkit.layout.dimension import Dimension as D
|
||||
from prompt_toolkit.styles import BaseStyle
|
||||
from prompt_toolkit.validation import Validator
|
||||
from prompt_toolkit.widgets import (
|
||||
Box,
|
||||
Button,
|
||||
CheckboxList,
|
||||
Dialog,
|
||||
Label,
|
||||
ProgressBar,
|
||||
RadioList,
|
||||
TextArea,
|
||||
ValidationToolbar,
|
||||
)
|
||||
|
||||
__all__ = [
|
||||
"yes_no_dialog",
|
||||
"button_dialog",
|
||||
"input_dialog",
|
||||
"message_dialog",
|
||||
"radiolist_dialog",
|
||||
"checkboxlist_dialog",
|
||||
"progress_dialog",
|
||||
]
|
||||
|
||||
|
||||
def yes_no_dialog(
|
||||
title: AnyFormattedText = "",
|
||||
text: AnyFormattedText = "",
|
||||
yes_text: str = "Yes",
|
||||
no_text: str = "No",
|
||||
style: Optional[BaseStyle] = None,
|
||||
) -> Application[bool]:
|
||||
"""
|
||||
Display a Yes/No dialog.
|
||||
Return a boolean.
|
||||
"""
|
||||
|
||||
def yes_handler() -> None:
|
||||
get_app().exit(result=True)
|
||||
|
||||
def no_handler() -> None:
|
||||
get_app().exit(result=False)
|
||||
|
||||
dialog = Dialog(
|
||||
title=title,
|
||||
body=Label(text=text, dont_extend_height=True),
|
||||
buttons=[
|
||||
Button(text=yes_text, handler=yes_handler),
|
||||
Button(text=no_text, handler=no_handler),
|
||||
],
|
||||
with_background=True,
|
||||
)
|
||||
|
||||
return _create_app(dialog, style)
|
||||
|
||||
|
||||
_T = TypeVar("_T")
|
||||
|
||||
|
||||
def button_dialog(
|
||||
title: AnyFormattedText = "",
|
||||
text: AnyFormattedText = "",
|
||||
buttons: List[Tuple[str, _T]] = [],
|
||||
style: Optional[BaseStyle] = None,
|
||||
) -> Application[_T]:
|
||||
"""
|
||||
Display a dialog with button choices (given as a list of tuples).
|
||||
Return the value associated with button.
|
||||
"""
|
||||
|
||||
def button_handler(v: _T) -> None:
|
||||
get_app().exit(result=v)
|
||||
|
||||
dialog = Dialog(
|
||||
title=title,
|
||||
body=Label(text=text, dont_extend_height=True),
|
||||
buttons=[
|
||||
Button(text=t, handler=functools.partial(button_handler, v))
|
||||
for t, v in buttons
|
||||
],
|
||||
with_background=True,
|
||||
)
|
||||
|
||||
return _create_app(dialog, style)
|
||||
|
||||
|
||||
def input_dialog(
|
||||
title: AnyFormattedText = "",
|
||||
text: AnyFormattedText = "",
|
||||
ok_text: str = "OK",
|
||||
cancel_text: str = "Cancel",
|
||||
completer: Optional[Completer] = None,
|
||||
validator: Optional[Validator] = None,
|
||||
password: FilterOrBool = False,
|
||||
style: Optional[BaseStyle] = None,
|
||||
) -> Application[str]:
|
||||
"""
|
||||
Display a text input box.
|
||||
Return the given text, or None when cancelled.
|
||||
"""
|
||||
|
||||
def accept(buf: Buffer) -> bool:
|
||||
get_app().layout.focus(ok_button)
|
||||
return True # Keep text.
|
||||
|
||||
def ok_handler() -> None:
|
||||
get_app().exit(result=textfield.text)
|
||||
|
||||
ok_button = Button(text=ok_text, handler=ok_handler)
|
||||
cancel_button = Button(text=cancel_text, handler=_return_none)
|
||||
|
||||
textfield = TextArea(
|
||||
multiline=False,
|
||||
password=password,
|
||||
completer=completer,
|
||||
validator=validator,
|
||||
accept_handler=accept,
|
||||
)
|
||||
|
||||
dialog = Dialog(
|
||||
title=title,
|
||||
body=HSplit(
|
||||
[
|
||||
Label(text=text, dont_extend_height=True),
|
||||
textfield,
|
||||
ValidationToolbar(),
|
||||
],
|
||||
padding=D(preferred=1, max=1),
|
||||
),
|
||||
buttons=[ok_button, cancel_button],
|
||||
with_background=True,
|
||||
)
|
||||
|
||||
return _create_app(dialog, style)
|
||||
|
||||
|
||||
def message_dialog(
|
||||
title: AnyFormattedText = "",
|
||||
text: AnyFormattedText = "",
|
||||
ok_text: str = "Ok",
|
||||
style: Optional[BaseStyle] = None,
|
||||
) -> Application[None]:
|
||||
"""
|
||||
Display a simple message box and wait until the user presses enter.
|
||||
"""
|
||||
dialog = Dialog(
|
||||
title=title,
|
||||
body=Label(text=text, dont_extend_height=True),
|
||||
buttons=[Button(text=ok_text, handler=_return_none)],
|
||||
with_background=True,
|
||||
)
|
||||
|
||||
return _create_app(dialog, style)
|
||||
|
||||
|
||||
def radiolist_dialog(
|
||||
title: AnyFormattedText = "",
|
||||
text: AnyFormattedText = "",
|
||||
ok_text: str = "Ok",
|
||||
cancel_text: str = "Cancel",
|
||||
values: Optional[List[Tuple[_T, AnyFormattedText]]] = None,
|
||||
style: Optional[BaseStyle] = None,
|
||||
) -> Application[_T]:
|
||||
"""
|
||||
Display a simple list of element the user can choose amongst.
|
||||
|
||||
Only one element can be selected at a time using Arrow keys and Enter.
|
||||
The focus can be moved between the list and the Ok/Cancel button with tab.
|
||||
"""
|
||||
if values is None:
|
||||
values = []
|
||||
|
||||
def ok_handler() -> None:
|
||||
get_app().exit(result=radio_list.current_value)
|
||||
|
||||
radio_list = RadioList(values)
|
||||
|
||||
dialog = Dialog(
|
||||
title=title,
|
||||
body=HSplit(
|
||||
[Label(text=text, dont_extend_height=True), radio_list],
|
||||
padding=1,
|
||||
),
|
||||
buttons=[
|
||||
Button(text=ok_text, handler=ok_handler),
|
||||
Button(text=cancel_text, handler=_return_none),
|
||||
],
|
||||
with_background=True,
|
||||
)
|
||||
|
||||
return _create_app(dialog, style)
|
||||
|
||||
|
||||
def checkboxlist_dialog(
|
||||
title: AnyFormattedText = "",
|
||||
text: AnyFormattedText = "",
|
||||
ok_text: str = "Ok",
|
||||
cancel_text: str = "Cancel",
|
||||
values: Optional[List[Tuple[_T, AnyFormattedText]]] = None,
|
||||
style: Optional[BaseStyle] = None,
|
||||
) -> Application[List[_T]]:
|
||||
"""
|
||||
Display a simple list of element the user can choose multiple values amongst.
|
||||
|
||||
Several elements can be selected at a time using Arrow keys and Enter.
|
||||
The focus can be moved between the list and the Ok/Cancel button with tab.
|
||||
"""
|
||||
if values is None:
|
||||
values = []
|
||||
|
||||
def ok_handler() -> None:
|
||||
get_app().exit(result=cb_list.current_values)
|
||||
|
||||
cb_list = CheckboxList(values)
|
||||
|
||||
dialog = Dialog(
|
||||
title=title,
|
||||
body=HSplit(
|
||||
[Label(text=text, dont_extend_height=True), cb_list],
|
||||
padding=1,
|
||||
),
|
||||
buttons=[
|
||||
Button(text=ok_text, handler=ok_handler),
|
||||
Button(text=cancel_text, handler=_return_none),
|
||||
],
|
||||
with_background=True,
|
||||
)
|
||||
|
||||
return _create_app(dialog, style)
|
||||
|
||||
|
||||
def progress_dialog(
|
||||
title: AnyFormattedText = "",
|
||||
text: AnyFormattedText = "",
|
||||
run_callback: Callable[[Callable[[int], None], Callable[[str], None]], None] = (
|
||||
lambda *a: None
|
||||
),
|
||||
style: Optional[BaseStyle] = None,
|
||||
) -> Application[None]:
|
||||
"""
|
||||
:param run_callback: A function that receives as input a `set_percentage`
|
||||
function and it does the work.
|
||||
"""
|
||||
loop = get_event_loop()
|
||||
progressbar = ProgressBar()
|
||||
text_area = TextArea(
|
||||
focusable=False,
|
||||
# Prefer this text area as big as possible, to avoid having a window
|
||||
# that keeps resizing when we add text to it.
|
||||
height=D(preferred=10 ** 10),
|
||||
)
|
||||
|
||||
dialog = Dialog(
|
||||
body=HSplit(
|
||||
[
|
||||
Box(Label(text=text)),
|
||||
Box(text_area, padding=D.exact(1)),
|
||||
progressbar,
|
||||
]
|
||||
),
|
||||
title=title,
|
||||
with_background=True,
|
||||
)
|
||||
app = _create_app(dialog, style)
|
||||
|
||||
def set_percentage(value: int) -> None:
|
||||
progressbar.percentage = int(value)
|
||||
app.invalidate()
|
||||
|
||||
def log_text(text: str) -> None:
|
||||
loop.call_soon_threadsafe(text_area.buffer.insert_text, text)
|
||||
app.invalidate()
|
||||
|
||||
# Run the callback in the executor. When done, set a return value for the
|
||||
# UI, so that it quits.
|
||||
def start() -> None:
|
||||
try:
|
||||
run_callback(set_percentage, log_text)
|
||||
finally:
|
||||
app.exit()
|
||||
|
||||
def pre_run() -> None:
|
||||
run_in_executor_with_context(start)
|
||||
|
||||
app.pre_run_callables.append(pre_run)
|
||||
|
||||
return app
|
||||
|
||||
|
||||
def _create_app(dialog: AnyContainer, style: Optional[BaseStyle]) -> Application[Any]:
|
||||
# Key bindings.
|
||||
bindings = KeyBindings()
|
||||
bindings.add("tab")(focus_next)
|
||||
bindings.add("s-tab")(focus_previous)
|
||||
|
||||
return Application(
|
||||
layout=Layout(dialog),
|
||||
key_bindings=merge_key_bindings([load_key_bindings(), bindings]),
|
||||
mouse_support=True,
|
||||
style=style,
|
||||
full_screen=True,
|
||||
)
|
||||
|
||||
|
||||
def _return_none() -> None:
|
||||
" Button handler that returns None. "
|
||||
get_app().exit()
|
||||
Loading…
Add table
Add a link
Reference in a new issue