Support easier widget "support classes" importing

CHANGELOG.md

@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](http://semver.org/).
 
 ## [0.11.0] - Unreleased
 
+### Changed
+
+- Breaking change: `TreeNode` can no longer be imported from `textual.widgets`; it is now available via `from textual.widgets.tree import TreeNode`. https://github.com/Textualize/textual/pull/1637
+
 ### Fixed
 
 - Fixed stuck screen  https://github.com/Textualize/textual/issues/1632

examples/json_tree.py

@@ -4,7 +4,8 @@
 from rich.text import Text
 
 from textual.app import App, ComposeResult
-from textual.widgets import Header, Footer, Tree, TreeNode
+from textual.widgets import Header, Footer, Tree
+from textual.widgets.tree import TreeNode
 
 
 class TreeApp(App):

src/textual/widgets/__init__.py

@@ -23,7 +23,6 @@
     from ._static import Static
     from ._text_log import TextLog
     from ._tree import Tree
-    from ._tree_node import TreeNode
     from ._welcome import Welcome
     from ..widget import Widget
 
@@ -44,7 +43,6 @@
     "Static",
     "TextLog",
     "Tree",
-    "TreeNode",
     "Welcome",
 ]
 

src/textual/widgets/data_table.py

@@ -0,0 +1,5 @@
+"""Make non-widget DataTable support classes available."""
+
+from ._data_table import Column, Row
+
+__all__ = ["Column", "Row"]

src/textual/widgets/tree.py

@@ -0,0 +1,5 @@
+"""Make non-widget Tree support classes available."""
+
+from ._tree import TreeNode
+
+__all__ = ["TreeNode"]

tests/tree/test_tree_node_children.py

@@ -1,5 +1,6 @@
 import pytest
-from textual.widgets import Tree, TreeNode
+from textual.widgets import Tree
+from textual.widgets.tree import TreeNode
 
 
 def label_of(node: TreeNode[None]):

tests/tree/test_tree_node_label.py

@@ -1,4 +1,5 @@
-from textual.widgets import Tree, TreeNode
+from textual.widgets import Tree
+from textual.widgets.tree import TreeNode
 from rich.text import Text
 
 

tests/tree/test_tree_node_parent.py

@@ -1,4 +1,4 @@
-from textual.widgets import TreeNode, Tree
+from textual.widgets import Tree
 
 
 def test_tree_node_parent() -> None: