1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
|
from textual import on, work
from textual.app import App, ComposeResult
from textual.screen import Screen
from textual.widgets import Button, Label
class QuestionScreen(Screen[bool]):
"""Screen with a parameter."""
def __init__(self, question: str) -> None:
self.question = question
super().__init__()
def compose(self) -> ComposeResult:
yield Label(self.question)
yield Button("Yes", id="yes", variant="success")
yield Button("No", id="no")
@on(Button.Pressed, "#yes")
def handle_yes(self) -> None:
self.dismiss(True) # (1)!
@on(Button.Pressed, "#no")
def handle_no(self) -> None:
self.dismiss(False) # (2)!
class QuestionsApp(App):
"""Demonstrates wait_for_dismiss"""
CSS_PATH = "questions01.tcss"
@work # (3)!
async def on_mount(self) -> None:
if await self.push_screen_wait( # (4)!
QuestionScreen("Do you like Textual?"),
):
self.notify("Good answer!")
else:
self.notify(":-(", severity="error")
if __name__ == "__main__":
app = QuestionsApp()
app.run()
|