tree auto width

Author: willmcgugan

src/textual/app.py

@@ -1080,9 +1080,9 @@ def _get_screen(self, screen: Screen | str) -> tuple[Screen, AwaitMount]:
         _screen = self.get_screen(screen)
         if not _screen.is_running:
             widgets = self._register(self, _screen)
-            return (_screen, AwaitMount(widgets))
+            return (_screen, AwaitMount(_screen, widgets))
         else:
-            return (_screen, AwaitMount([]))
+            return (_screen, AwaitMount(_screen, []))
 
     def _replace_screen(self, screen: Screen) -> Screen:
         """Handle the replaced screen.
@@ -1128,7 +1128,7 @@ def switch_screen(self, screen: Screen | str) -> AwaitMount:
             self.screen.post_message_no_wait(events.ScreenResume(self))
             self.log.system(f"{self.screen} is current (SWITCHED)")
             return await_mount
-        return AwaitMount([])
+        return AwaitMount(self.screen, [])
 
     def install_screen(self, screen: Screen, name: str | None = None) -> AwaitMount:
         """Install a screen.
@@ -1563,7 +1563,6 @@ def _register(
                 if widget.children:
                     self._register(widget, *widget.children)
                 apply_stylesheet(widget)
-
         return list(widgets)
 
     def _unregister(self, widget: Widget) -> None:

src/textual/css/constants.py

@@ -36,20 +36,21 @@
 VALID_TEXT_ALIGN: Final = {"start", "end", "left", "right", "center", "justify"}
 VALID_SCROLLBAR_GUTTER: Final = {"auto", "stable"}
 VALID_STYLE_FLAGS: Final = {
-    "none",
-    "not",
-    "bold",
+    "b",
     "blink",
+    "bold",
+    "dim",
+    "i",
     "italic",
-    "underline",
+    "none",
+    "not",
+    "o",
     "overline",
+    "reverse",
     "strike",
-    "b",
-    "i",
     "u",
+    "underline",
     "uu",
-    "o",
-    "reverse",
 }
 
 NULL_SPACING: Final = Spacing.all(0)

src/textual/dom.py

@@ -507,8 +507,8 @@ def text_style(self) -> Style:
     @property
     def rich_style(self) -> Style:
         """Get a Rich Style object for this DOMNode."""
-        background = WHITE
-        color = BLACK
+        background = Color(0, 0, 0, 0)
+        color = Color(255, 255, 255, 0)
         style = Style()
         for node in reversed(self.ancestors_with_self):
             styles = node.styles
@@ -520,7 +520,8 @@ def rich_style(self) -> Style:
             if styles.has_rule("auto_color") and styles.auto_color:
                 color = background.get_contrast_text(color.a)
         style += Style.from_color(
-            (background + color).rich_color, background.rich_color
+            (background + color).rich_color if (background.a or color.a) else None,
+            background.rich_color if background.a else None,
         )
         return style
 

src/textual/widget.py

@@ -85,7 +85,8 @@ class AwaitMount:
 
     """
 
-    def __init__(self, widgets: Sequence[Widget]) -> None:
+    def __init__(self, parent: Widget, widgets: Sequence[Widget]) -> None:
+        self._parent = parent
         self._widgets = widgets
 
     def __await__(self) -> Generator[None, None, None]:
@@ -97,6 +98,7 @@ async def await_mount() -> None:
                 ]
                 if aws:
                     await wait(aws)
+                    self._parent.refresh(layout=True)
 
         return await_mount().__await__()
 
@@ -595,12 +597,12 @@ def mount(
         else:
             parent = self
 
-        return AwaitMount(
-            self.app._register(
-                parent, *widgets, before=insert_before, after=insert_after
-            )
+        mounted = self.app._register(
+            parent, *widgets, before=insert_before, after=insert_after
         )
 
+        return AwaitMount(self, mounted)
+
     def move_child(
         self,
         child: int | Widget,

src/textual/widgets/_tree.py

@@ -614,6 +614,11 @@ def _tree_lines(self) -> list[_TreeLine]:
         assert self._tree_lines_cached is not None
         return self._tree_lines_cached
 
+    def _on_idle(self) -> None:
+        """Check tree needs a rebuild on idle."""
+        # Property calls build if required
+        self._tree_lines
+
     def _build(self) -> None:
         """Builds the tree by traversing nodes, and creating tree lines."""
 
@@ -805,7 +810,7 @@ async def _on_click(self, event: events.Click) -> None:
             cursor_line = meta["line"]
             if meta.get("toggle", False):
                 node = self.get_node_at_line(cursor_line)
-                if node is not None:
+                if node is not None and self.auto_expand:
                     self._toggle_node(node)
 
             else: