ProgressQuest-cli/pqcli/ui/urwid/views/confirm_view.py

import typing as T

import urwid

from pqcli.ui.urwid.widgets import CustomButton, CustomLineBox, NPile


class ConfirmView(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()