U ºñdZú<ã@sŽddlmZmZddlZddlmZGdd„deƒZGdd„dejƒZ Gdd „d e ƒZ Gd d „d e ƒZ Gd d „d ej ƒZGdd„dejƒZdS)é)ÚdivisionÚprint_functionN)ÚSelectableIconc@s eZdZdS)ÚTreeWidgetErrorN)Ú__name__Ú __module__Ú __qualname__©r r ú1/usr/lib/python3/dist-packages/urwid/treetools.pyr&src@s eZdZdZdZeddƒZeddƒZdd„Zdd „Z d d „Z d d „Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$S)%Ú TreeWidgetz9A widget representing something in a nested tree display.éú+rú-cCs8||_d|_t|dƒ |_d|_| ¡}|j |¡dS)NÚget_first_childT)Ú_nodeÚ _innerwidgetÚhasattrÚis_leafÚexpandedÚget_indented_widgetÚ_TreeWidget__superÚ__init__)ÚselfÚnodeÚwidgetr r r r0s zTreeWidget.__init__cCs|j S)zS Allow selection of non-leaf nodes so children may be (un)expanded )r©rr r r Ú selectable8szTreeWidget.selectablecCsL| ¡}|js4tjdd|j|jg|jf|gdd}| ¡}tj|d|dS)NZfixedé)Z dividechars)Úrelativeéd)ÚwidthÚleft) Úget_inner_widgetrÚurwidZColumnsÚunexpanded_iconÚ expanded_iconrÚget_indent_colsZPadding)rrÚ indent_colsr r r r>sÿþþÿzTreeWidget.get_indented_widgetcCs |j|jg|j|jjjd<dS)z-Update display widget text for parent widgetsrN)r$r%rÚ_wZ base_widgetZ widget_listrr r r Úupdate_expanded_iconHs ÿÿzTreeWidget.update_expanded_iconcCs|j| ¡ ¡S©N)r'Úget_nodeÚ get_depthrr r r r&NszTreeWidget.get_indent_colscCs|jdkr| ¡|_|jSr*)rÚload_inner_widgetrr r r r"Qs  zTreeWidget.get_inner_widgetcCst | ¡¡Sr*)r#ZTextÚget_display_textrr r r r-VszTreeWidget.load_inner_widgetcCs|jSr*)rrr r r r+YszTreeWidget.get_nodecCs | ¡ ¡dt| ¡ ¡ƒS)Nz: )r+Úget_keyÚstrÚ get_valuerr r r r.\sÿzTreeWidget.get_display_textcCs~| ¡}|dk r|S| ¡}| ¡}| ¡}|dkrf|dkrf| ¡}| ¡}|d8}|| ¡ks,t‚q,|dkrrdS| ¡SdS)z5Return the next TreeWidget depth first from this one.Nrr)Ú first_childr+Ú next_siblingr,Ú get_parentÚAssertionErrorÚ get_widget)rZ firstchildÚthisnodeZnextnodeÚdepthr r r Ú next_inorder`szTreeWidget.next_inordercCsp|j}| ¡}|dk r8| ¡}| ¡}|dkr2|S|Sn4| ¡}|dkrT|dkrTdS|dkrd| ¡}| ¡SdS)z9Return the previous TreeWidget depth first from this one.Nr)rÚ prev_siblingr6Ú last_childr,r4)rr7ZprevnodeZ prevwidgetÚ lastchildr8r r r Ú prev_inorderwszTreeWidget.prev_inordercCsZ|jr |S|dkr"d|_| ¡n4|dkr:d|_| ¡n|j ¡rR|j ||¡S|SdS)z2Handle expand & collapse requests (non-leaf nodes))r ÚrightTrFN)rrr)r(rrÚkeypress©rÚsizeÚkeyr r r r?s   zTreeWidget.keypresscCsH|js|dks|dkrdS|dkrD|| ¡krD|j |_| ¡dSdS)Nz mouse pressrFrT)rr&rr))rrAZeventZbuttonÚcolÚrowÚfocusr r r Ú mouse_events zTreeWidget.mouse_eventcCs4|js |jsdS|j ¡r,|j ¡}| ¡SdSdS)zReturn first child if expanded.N)rrrÚ has_childrenrr6)rZ firstnoder r r r2¨s    zTreeWidget.first_childcCsJ|js |jsdS|j ¡r*|j ¡ ¡}ndS| ¡}|dkrB|S|SdS)zReturn last child if expanded.N)rrrrGÚget_last_childr6r;)rr<Zlastdescendantr r r r;³s  zTreeWidget.last_childN)rrrÚ__doc__r'rr$r%rrrr)r&r"r-r+r.r9r=r?rFr2r;r r r r r *s&     r c@sŒeZdZdZd"dd„Zd#dd„Zdd „Zd d „Zd d „Zdd„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zd d!„ZdS)$ÚTreeNodea: Store tree contents and cache TreeWidget objects. A TreeNode consists of the following elements: * key: accessor token for parent nodes * value: subclass-specific data * parent: a TreeNode which contains a pointer back to this object * widget: The widget used to render the object NcCs"||_||_||_||_d|_dSr*)Ú_keyÚ_parentÚ_valueÚ_depthÚ_widget©rÚvalueÚparentrBr8r r r rÍs zTreeNode.__init__FcCs"|jdks|dkr| ¡|_|jS)z! Return the widget for this node.NT)rOÚ load_widget©rÚreloadr r r r6Ôs zTreeNode.get_widgetcCst|ƒSr*)r rr r r rSÚszTreeNode.load_widgetcCs<|jdkr|jdkrd|_n|jdkr6|j ¡d|_|jS)Nrr)rNrLr,rr r r r,Ýs  zTreeNode.get_depthcCs.| ¡dkrdS| ¡}| ¡}| |¡SdS©Nr)r,r/r4Úget_child_index)rrBrRr r r Ú get_indexäs  zTreeNode.get_indexcCs|jSr*©rKrr r r r/ìszTreeNode.get_keycCs ||_dSr*rY©rrBr r r Úset_keyïszTreeNode.set_keycCs| ¡ |j|¡dSr*)r4Úchange_child_keyrKrZr r r Ú change_keyòszTreeNode.change_keycCs&|jdkr | ¡dkr | ¡|_|jSrV)rLr,Ú load_parentrr r r r4õs zTreeNode.get_parentcCs tdƒ‚dS)z¦Provide TreeNode with a parent for the current node. This function is only required if the tree was instantiated from a child node (virtual function)ú(virtual function. Implement in subclassN©rrr r r r^úszTreeNode.load_parentcCs|jSr*)rMrr r r r1szTreeNode.get_valuecCs | ¡dkSrV)r,rr r r Úis_rootszTreeNode.is_rootcCs&| ¡dkr| ¡ | ¡¡SdSdSrV)r,r4Ú next_childr/rr r r r3s zTreeNode.next_siblingcCs&| ¡dkr| ¡ | ¡¡SdSdSrV)r,r4Ú prev_childr/rr r r r: s zTreeNode.prev_siblingcCs|}| ¡dk r| ¡}q|Sr*)r4)rÚrootr r r Úget_roots  zTreeNode.get_root)NNN)F)rrrrIrr6rSr,rXr/r[r]r4r^r1rar3r:rer r r r rJÄs   rJc@s†eZdZdZd dd„Zd!dd„Zdd „Zd d „Zd"d d „Zdd„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„ZdS)#Ú ParentNodez"Maintain sort order for TreeNodes.NcCs$tj|||||dd|_i|_dS)N)rRrBr8)rJrÚ _child_keysÚ _childrenrPr r r rszParentNode.__init__FcCs"|jdks|dkr| ¡|_|jS)z,Return a possibly ordered list of child keysNT)rgÚload_child_keysrTr r r Úget_child_keys!s zParentNode.get_child_keyscCs tdƒ‚dS)zPProvide ParentNode with an ordered list of child keys (virtual function)r_Nr`rr r r ri'szParentNode.load_child_keyscCs| |¡}| ¡S)z8Return the widget for a given key. Create if necessary.)Úget_child_noder6)rrBZchildr r r Úget_child_widget,s zParentNode.get_child_widgetcCs,||jks|dkr"| |¡|j|<|j|S)z|j ¡\}}| |¡|j ¡\}}||kr:| |d¡dS)zCollapse parent directory.rN)Úbodyrzr…r?)rrArÚposZpwidgetZpposr r r r†µs  z!TreeListBox.collapse_focus_parentcCs|j ¡\}}| ¡}|dkr"dS| |¡\}}}|\}} } } } |\} }|D].\}}}||8}||krL| |||¡dSqL| || ¡¡dS)z(Move focus to parent of widget in focus.N)rŠrzr4Zcalculate_visibleÚ change_focus)rrArr‹Ú parentposZmiddleÚtopZbottomÚ row_offsetZ focus_widgetZ focus_posZ focus_rowsZcursorZtrim_topZ fill_aboveZrowsr r r r…¿sz TreeListBox.move_focus_to_parentcCs&|j ¡\}}| ¡}| ||¡dS)zMove focus to very top.N)rŠrzrerŒ)rrArr‹Úrootnoder r r r‡ÖszTreeListBox.focus_homec CsL|\}}|j ¡\}}| ¡}| ¡}| ¡}| ¡} | || |d¡dS)zMove focus to far bottom.rN)rŠrzrer6r;r+rŒ) rrAZmaxrowZmaxcolrr‹rZ rootwidgetZ lastwidgetZlastnoder r r rˆÝszTreeListBox.focus_endN) rrrrIr?r‚r†r…r‡rˆr r r r r s  r)Z __future__rrr#Z urwid.wimprÚ RuntimeErrorrZ WidgetWrapr ÚobjectrJrfZ ListWalkerrxZListBoxrr r r r Ús  Ue"