Alternative compose

CHANGELOG.md

@@ -5,6 +5,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](http://keepachangelog.com/)
 and this project adheres to [Semantic Versioning](http://semver.org/).
 
+## Unreleased
+
+### Changed
+
+- Added alternative method of composing Widgets
+
 ## [0.11.1] - 2023-02-17
 
 ### Fixed

examples/code_browser.py

@@ -38,10 +38,10 @@ def compose(self) -> ComposeResult:
         """Compose our UI."""
         path = "./" if len(sys.argv) < 2 else sys.argv[1]
         yield Header()
-        yield Container(
-            DirectoryTree(path, id="tree-view"),
-            Vertical(Static(id="code", expand=True), id="code-view"),
-        )
+        with Container():
+            yield DirectoryTree(path, id="tree-view")
+            with Vertical(id="code-view"):
+                yield Static(id="code", expand=True)
         yield Footer()
 
     def on_mount(self, event: events.Mount) -> None:

src/textual/_compose.py

@@ -0,0 +1,39 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .app import App
+    from .widget import Widget
+
+
+def compose(node: App | Widget) -> list[Widget]:
+    """Compose child widgets.
+
+    Args:
+        node: The parent node.
+
+    Returns:
+        A list of widgets.
+    """
+    app = node.app
+    nodes: list[Widget] = []
+    compose_stack: list[Widget] = []
+    composed: list[Widget] = []
+    app._compose_stacks.append(compose_stack)
+    app._composed.append(composed)
+    try:
+        for child in node.compose():
+            if composed:
+                nodes.extend(composed)
+                composed.clear()
+            if compose_stack:
+                compose_stack[-1]._nodes._append(child)
+            else:
+                nodes.append(child)
+        if composed:
+            nodes.extend(composed)
+            composed.clear()
+    finally:
+        app._compose_stacks.pop()
+    return nodes

src/textual/app.py

@@ -46,6 +46,7 @@
 from ._ansi_sequences import SYNC_END, SYNC_START
 from ._asyncio import create_task
 from ._callback import invoke
+from ._compose import compose
 from ._context import active_app
 from ._event_broker import NoHandler, extract_handler_actions
 from ._path import _make_path_object_relative
@@ -388,6 +389,9 @@ def __init__(
         self._installed_screens: dict[str, Screen | Callable[[], Screen]] = {}
         self._installed_screens.update(**self.SCREENS)
 
+        self._compose_stacks: list[list[Widget]] = []
+        self._composed: list[list[Widget]] = []
+
         self.devtools: DevtoolsClient | None = None
         if "devtools" in self.features:
             try:
@@ -1606,7 +1610,7 @@ async def on_screenshot():
 
     async def _on_compose(self) -> None:
         try:
-            widgets = list(self.compose())
+            widgets = compose(self)
         except TypeError as error:
             raise TypeError(
                 f"{self!r} compose() returned an invalid response; {error}"

src/textual/cli/previews/colors.py

@@ -41,18 +41,14 @@ def compose(self) -> ComposeResult:
         ]
 
         for color_name in ColorSystem.COLOR_NAMES:
-            items: list[Widget] = [Label(f'"{color_name}"')]
-            for level in LEVELS:
-                color = f"{color_name}-{level}" if level else color_name
-                item = ColorItem(
-                    ColorBar(f"${color}", classes="text label"),
-                    ColorBar("$text-muted", classes="muted"),
-                    ColorBar("$text-disabled", classes="disabled"),
-                    classes=color,
-                )
-                items.append(item)
-
-            yield ColorGroup(*items, id=f"group-{color_name}")
+            with ColorGroup(id=f"group-{color_name}"):
+                yield Label(f'"{color_name}"')
+                for level in LEVELS:
+                    color = f"{color_name}-{level}" if level else color_name
+                    with ColorItem(classes=color):
+                        yield ColorBar(f"${color}", classes="text label")
+                        yield ColorBar("$text-muted", classes="muted")
+                        yield ColorBar("$text-disabled", classes="disabled")
 
 
 class ColorsApp(App):

src/textual/cli/previews/easing.py

@@ -73,16 +73,14 @@ def compose(self) -> ComposeResult:
         )
 
         yield EasingButtons()
-        yield Vertical(
-            Horizontal(
-                Label("Animation Duration:", id="label"), duration_input, id="inputs"
-            ),
-            Horizontal(
-                self.animated_bar,
-                Container(self.opacity_widget, id="other"),
-            ),
-            Footer(),
-        )
+        with Vertical():
+            with Horizontal(id="inputs"):
+                yield Label("Animation Duration:", id="label")
+                yield duration_input
+            with Horizontal():
+                yield self.animated_bar
+                yield Container(self.opacity_widget, id="other")
+            yield Footer()
 
     def on_button_pressed(self, event: Button.Pressed) -> None:
         self.bell()

src/textual/widget.py

@@ -5,6 +5,7 @@
 from fractions import Fraction
 from itertools import islice
 from operator import attrgetter
+from types import TracebackType
 from typing import (
     TYPE_CHECKING,
     ClassVar,
@@ -38,6 +39,7 @@
 from ._animator import DEFAULT_EASING, Animatable, BoundAnimator, EasingFunction
 from ._arrange import DockArrangeResult, arrange
 from ._asyncio import create_task
+from ._compose import compose
 from ._context import active_app
 from ._easing import DEFAULT_SCROLL_EASING
 from ._layout import Layout
@@ -363,6 +365,22 @@ def offset(self) -> Offset:
     def offset(self, offset: Offset) -> None:
         self.styles.offset = ScalarOffset.from_offset(offset)
 
+    def __enter__(self) -> None:
+        self.app._compose_stacks[-1].append(self)
+
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: TracebackType | None,
+    ) -> None:
+        compose_stack = self.app._compose_stacks[-1]
+        composed = compose_stack.pop()
+        if compose_stack:
+            compose_stack[-1]._nodes._append(composed)
+        else:
+            self.app._composed[-1].append(composed)
+
     ExpectType = TypeVar("ExpectType", bound="Widget")
 
     @overload
@@ -2444,7 +2462,7 @@ async def handle_key(self, event: events.Key) -> bool:
 
     async def _on_compose(self) -> None:
         try:
-            widgets = list(self.compose())
+            widgets = compose(self)
         except TypeError as error:
             raise TypeError(
                 f"{self!r} compose() returned an invalid response; {error}"