artin/rich
1
0
Fork 0
mirror of https://github.com/Textualize/rich.git synced 2026-02-06 10:58:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge 7851ded0fe into 0752ff0472

Browse Source
This commit is contained in:
zitro 2026-02-06 10:50:26 +08:00 committed by GitHub
commit 84f85c6f37
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
rich/tree.py
View File

@ -43,6 +43,7 @@ class Tree(JupyterMixin):
expanded: bool = True,
highlight: bool = False,
hide_root: bool = False,
end_new_line: bool = True,
) -> None:
self.label = label
self.style = style
@ -51,6 +52,7 @@ class Tree(JupyterMixin):
self.expanded = expanded
self.highlight = highlight
self.hide_root = hide_root
self.end_new_line = end_new_line
def add(
self,
@ -79,6 +81,7 @@ class Tree(JupyterMixin):
guide_style=self.guide_style if guide_style is None else guide_style,
expanded=expanded,
highlight=self.highlight if highlight is None else highlight,
end_new_line=True,
)
self.children.append(node)
return node
@ -155,7 +158,11 @@ class Tree(JupyterMixin):
post_style=remove_guide_styles,
)
yield from line
yield new_line
# Equivalent of `if not last or (depth == 0) or (depth != 0 and self.end_new_line):`
if not last or depth == 0 or self.end_new_line:
yield new_line
if first and prefix:
prefix[-1] = make_guide(
SPACE if last else CONTINUE, prefix[-1].style or null_style