Pilgrim/src/pilgrim/ui/screens/rename_entry_modal.py

59 lines
2.2 KiB
Python

from textual.app import ComposeResult
from textual.binding import Binding
from textual.containers import Vertical, Horizontal
from textual.screen import ModalScreen
from textual.widgets import Label, Input, Button
class RenameEntryModal(ModalScreen[str]):
"""A modal screen to rename a diary entry."""
BINDINGS = [
Binding("escape", "cancel", "Cancel"),
]
def __init__(self, current_name: str):
super().__init__()
self._current_name = current_name
self.name_input = Input(
value=self._current_name,
placeholder="Type the new name...",
id="rename_input",
classes="RenameEntryModal-name-input"
)
def compose(self) -> ComposeResult:
with Vertical(id="rename_entry_dialog", classes="RenameEntryModal-dialog"):
yield Label("Rename Entry", classes="dialog-title RenameEntryModal-title")
yield Label("New Entry Title:", classes="RenameEntryModal-label")
yield self.name_input
with Horizontal(classes="dialog-buttons RenameEntryModal-buttons"):
yield Button("Save", variant="primary", id="save", classes="RenameEntryModal-save-button")
yield Button("Cancel", variant="default", id="cancel", classes="RenameEntryModal-cancel-button")
def on_mount(self) -> None:
"""Focuses on the input when the screen is mounted."""
self.name_input.focus()
self.name_input.cursor_position = len(self.name_input.value)
def on_button_pressed(self, event: Button.Pressed) -> None:
"""Handles button clicks."""
if event.button.id == "save":
new_name = self.name_input.value.strip()
if new_name:
self.dismiss(new_name) # Returns the new name
else:
self.dismiss(None) # Considers empty name as cancellation
else:
self.dismiss(None) # Returns None for cancellation
def on_input_submitted(self, event: Input.Submitted) -> None:
"""Allows saving by pressing Enter."""
new_name = event.value.strip()
if new_name:
self.dismiss(new_name)
else:
self.dismiss(None)
def action_cancel(self) -> None:
self.dismiss(None)