mirror of
https://github.com/Textualize/textual.git
synced 2025-10-17 02:38:12 +03:00
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Iterable
|
|
|
|
from textual.app import App, ComposeResult, SystemCommand
|
|
from textual.containers import Grid
|
|
from textual.screen import ModalScreen, Screen
|
|
from textual.widgets import Button, Label
|
|
|
|
|
|
class QuitScreen(ModalScreen[bool]):
|
|
"""Screen with a dialog to quit."""
|
|
|
|
def compose(self) -> ComposeResult:
|
|
yield Grid(
|
|
Label("Are you sure you want to quit?", id="question"),
|
|
Button("Quit", variant="error", id="quit"),
|
|
Button("Cancel", variant="primary", id="cancel"),
|
|
id="dialog",
|
|
)
|
|
|
|
def on_button_pressed(self, event: Button.Pressed) -> None:
|
|
if event.button.id == "quit":
|
|
self.dismiss(True)
|
|
else:
|
|
self.dismiss(False)
|
|
|
|
|
|
class ModalApp(App):
|
|
"""An app with a modal dialog."""
|
|
|
|
BINDINGS = [("q", "request_quit", "Quit")]
|
|
|
|
def __init__(self) -> None:
|
|
self.check_quit_called = False
|
|
super().__init__()
|
|
|
|
def get_system_commands(self, screen: Screen) -> Iterable[SystemCommand]:
|
|
yield from super().get_system_commands(screen)
|
|
yield SystemCommand(
|
|
"try a modal quit dialog", "this should work", self.action_request_quit
|
|
)
|
|
|
|
def action_request_quit(self) -> None:
|
|
"""Action to display the quit dialog."""
|
|
|
|
def check_quit(quit: bool | None) -> None:
|
|
"""Called when QuitScreen is dismissed."""
|
|
self.check_quit_called = True
|
|
|
|
self.push_screen(QuitScreen(), check_quit)
|
|
|
|
|
|
async def test_command_dismiss():
|
|
"""Regression test for https://github.com/Textualize/textual/issues/5512"""
|
|
app = ModalApp()
|
|
|
|
async with app.run_test() as pilot:
|
|
await pilot.press("ctrl+p", *"modal quit", "enter")
|
|
await pilot.pause()
|
|
await pilot.press("enter")
|
|
assert app.check_quit_called
|