mirror of
https://github.com/rr-/pq-cli.git
synced 2026-09-29 06:44:00 -06:00
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
import typing as T
|
|
|
|
import urwid
|
|
|
|
from pqcli.ui.custom_button import CustomButton
|
|
from pqcli.ui.custom_line_box import CustomLineBox
|
|
from pqcli.ui.layout import NPile
|
|
|
|
|
|
class ConfirmDialog(urwid.Overlay):
|
|
signals = ["confirm", "cancel"]
|
|
|
|
def __init__(self, label: str, parent: urwid.Widget) -> None:
|
|
question = urwid.Text(("bold", label), "center")
|
|
yes_btn = CustomButton(
|
|
"Yes", hint="Y", on_press=lambda _user_data: self.confirm()
|
|
)
|
|
no_btn = CustomButton(
|
|
"No", hint="N", on_press=lambda _user_data: self.cancel()
|
|
)
|
|
|
|
line_box = CustomLineBox(
|
|
urwid.Filler(NPile([question, no_btn, yes_btn], outermost=True))
|
|
)
|
|
|
|
super().__init__(
|
|
line_box, parent, "center", len(label) + 6, "middle", 5
|
|
)
|
|
|
|
def keypress(self, size: T.Any, key: str) -> T.Optional[str]:
|
|
if key in {"y", "Y"}:
|
|
self.confirm()
|
|
return None
|
|
if key in {"n", "N"}:
|
|
self.cancel()
|
|
return None
|
|
return T.cast(T.Optional[str], super().keypress(size, key))
|
|
|
|
def confirm(self) -> None:
|
|
self._emit("confirm")
|
|
|
|
def cancel(self) -> None:
|
|
self._emit("cancel")
|
|
|
|
def tick(self) -> None:
|
|
if hasattr(self.bottom_w, "tick"):
|
|
self.bottom_w.tick()
|