AHjY|UdZddlmZddlmZddlmZmZmZm Z m Z m Z m Z ddl ZddlmZmZddlmZmZddlmZmZdd lmZdd lmZdd lmZdd lmZm Z dd l!m"Z"ddl#m$Z$m%Z%m&Z&ddl'm(Z(ddl)m*Z*m+Z+ddl,m-Z-ddl.m/Z/erddl0m1Z1m2Z2e de3Z4 e dZ5 e dZ6 dZ7de8d<ejrddiZ:Gdde;Z<Gdde;Z=Gd d!e;Z>eGd"d#ee5Z?Gd$d%ed&Z@ejjGd'd(ee5ZCGd)d*ee5e-d+ZDy),zProvides a tree widget.) annotations) dataclass) TYPE_CHECKINGClassVarGenericIterableNewTypeTypeVarcastN) NULL_STYLEStyle)TextTextType)eventson)ImmutableSequenceView) loop_last)line_pad)Binding BindingType)LRUCache)RegionSizeclamp)Message)reactivevar) ScrollView)Strip)Self TypeAliasNodeID TreeDataTypeEventTreeDataTypeztuple[int | tuple, ...]r! LineCacheKeytoggleTceZdZdZy)RemoveRootErrorzaException raised when trying to remove the root of a [`TreeNode`][textual.widgets.tree.TreeNode].N__name__ __module__ __qualname____doc__M/root/tools/cai/cai_env/lib/python3.12/site-packages/textual/widgets/_tree.pyr(r(-skr/r(ceZdZdZy) UnknownNodeIDz]Exception raised when referring to an unknown [`TreeNode`][textual.widgets.tree.TreeNode] ID.Nr)r.r/r0r2r21sgr/r2ceZdZdZy) AddNodeErrorzEException raised when there is an error with a request to add a node.Nr)r.r/r0r4r45sOr/r4c<eZdZUded<ded<eddZd dZy) _TreeLinelist[TreeNode[TreeDataType]]pathboollastc |jdS)z#The node associated with this line.)r8selfs r0nodez_TreeLine.node>syy}r/c|r'tdt|jdz |z}|Sd}t|jdkDr|t|jdz |zz }|S)zGet the cell width of the line as rendered. Args: guide_depth: The guide depth (cells in the indentation). Returns: Width in cells. r)maxlenr8)r> guide_depth show_rootwidths r0_get_guide_widthz_TreeLine._get_guide_widthCsd C NQ./;>E  E499~!#dii.1, ;; r/N)returnTreeNode[TreeDataType])rDintrEr9rHrJ)r*r+r,__annotations__propertyr?rGr.r/r0r6r69s$ && J r/r6ceZdZdZy) TreeNodesz&An immutable collection of `TreeNode`.Nr)r.r/r0rNrNVs0r/rNrIceZdZdZ d/ddd d0dZd1dZd2dZed3dZed4d Z ed4d Z ed5d Z ed6d Z e jd7d Z ed6dZejd8dZed9dZed:dZed:dZed:dZed6dZed6dZed6dZed6dZed6dZejd;dZddZd=dZd=dZd=d Zd=d!Zed?d"Z e jd@d#Z dAd$Z! d/ddd%dd& dBd'Z" d/ddd( dCd)Z#d2d*Z$d2d+Z%d2d,Z&d2d-Z'd2d.Z(y)DTreeNodez5An object that represents a "node" in a tree control.NT)expanded allow_expandc||_||_||_|j||_||_ ||_g|_d|_d|_ ||_ d|_ d|_ y)aInitialise the node. Args: tree: The tree that the node is being attached to. parent: The parent node that this node is being attached to. id: The ID of the node. label: The label for the node. data: Optional data to associate with the node. expanded: Should the node be attached in an expanded state? allow_expand: Should the node allow being expanded by the user? Frr<N) _tree_parent_id process_label_labeldata _expanded _children_hover_ _selected_ _allow_expand_updates_line)r>treeparentidlabelrYrQrRs r0__init__zTreeNode.__init__^sg,  ((/  :!79 )  r/c#XK|jj|jywN)rXplainrYr=s r0 __rich_repr__zTreeNode.__rich_repr__skkiis(*cJd|_d|_|xjdz c_y)NFrA)r\r]r_r=s r0_resetzTreeNode._resets   r/c|jS)z'The tree that this node is attached to.)rTr=s r0raz TreeNode.treezzr/c,t|jS)zThe child nodes of a TreeNode.)rNr[r=s r0childrenzTreeNode.childrens((r/c^|j t|gS|jjS)z*The siblings of this node (includes self).)rbrNror=s r0siblingszTreeNode.siblingss+ ;; dV$ $;;'' 'r/c|jS)zhovers r0ruzTreeNode._hovers   r/c|jS)zCheck if the node is selected.)r]r=s r0 _selectedzTreeNode._selectedsr/c<|xjdz c_||_yrx)r_r])r>selecteds r0r{zTreeNode._selecteds  "r/c|jS)zThe ID of the node.)rVr=s r0rcz TreeNode.ids xxr/c|jS)zThe parent of the node.)rUr=s r0rbzTreeNode.parentrvr/cl|j}|j|dz} ||S#t$rYywxYw)z The next sibling below the node.rANrqindex IndexErrorr>rqrs r0 next_siblingzTreeNode.next_siblingsA==t$q( E? "  s ' 33cx|j}|j|dz }|dkry ||S#t$rYywxYw)z$The previous sibling below the node.rArNrrs r0previous_siblingzTreeNode.previous_siblingsK==t$q( 19 E? "  s - 99c|jS)zIs the node expanded?rZr=s r0 is_expandedzTreeNode.is_expandeds~~r/c|j S)zIs the node collapsed?rr=s r0 is_collapsedzTreeNode.is_collapseds>>!!r/c|jyt|jjxr|jjd|k(S)z*Is this the last child node of its parent?Tr<)rUr9r[r=s r0is_lastzTreeNode.is_lastsD <<  LL " " It||'='=b'AT'I  r/c4||jjk(S)z"Is this node the root of the tree?)rTrootr=s r0is_rootzTreeNode.is_rootstzz&&r/c|jS)zIs this node allowed to expand?)r^r=s r0rRzTreeNode.allow_expands!!!r/c<||_|xjdz c_yrx)r^r_)r>rRs r0rRzTreeNode.allow_expands)  r/cd|_|xjdz c_|jjtj |j |j|r#|jD]}|j|yy)zMark the node as expanded (its children are shown). Args: expand_all: If `True` expand all offspring at all depths. TrAN) rZr_rT post_messageTree NodeExpanded set_senderro_expand)r> expand_allchilds r0rzTreeNode._expandsl     1 1$ 7 B B4:: NO  * j) * r/c\|jd|jj|S)zdExpand the node (show its children). Returns: The `TreeNode` instance. FrrT _invalidater=s r0expandzTreeNode.expands% U   r/c\|jd|jj|S)z{Expand the node (show its children) and all those below it. Returns: The `TreeNode` instance. Trr=s r0rzTreeNode.expand_alls% T   r/cd|_|xjdz c_|jjtj |j |j|r#|jD]}|j|yy)zMark the node as collapsed (its children are hidden). Args: collapse_all: If `True` collapse all offspring at all depths. FrAN) rZr_rTrr NodeCollapsedrro _collapse)r> collapse_allrs r0rzTreeNode._collapsesl     2 24 8 C CDJJ OP  . - . r/c\|jd|jj|S)zfCollapse the node (hide its children). Returns: The `TreeNode` instance. FrrTrr=s r0collapsezTreeNode.collapse's% u   r/c\|jd|jj|S)z}Collapse the node (hide its children) and all those below it. Returns: The `TreeNode` instance. Trr=s r0rzTreeNode.collapse_all1s% t   r/cb|jr|j|S|j|S)zaToggle the node's expanded state. Returns: The `TreeNode` instance. )rZrrr=s r0r&zTreeNode.toggle;s+ >> MMO  KKM r/cb|jr|j|S|j|S)zToggle the node's expanded state and make all those below it match. Returns: The `TreeNode` instance. )rZrrr=s r0 toggle_allzTreeNode.toggle_allGs/ >>      OO  r/c|jS)zThe label for the node.)rXr=s r0rdzTreeNode.labelSs{{r/c&|j|yrg) set_label)r> new_labels r0rdzTreeNode.labelXs y!r/c|xjdz c_|jj|}||_|jj |jj |y)zySet a new label for the node. Args: label: A ``str`` or ``Text`` object with the new label. rAN)r_rTrWrX call_later _refresh_node)r>rd text_labels r0rzTreeNode.set_label\sK  ZZ--e4    djj66=r/FbeforeafterrrRcJ| | tdt|j}|Kt|tr|}n8t|t r |jj |}n td|nt|tr#|dz}|dkrT|t|jz }n;t|t r |jj |dz}n td|jj|}|jj|||} || _ || _ |xjdz c_|jj!|| |jj#| S#t$r tdwxYw#t$r tdwxYw)aAdd a node to the sub-tree. Args: label: The new node's label. data: Data associated with the new node. before: Optional index or `TreeNode` to add the node before. after: Optional index or `TreeNode` to add the node after. expand: Node should be expanded. allow_expand: Allow user to expand the node via keyboard or mouse. Returns: A new Tree node Raises: AddNodeError: If there is a problem with the addition request. Note: Only one of `before` or `after` can be provided. If both are provided a `AddNodeError` will be raised. z1Unable to add a node both before and after a nodez;The node specified for `before` is not a child of this nodezE`before` argument must be an index or a TreeNode object to add beforerArz:The node specified for `after` is not a child of this nodezC`after` argument must be an index or a TreeNode object to add after)r4rCro isinstancerJrPr ValueError TypeErrorrTrW _add_noderZr^r_r[insertr) r>rdrYrrrrR insert_indexrr?s r0addz TreeNode.addgs<  %"3RS S .  &#&% FH-#'==#6#6v#>L  [  %%$qy 19 C $66LE8,#'==#6#6u#=#AL  YZZ--e4 zz##D*d;)   lD1   E"&U""&Ts E58F 5F  F")rrc4|j||||dd}|S)a1Add a 'leaf' node (a node that can not expand). Args: label: Label for the node. data: Optional data. before: Optional index or `TreeNode` to add the node before. after: Optional index or `TreeNode` to add the node after. Returns: New node. Raises: AddNodeError: If there is a problem with the addition request. Note: Only one of `before` or `after` can be provided. If both are provided a `AddNodeError` will be raised. Fr)r)r>rdrYrrr?s r0add_leafzTreeNode.add_leafs04xx     r/cXt|jD]}|jy)zRemove child nodes of this node. Note: This is the internal support method for `remove_children`. Call `remove_children` to ensure the tree gets refreshed. N)reversedr[_remove)r>rs r0_remove_childrenzTreeNode._remove_childrens%dnn- E MMO r/c|j|jJ|jj|jjj|=|jj |j =y)zRemove the current node and all its children. Note: This is the internal support method for `remove`. Call `remove` to ensure the tree gets refreshed. N)rrUr[rrT _tree_nodesrcr=s r0rzTreeNode._removes] ||''' LL " "4<<#9#9#?#?#E F JJ " "477 +r/c|jr td|j|jj y)zRemove this node from the tree. Raises: RemoveRootError: If there is an attempt to remove the root. z*Attempt to remove the root node of a Tree.N)rr(rrTrr=s r0removezTreeNode.removes0 <<!"NO O   r/cX|j|jjy)z$Remove any child nodes of this node.N)rrTrr=s r0remove_childrenzTreeNode.remove_childrens   r/cx|xjdz c_|jj|jy)z*Initiate a refresh (repaint) of this node.rAN)r_rT _refresh_liner`r=s r0refreshzTreeNode.refreshs&     ,r/rg)raTree[TreeDataType]rbTreeNode[TreeDataType] | Nonercr"rdrrYTreeDataType | NonerQr9rRr9rHNone)rHzrich.repr.ResultrHr)rHr)rHzTreeNodes[TreeDataType]rHrJ)rHr9)ryr9rHr)r}r9rHrrHr"rHr)rRr9rHr)rr9rHrrHr )rr9rHr)rHr)rrrHr)rdrrHr)rdrrYrr#int | TreeNode[TreeDataType] | Nonerrrr9rRr9rHrI) rdrrYrrrrrrHrI))r*r+r,r-rerirkrLrarorqrsrusetterr{rcrbrrrrrrrRrrrrrrr&rrdrrrrrrrrr.r/r0rPrPZs?%) #!# #.#  #  # " ### #J ))(( ]]##  ""  ''"" * .   \\"" >%)K 7;59!KK"K 4 K 3 KKK K`%)" 7;59 """" 4 " 3 " "H , !! -r/rPceZdZUdZdZ dZ edddded d d ded d ddeddddeddddeddddeddddeddddeddd dg Zd!ed"< hd#Z d$ed%< d&Z e d'Z e d(Z e d(d')Z e d'Z e d*d+Z e d'Z e dZ d,d-d.d/Zd0ed1<Gd2d3eeeZGd4d5eeeZGd6d7eeeZGd8d9eeeZ dvd:d:d:dd; dwfd<Zdvdxd=Zedyd>Zedzd?Z d{d@Z! d| d}dAZ" d~dBZ#ddCZ$ddDZ%ddEZ&dvddFZ' d| ddGZ(d|ddHZ)ddIZ*ddJZ+e,eddKZ-ddLZ.ddMZ/ddNZ0ddOZ1ddPZ2ddQZ3ddRZ4ddSZ5ddTZ6ddUZ7ddVZ8ddWZ9ddXZ:ddYZ;dddZZ< d dd[Z=dd\Z>dd]Z?dd^Z@edd_ZAdd`ZBddaZCdfdb ZDddcZEdddZFddeZGddfZHddgZIdfdh ZJddiZKddjZLddkZMddlZNddmZOddnZPddoZQddpZRddqZSddrZTddsZUddtZVdduZWxZXS)rz6A widget for displaying and navigating data in a tree.u▶ u▼ z shift+left cursor_parentzCursor to parentF)showz shift+rightcursor_parent_next_siblingzCursor to next ancestorzshift+upcursor_previous_siblingzCursor to previous siblingz shift+downcursor_next_siblingzCursor to next siblingenter select_cursorSelectspace toggle_nodeTogglez shift+spacetoggle_expand_allzExpand or collapse allup cursor_upz Cursor Updown cursor_downz Cursor DownzClassVar[list[BindingType]]BINDINGS> tree--label tree--cursor tree--guidestree--highlighttree--guides-hovertree--highlight-linetree--guides-selectedzClassVar[set[str]]COMPONENT_CLASSESa Tree { background: $surface; color: $foreground; & > .tree--label {} & > .tree--guides { color: $surface-lighten-2; } & > .tree--guides-hover { color: $surface-lighten-2; } & > .tree--guides-selected { color: $block-cursor-blurred-background; } & > .tree--cursor { text-style: $block-cursor-blurred-text-style; background: $block-cursor-blurred-background; } & > .tree--highlight {} & > .tree--highlight-line { background: $block-hover-background; } &:focus { background-tint: $foreground 5%; & > .tree--cursor { color: $block-cursor-foreground; background: $block-cursor-background; text-style: $block-cursor-text-style; } & > .tree--guides { color: $surface-lighten-3; } & > .tree--guides-hover { color: $surface-lighten-3; } & > .tree--guides-selected { color: $block-cursor-background; } } &:light { /* In light mode the guides are darker*/ & > .tree--guides { color: $surface-darken-1; } & > .tree--guides-hover { color: $block-cursor-background; } & > .tree--guides-selected { color: $block-cursor-background; } } &:ansi { color: ansi_default; & > .tree--guides { color: ansi_green; } &:nocolor > .tree--cursor{ text-style: reverse; } } } Tr<) always_update)init) u│ u└─u├─)ru┃ u┗━u┣━)ru║ u╚═u╠═)defaultbolddoublez$dict[str, tuple[str, str, str, str]]LINESc6eZdZdZdfd ZeddZxZS)Tree.NodeCollapsedzEvent sent when a node is collapsed. Can be handled using `on_tree_node_collapsed` in a subclass of `Tree` or in a parent node in the DOM. c2||_ t| yrgr?superrer>r? __class__s r0rezTree.NodeCollapsed.__init__s59DI . G  r/c.|jjSzThe tree that sent the message.r?rar=s r0controlzTree.NodeCollapsed.control99>> !r/r?zTreeNode[EventTreeDataType]rHrrHzTree[EventTreeDataType]r*r+r,r-rerLr __classcell__rs@r0rr!    "  "r/rc6eZdZdZdfd ZeddZxZS)Tree.NodeExpandedzEvent sent when a node is expanded. Can be handled using `on_tree_node_expanded` in a subclass of `Tree` or in a parent node in the DOM. c2||_ t| yrgrrs r0rezTree.NodeExpanded.__init__59DI - G  r/c.|jjSrrr=s r0rzTree.NodeExpanded.controlrr/rrr r s@r0rrr r/rc6eZdZdZdfd ZeddZxZS)Tree.NodeHighlightedzEvent sent when a node is highlighted. Can be handled using `on_tree_node_highlighted` in a subclass of `Tree` or in a parent node in the DOM. c2||_ t| yrgrrs r0rezTree.NodeHighlighted.__init__s59DI 0 G  r/c.|jjSrrr=s r0rzTree.NodeHighlighted.controlrr/rrr r s@r0NodeHighlightedrr r/rc6eZdZdZdfd ZeddZxZS)Tree.NodeSelectedzEvent sent when a node is selected. Can be handled using `on_tree_node_selected` in a subclass of `Tree` or in a parent node in the DOM. c2||_ t| yrgrrs r0rezTree.NodeSelected.__init__rr/c.|jjSrrr=s r0rzTree.NodeSelected.controlrr/rrr r s@r0 NodeSelectedrr r/rNnamercclassesdisabledc|j|}d|_i|_d|_|j d|||_ t d|_d|_d|_ t|-||||y)azInitialise a Tree. Args: label: The label of the root node of the tree. data: The optional data to associate with the root node of the tree. name: The name of the Tree. id: The ID of the tree in the DOM. classes: The CSS classes of the tree. disabled: Whether the tree is disabled or not. rNir) rWr_r _current_idrrr _line_cache_tree_lines_cached _cursor_noderre) r>rdrYrrcrrrrs r0rez Tree.__init__su*''.  ACNN4T: (:B4.HL;? dr7XNr/c`| |j}ddlm}|dfd d||y)zAdds JSON data to a node. Args: json_data: An object decoded from JSON. node: Node to add data to. Nr)ReprHighlightercht|trQ|jtd||j D] \}}|j d}|||"yt|t rU|jtd|t|D])\}}|j d}t|||+yd|_ |r=tjtjd|d t|}ntt|}|j|y)zAdds a node to the tree. Args: name: Name of the node. node: Parent node. data: Data associated with the node. z{} z[] Fz[b]z[/b]=N) rdictrritemsrlist enumeratestrrRassemble from_markuprepr) rr?rYkeyvaluenew_noderrdadd_node highlighters r0r4zTree.add_json..add_node$s$%teD6N34"&**,3JC#xx|HS(E23D$'tc$L12$-dO:LE5#xx|HSZ59:%*! MM((3tfE):;[d=TE!d,Eu%r/r()rr-r?rPrYobjectrHr)rrich.highlighterr&)r> json_datar?r&r4r5s @@r0add_jsonz Tree.add_jsons1 <99D4%'  &8 T9%r/c|jS)z9The currently selected node, or ``None`` if no selection.)r$r=s r0 cursor_nodezTree.cursor_nodeBs   r/c2t|jdz S)zThe index of the last line.rA)rC _tree_linesr=s r0 last_linezTree.last_lineGs4##$q((r/c|t|trtj|}n|}|j d}|S)zProcess a `str` or `Text` value into a label. May be overridden in a subclass to change how labels are rendered. Args: label: Label. Returns: A Rich Text object. r)rr-rr/split)r>rdr first_lines r0rWzTree.process_labelLs< eS !))%0JJ%%'* r/ct|||j|||}||j|j<|xjdz c_|S)NrQrA)rP_new_idrrVr_)r>rbrdrYrr?s r0rzTree._add_node^sEfdllneTFS%)"   r/c|jj}|j||jr.|jr |j n |j |tzf}nd|f}tj||}|S)a>Render a label for the given node. Override this to modify how labels are rendered. Args: node: A tree node. base_style: The base style of the widget. style: The additional style for the label. Returns: A Rich Text object containing the label. r() rXcopystylizer^rICON_NODE_EXPANDED ICON_NODE TOGGLE_STYLErr.)r>r? base_stylestyle node_labelprefixtexts r0 render_labelzTree.render_labeljst[[%%' 5!   +/+;+;''\)F *%F}}VZ0 r/cP|j|tt}|jS)a.Get the width of the nodes label. The default behavior is to call `render_label` and return the cell length. This method may be overridden in a sub-class if it can be done more efficiently. Args: node: A node. Returns: Width in cells. )rPr cell_len)r>r?rds r0get_label_widthzTree.get_label_widths"!!$ J?~~r/cF|jjd|_y)zClear line cache.N)r"clearr#r=s r0_clear_line_cachezTree._clear_line_caches  "&r/cJ|jd|_|jj}|jj}|jj }t |d|j||||_|xjdz c_|j|S)zWClear all nodes under root. Returns: The `Tree` instance. rNrCrA) rVr!rrXrYrrPrDr_r)r> root_label root_data root_expandeds r0rUz Tree.clears  YY%% IINN  --    LLN  "       r/cj|j||j_||j_|S)zClear the tree and reset the root node. Args: label: The label for the root node. data: Optional data for the root node. Returns: The `Tree` instance. )rUrrdrY)r>rdrYs r0resetz Tree.resets(    r/c|j}|dn |j|_|H|j;|j|j|xrt |j|z dkDyyy)zMove the cursor to the given node, or reset cursor. Args: node: A tree node, or None to reset cursor. animate: Enable animation Nr<rAanimate) cursor_liner`r;scroll_to_nodeabs)r>r?r_previous_cursor_lines r0 move_cursorzTree.move_cursorsv $//!%24::   0 0 <     TC(8(8;O(O$PST$T  != r/c|j|k(ry |j|j}|j ||y#t$rtd|dwxYw)aMove the cursor to the given line. Args: line: The line number (negative indexes are offsets from the last line). animate: Enable scrolling animation. Raises: IndexError: If the line doesn't exist. Nz No line no. z in the treer^)r`r=r?rrd)r>rsr_r?s r0move_cursor_to_linezTree.move_cursor_to_linesm   t #  @##D)..D w/ @|D6>? ? @s ?Act|j||%|jtj|yy)zMove the cursor to the given node and select it, or reset cursor. Args: node: A tree node to move the cursor to and select, or None to reset cursor. N)rdrrrr>r?s r0 select_nodezTree.select_nodes6      d//5 6 r/cd|jtjd|jy)zHide and reset the cursor.r<N) set_reactiverr`rr=s r0unselectz Tree.unselects$ $**B/ r/cX|j}|jr|j|yy)zDWhen the node is selected, expand the node if `auto_expand` is True.N)r? auto_expand _toggle_node)r>eventr?s r0_expand_node_on_selectzTree._expand_node_on_selects)zz      d # r/cX |j|}|jS#t$rYywxYw)zGet the node for a given line. Args: line_no: A line number. Returns: A tree node, or ``None`` if there is no node at that line. Nr=r?r)r>line_norss r0get_node_at_linezTree.get_node_at_lines7 ##G,D99   s  ))c\ |j|S#t$rtd|ddwxYw)zGet a tree node by its ID. Args: node_id: The ID of the node to get. Returns: The node associated with that ID. Raises: UnknownNodeID: Raised if the `TreeNode` ID is unknown. zUnknown NodeID (z ) in treeN)rKeyErrorr2)r>node_ids r0get_node_by_idzTree.get_node_by_idsA Q##G, , Q"27)9 EFD P Qs+cHt|dt|jdz S)zPrevent cursor line from going outside of range. Args: value: The value to test. Return: A valid version of the given value. rrA)rrCr=r>r2s r0validate_cursor_linezTree.validate_cursor_lines#UAs4#3#34q899r/ct|ddS)zRestrict guide depth to reasonable range. Args: value: The value to test. Return: A valid version of the given value.  )rr{s r0validate_guide_depthzTree.validate_guide_depth$sUAr""r/c|j|xjdz c_|jj|j dy)zInvalidate caches.rAT)layoutN)rVr_rrkrr=s r0rzTree._invalidate/s:      D !r/c`|jj}|rd|vr |d|_yd|_y)Nrsr<)rLmeta hover_line)r>rprs r0_on_mouse_movezTree._on_mouse_move6s,{{ FdN"6lDO DOr/cd|_yNr<)r)r>_s r0 _on_leavezTree._on_leave=s r/cZ|j}|xjdz c_t|S)zNCreate a new node ID. Returns: A unique node ID. rA)r!r")r>rcs r0rDz Tree._new_idAs*    Abzr/cd|dk(ry |j|}|jS#t$rYywxYwrrs)r>rs tree_lines r0 _get_nodezTree._get_nodeKsA 2: "((.I>> !  s # //c |j|}|j|j|j}|j |j }t|||dS#t$rYywxYw)aiReturns the region occupied by the label of the node at line `line`. This can be used, e.g., when scrolling to that line such that the label is visible after the scroll. Args: line: A line number. Returns: The region occupied by the label, or `None` if the line is not in the tree. NrA)r=rrGrDrErSr?r)r>rsrregion_x region_widths r0_get_label_regionzTree._get_label_regionUsn ((.I--d.>.>O++INN; hlA66   sA A,+A,c|j|}||j|d|_|j|}||j|d|_yyNFT)rrru)r>previous_hover_liner previous_noder?s r0watch_hover_linezTree.watch_hover_linejs[':;  $   } -#(M ~~j)     t $DK r/cj|j|}|j|}|jd|j_|d|_|d|_||_nd|_||k(ry||j ||8|j |||k7r!|j |j |yyyr)rr;r{r$rrr)r> previous_linersrr?s r0watch_cursor_linezTree.watch_cursor_lineus}5 ~~d#    ').D   &  $&+M #  !DN $D  $D  D   $   } -     t $$!!$"6"6t"<=% r/c$|jyrg)r)r>rDs r0watch_guide_depthzTree.watch_guide_depths r/c2d|_|jyr)r`r)r>rEs r0watch_show_rootzTree.watch_show_roots r/cn|j|}|"|j||d|jddyy)zyScroll to the given line. Args: line: A line number. animate: Enable animation. NTF)r_forcecenterorigin_visiblex_axis)rscroll_to_region center_scroll)r>rsr_regions r0scroll_to_linezTree.scroll_to_linesJ''-    ! !))$ "  r/cN|j}|dk7r|j||yy)zScroll to the given node. Args: node: Node to scroll into view. animate: Animate scrolling. r<r^N)r`r)r>r?r_rss r0razTree.scroll_to_nodes,zz 2:   g  6 r/ctd||jjz |jjd}|j |y)zbRefresh (repaint) a given line in the tree. Args: line: Line number. rrAN)r scroll_offsetysizerFr)r>rsrs r0rzTree._refresh_lines9 4$"4"4"6"66 K Vr/cN|j|}||j|yyrg)rrr>rsr?s r0_refresh_node_linezTree._refresh_node_lines)~~d#     t $ r/c|jj}|jj}|j|||z}t ||D]%\}}||j vs|j|'y)z[Refresh a node and all its children. Args: node: A tree node. N)rrrheightr=r,r8r)r>r?scroll_yr visible_linesrtrss r0rzTree._refresh_nodeso %%''!!((Hv4EF &}h? ,MGTtyy ""7+ ,r/cn|j|j|jJ|jSrg)r#_buildr=s r0r=zTree._tree_liness5  " " * KKM&&222&&&r/cK|j4d{|jdddd{y7!7#1d{7swYyxYww)z#Check tree needs a rebuild on idle.N)lockr=r>rps r0_on_idlez Tree._on_idlesG99            s;A5A 9 A7AAA A A Ac tg j|j} d fd |jr g|dn%|jjD] }g|d |_|j |j |j d fd } r t Dcgc] }|| c}}n|jj}t|t |_ |jdk7rH|j|jj |_|jt k\rd|_yyycc}w)z=Builds the tree by traversing nodes, and creating tree lines.cg||}t|_|||jr(t|jD]\}}|||yyrg)rCr`rZrr[) r8r?r: child_pathrTreeLineadd_liner4liness r0r4zTree._build..add_nodes`'4JUDJ Xj$/ 0~~#,T^^#<6KD%Z56r/TcN|j|jzSrg)r?rG)rsrSrDrEs r0get_line_widthz#Tree._build..get_line_widths+"499-0E0EY1 r/r<N)r8r7r?rIr:r9rHr)rsz_TreeLine[TreeDataType]rHrJ)r6appendrrEr[r#rDrSrBrrFrrC virtual_sizer`r;r`) r>rr?rrsrFrrr4rSrDrrEs @@@@@@@r0rz Tree._buildsH/1<<yy 6. 66L 6TX 6  6 6 >> Rt $ ++ )T4( )"'&& NN ..  %@$-@AEIIOOE E 3   r !+#'#3#3#9#9 3u:-#% . " As3EcL|j|_t| |Srg)get_pseudo_class_state_pseudo_class_stater render_lines)r>croprs r0rzTree.render_liness$#'#>#>#@ w#D))r/c|jj}|j\}}|j}|j ||z|||z|Srg)rrFr rich_style _render_line)r>rrFscroll_xrrLs r0 render_linezTree.render_linesP !//(  L  u     r/c j}jj}|t|k\rt j ||S||}j dk\xrtd|jD}|||jjtd|jDf} | jvrj| } nsjdjjdk(} jdjjdk(} jdjjdk(} j!dd}|j!ddz}|j!ddz}|jdj"}|jdj$xr j&}dfd }|rj!d }n|}|t)d |i z }t+| }|j,}|}d}|jddD]x}| }|r|}| }|r|}| }|||\}}}}|j.r|n|}||jdk7r ||| |xs |j"}|xs |j$xr j&}zt|jdkDrA|||\}}}}|j0r|j-|| n|j-|| j!dd} j |k(r| j!ddz } j2|k(r| j!ddz } j5|jd|| j7}!|!j9t)d|j:j<i |j-|!t?|jAjBjD}"tGjHj|}#tK|"d|#|jLz |}"t |"x} j| <| jO||} | S)Nrc34K|]}|jywrg)ru.0r?s r0 z$Tree._render_line..(s/R /Rc34K|]}|jywrg)r_rs r0rz$Tree._render_line..0s6D$--6rrrrT)partialc&jrI|sGjd}|jrjd}n|jrjd}nd}t dj dz t fd|D}td|S) a%Get the guide strings for a given style. Args: style: A Style object. hidden: Switch to hide guides (make them invisible). Returns: Strings for space, vertical, terminator and cross. rrr)rrrrrr~c3>K|]}|d|dzdyw)rrA Nr.)r charactersrDs r0rz8Tree._render_line..get_guides..^s2$""!}ojmk&A%B!D$stuple[str, str, str, str]) show_guidesrr underline2rBrDtupler )rLhiddenr guide_linesrDr>s @r0 get_guidesz%Tree._render_line..get_guidesIs##F JJy1Ezz $ 6 2)) $ 8 44E!!T%5%5%9: #$&+$ 7EEr/rrs)r)rLrAr<rrrFr?)rLr rr9rHr)(r=rrFrCrblankranyr8r_rrr"get_component_stylescoloraget_component_rich_stylerur{ has_focusr rrrr:r`rPrFrGr?rVr+renderappconsolerBrrrRr)$r>rx1x2rK tree_linesrFrsis_hover cache_keystrip base_hidden hover_hiddenselected_hiddenbase_guide_styleguide_hover_styleguide_selected_styleryr}r line_styleguides guides_append guide_stylerr?rverticalrguide terminatorcross label_stylerdsegments pad_widths$` r0rzTree._render_lines4%%   J ;;uj1 1!}??a'RC/R /R,R    MM  $ $ 6DII6 6   (( ($$Y/E33NCIIKKqPK445IJPPRRVWWL))*ABHHJJaO  $<< =  !143P3P$d4Q4! $4d6S6S'7T7$ IIaL''Eyy|--@$..H F6!::;QR ' %fa[1 1J +F"MMM*KF !"  K$"3K)F"6K,F(2;(G%xA!%8499R=(!%{;,$6JDNN K 499~!*4[&*I'1j%99MM*KM@MM%{M;77 t7TK!#t<<%t =  1$t<<"E =  %%diimZMRRTE MM%fdiimm%<= > MM% FMM$((*:*:;rprr`r?s r0 _on_clickzTree._on_clicks99 ; ;;;##D~"6l 88He,00=D'))$/(3D$///::: ; ; ;; ; ; ; ;sWB:BB:A2B% B! B% B:B#B:!B%#B:%B7+B. ,B73B:cBt||jyrg)rnotify_style_updater)r>rs r0r zTree.notify_style_updates #% r/c|jdk(r|j|_n|xjdzc_|j|jdy)zMove the cursor up one node.r<rAFr^N)r`r>rr=s r0action_cursor_upzTree.action_cursor_upsF   r !#~~D     !  D,,errrr=s r0action_page_upzTree.action_page_upsR   r !#~~D  D::AAAEE D,,-r/cHd|_|j|jy)z'Move the cursor to the top of the tree.rNrr=s r0action_scroll_homezTree.action_scroll_homes D,,-r/c\|j|_|j|jy)z~Move the cursor to the bottom of the tree. Note: Here bottom means vertically, not branch depth. N)r>r`rr=s r0action_scroll_endzTree.action_scroll_ends$  >> D,,-r/c |j|j}|j|jdy#t$rYywxYw)z-Toggle the expanded state of the target node.r<N)r=r`ror8r)r>rss r0action_toggle_nodezTree.action_toggle_nodesI -##D$4$45D   diim ,   s: AAc|jdkry |j|j}|jd}|jtj |y#t $rYywxYw)zCause a select event for the target node. Note: If `auto_expand` is `True` use of this action on a non-leaf node will cause both an expand/collapse event to occur, as well as a selected event. rNr<)r`r=r8rrrrrs r0action_select_cursorzTree.action_select_cursorsl   a   7##D$4$45D99R=D   d//5 6    sA A+*A+cv|j}|+|j|j|jdyyy)z#Move the cursor to the parent node.NTr^)r;rbrdr>r;s r0action_cursor_parentzTree.action_cursor_parents@&&  "{'9'9'E   [//  >(F "r/c|j}|5|j(|j|jjdyyy)z-Move the cursor to the parent's next sibling.NTr^)r;rbrdrrs r0!action_cursor_parent_next_siblingz&Tree.action_cursor_parent_next_sibling sF&&  "{'9'9'E   [//<r;rs r0action_cursor_previous_siblingz#Tree.action_cursor_previous_siblingsW&&  "*;; '  !3!3T B  !14 @ #r/c|j}|Y|j}|7|j*|jj}|j|dyy|j|dyy)z]Move the cursor to the next sibling, or to the paren't sibling if there are no more siblings.NTr^)r;rrbrd)r>r;rparent_siblings r0action_cursor_next_siblingzTree.action_cursor_next_siblingss&&  "&33L#%%1%0%7%7%D%DN$$^T$B2  t < #r/c|j|jjy|jj}|j}td|Dr%|D]}|js|j !n$|D]}|js|j !|j|j|dy)zExpand or collapse all siblings. If all the siblings are collapsed then they will be expanded. Otherwise they will all be collapsed. Nc34K|]}|jywrg)r)rrs r0rz0Tree.action_toggle_expand_all..4s8eu!!8rFr^) r;rbrqallrRrrcall_after_refreshrd)r>rqr;rs r0action_toggle_expand_allzTree.action_toggle_expand_all%s    #t'7'7'>'>'F ##,,&&  8x8 8! #%%LLN # " %%%NN$ %  0 0+uMr/rg)rdrrYrr str | Nonercr-rr-rr9rHr)r8r6r?zTreeNode | NonerHrrr)rdrrHr)F) rbrrdrrYrrr9rHrI)r?rIrKr rLr rHr)r?rIrHrJrr)rdrrYrrHr )r?rr_r9rHr)rsrJrHr)r?rrHr)rpzNodeSelected[TreeDataType]rHr)rtrJrHr)rxr"rHrI)r2rJrHrJ)rpzevents.MouseMoverHr)rz events.LeaverHrr)rsrJrHr)rsrJrHz Region | None)rrJrrJrHr)rrJrsrJrHr)rDrJrHr)rEr9rHr)T)rsrJr_r9rHr)r?rIr_r9rHr)r?rIrHr)rHzlist[_TreeLine[TreeDataType]])rpz events.IdlerHr)rrrHz list[Strip])rrJrHr) rrJrrJrrJrKr rHr)rpz events.ResizerHr)rpz events.ClickrHr)Yr*r+r,r-rIrHrrrKr DEFAULT_CSSrrErrr`rrDrnrrrr$rrrrrrer9rLr;r>rWrrPrSrVrUr\rdrfrirlrrqruryr|rrrrrDrrrrrrrrarrrr=rrrrrrror r r rrrrrrrr r"r$r'r,r r s@r0rrs@I75  o/AN  ( %    % (    ! $  (?u= .0Hu  k;U; }5A5-H)8-) BKHI$RJXb-K74.K-15)K)d)K8JMJ   3E /*" 12G"""w017"""'"34g"""w017"(%) O  " O O" O  O  O O O  OD,&\!!)). -  "    *8=FK 6 ' .  DI1<@ "0$7  $$  Q" : #"!"7* ><&=A 7* 759 7  7% ,''  -&^*  }~ ;==... .-7$? L A =Nr/r) can_focus)Er- __future__r dataclassesrtypingrrrrr r r rich.reprrich rich.styler r rich.textrrtextualrr textual._immutable_sequence_viewr textual._looprtextual._segment_toolsrtextual.bindingrr textual.cachertextual.geometryrrrtextual.messagertextual.reactiverrtextual.scroll_viewr textual.striprtyping_extensionsr r!rJr"r#r$r%rK from_metarJ Exceptionr(r2r4r6rNr0autorPrr.r/r0rFs:"!UUU($B#+0"00#**1 3 P~& R/0 4 i3u$/0 lilhIhP9P  % 81%&>?1f-w|$f-f-R zN7< *zNr/