109 lines
2.9 KiB
Python
109 lines
2.9 KiB
Python
"""Confirmation dialog screen for destructive actions."""
|
|
|
|
from textual import on
|
|
from textual.app import ComposeResult
|
|
from textual.binding import Binding
|
|
from textual.containers import Horizontal, Vertical, Container
|
|
from textual.screen import ModalScreen
|
|
from textual.widgets import Button, Label, Static
|
|
|
|
|
|
class ConfirmDialog(ModalScreen[bool]):
|
|
"""A modal confirmation dialog that returns True if confirmed, False otherwise."""
|
|
|
|
BINDINGS = [
|
|
Binding("escape", "cancel", "Cancel"),
|
|
Binding("enter", "confirm", "Confirm"),
|
|
Binding("y", "confirm", "Yes"),
|
|
Binding("n", "cancel", "No"),
|
|
]
|
|
|
|
DEFAULT_CSS = """
|
|
ConfirmDialog {
|
|
align: center middle;
|
|
}
|
|
|
|
ConfirmDialog #confirm-container {
|
|
width: 50;
|
|
height: auto;
|
|
min-height: 7;
|
|
background: $surface;
|
|
border: thick $primary;
|
|
padding: 1 2;
|
|
}
|
|
|
|
ConfirmDialog #confirm-title {
|
|
text-style: bold;
|
|
width: 1fr;
|
|
height: 1;
|
|
text-align: center;
|
|
margin-bottom: 1;
|
|
}
|
|
|
|
ConfirmDialog #confirm-message {
|
|
width: 1fr;
|
|
height: 1;
|
|
text-align: center;
|
|
margin-bottom: 1;
|
|
}
|
|
|
|
ConfirmDialog #confirm-buttons {
|
|
width: 1fr;
|
|
height: 3;
|
|
align: center middle;
|
|
margin-top: 1;
|
|
}
|
|
|
|
ConfirmDialog Button {
|
|
margin: 0 1;
|
|
}
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
title: str = "Confirm",
|
|
message: str = "Are you sure?",
|
|
confirm_label: str = "Yes",
|
|
cancel_label: str = "No",
|
|
**kwargs,
|
|
):
|
|
"""Initialize the confirmation dialog.
|
|
|
|
Args:
|
|
title: The dialog title
|
|
message: The confirmation message
|
|
confirm_label: Label for the confirm button
|
|
cancel_label: Label for the cancel button
|
|
"""
|
|
super().__init__(**kwargs)
|
|
self._title = title
|
|
self._message = message
|
|
self._confirm_label = confirm_label
|
|
self._cancel_label = cancel_label
|
|
|
|
def compose(self) -> ComposeResult:
|
|
with Container(id="confirm-container"):
|
|
yield Label(self._title, id="confirm-title")
|
|
yield Label(self._message, id="confirm-message")
|
|
with Horizontal(id="confirm-buttons"):
|
|
yield Button(self._cancel_label, id="cancel", variant="default")
|
|
yield Button(self._confirm_label, id="confirm", variant="error")
|
|
|
|
def on_mount(self) -> None:
|
|
"""Focus the cancel button by default (safer option)."""
|
|
self.query_one("#cancel", Button).focus()
|
|
|
|
@on(Button.Pressed, "#confirm")
|
|
def handle_confirm(self) -> None:
|
|
self.dismiss(True)
|
|
|
|
@on(Button.Pressed, "#cancel")
|
|
def handle_cancel(self) -> None:
|
|
self.dismiss(False)
|
|
|
|
def action_confirm(self) -> None:
|
|
self.dismiss(True)
|
|
|
|
def action_cancel(self) -> None:
|
|
self.dismiss(False)
|