Current File : //snap/core/17212/usr/lib/python3/dist-packages/urwid/__pycache__/treetools.cpython-35.pyc
p�6V�< � @ s� d Z d d l Z d d l m Z Gd d � d e � Z Gd d � d e j � Z Gd d � d e � Z Gd
d � d e � Z
Gd d
� d
e j � Z Gd d � d e j
� Z d S)z�
Urwid tree view
Features:
- custom selectable widgets for trees
- custom list walker for displaying widgets in a tree fashion
� N)�SelectableIconc @ s e Z d Z d S)�TreeWidgetErrorN)�__name__�
__module__� __qualname__� r r �1/usr/lib/python3/dist-packages/urwid/treetools.pyr $ s r c @ s� e Z d Z d Z d Z e d d � Z e d d � Z d d � Z d d � Z d
d � Z
d d
� Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d! � Z d" d# � Z d$ S)%�
TreeWidgetz9A widget representing something in a nested tree display.� �+r �-c C sN | | _ d | _ t | d � | _ d | _ | j � } | j j | � d S)N�get_first_childT)�_node� _innerwidget�hasattr�is_leaf�expanded�get_indented_widget�_TreeWidget__super�__init__)�self�node�widgetr r r r . s zTreeWidget.__init__c C s | j S)zS
Allow selection of non-leaf nodes so children may be (un)expanded
)r )r r r r �
selectable6 s zTreeWidget.selectablec C sq | j � } | j sL t j d d | j | j g | j f | g d d �} | j � } t j | d d d | �S) NZfixed� Z dividechars�width�relative�d �left)r r ) �get_inner_widgetr �urwidZColumns�unexpanded_icon�
expanded_iconr �get_indent_colsZPadding)r r � indent_colsr r r r < s z TreeWidget.get_indented_widgetc C s* | j | j g | j | j j j d <d S)z-Update display widget text for parent widgetsr N)r! r" r �_wZ base_widgetZ widget_list)r r r r �update_expanded_iconF s z TreeWidget.update_expanded_iconc C s | j | j � j � S)N)r$ �get_node� get_depth)r r r r r# L s zTreeWidget.get_indent_colsc C s% | j d k r | j � | _ | j S)N)r �load_inner_widget)r r r r r O s zTreeWidget.get_inner_widgetc C s t j | j � � S)N)r ZText�get_display_text)r r r r r) T s z TreeWidget.load_inner_widgetc C s | j S)N)r )r r r r r' W s zTreeWidget.get_nodec C s* | j � j � d t | j � j � � S)Nz: )r' �get_key�str� get_value)r r r r r* Z s zTreeWidget.get_display_textc C s� | j � } | d k r | S| j � } | j � } | j � } xV | d k r� | d k r� | j � } | j � } | d 8} | | j � k sC t � qC W| d k r� d S| j � Sd S)z5Return the next TreeWidget depth first from this one.Nr r )� first_childr' � next_siblingr( �
get_parent�AssertionError�
get_widget)r Z
firstchild�thisnodeZnextnode�depthr r r � next_inorder^ s
zTreeWidget.next_inorderc C s� | j } | j � } | d k rP | j � } | j � } | d k rI | S| SnJ | j � } | d k rx | d k rx d S| d k r� | j � } | j � Sd S)z9Return the previous TreeWidget depth first from this one.Nr )r � prev_siblingr2 �
last_childr( r0 )r r3 ZprevnodeZ
prevwidget� lastchildr4 r r r � prev_inorderu s zTreeWidget.prev_inorderc C s{ | j r
| S| d k r/ d | _ | j � nH | d k rQ d | _ | j � n& | j j � rs | j j | | � S| Sd S)z2Handle expand & collapse requests (non-leaf nodes)r �rightTr FN)r zright)r r r& r% r r �keypress)r �size�keyr r r r; � s
zTreeWidget.keypressc C sb | j s! | d k s! | d k r% d S| d k r^ | | j � k r^ | j | _ | j � d Sd S)Nz mouse pressr Fr T)r r# r r& )r r< ZeventZbutton�col�row�focusr r r � mouse_event� s !
zTreeWidget.mouse_eventc C sG | j s | j r d S| j j � r? | j j � } | j � Sd Sd S)z Return first child if expanded.N)r r r � has_childrenr
r2 )r Z firstnoder r r r. � s
zTreeWidget.first_childc C sf | j s | j r d S| j j � r> | j j � j � } n d S| j � } | d k r^ | S| Sd S)z Return last child if expanded.N)r r r rB �get_last_childr2 r7 )r r8 Zlastdescendantr r r r7 � s zTreeWidget.last_childN)r r r �__doc__r$ r r! r" r r r r& r# r r) r' r* r5 r9 r; rA r. r7 r r r r r ( s&
r c @ s� e Z d Z d Z d d d d d � Z d d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d! � Z d S)"�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
Nc C s1 | | _ | | _ | | _ | | _ d | _ d S)N)�_key�_parent�_value�_depth�_widget)r �value�parentr= r4 r r r r � s
zTreeNode.__init__Fc C s1 | j d k s | d k r* | j � | _ | j S)z! Return the widget for this node.NT)rJ � load_widget)r �reloadr r r r2 � s zTreeNode.get_widgetc C s
t | � S)N)r )r r r r rM � s zTreeNode.load_widgetc C sV | j d k r* | j d k r* d | _ n% | j d k rO | j j � d | _ | j S)Nr r )rI rG r( )r r r r r( � s
zTreeNode.get_depthc C s? | j � d k r d S| j � } | j � } | j | � Sd S)Nr )r( r+ r0 �get_child_index)r r= rL r r r � get_index� s
zTreeNode.get_indexc C s | j S)N)rF )r r r r r+ � s zTreeNode.get_keyc C s
| | _ d S)N)rF )r r= r r r �set_key� s zTreeNode.set_keyc C s | j � j | j | � d S)N)r0 �change_child_keyrF )r r= r r r �
change_key� s zTreeNode.change_keyc C s7 | j d k r0 | j � d k r0 | j � | _ | j S)Nr )rG r( � load_parent)r r r r r0 � s !zTreeNode.get_parentc C s t d � � d S)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)z(virtual function. Implement in subclassN)r )r r r r rT � s zTreeNode.load_parentc C s | j S)N)rH )r r r r r- � s zTreeNode.get_valuec C s | j � d k S)Nr )r( )r r r r �is_root s zTreeNode.is_rootc C s3 | j � d k r+ | j � j | j � � Sd Sd S)Nr )r( r0 �
next_childr+ )r r r r r/ s zTreeNode.next_siblingc C s3 | j � d k r+ | j � j | j � � Sd Sd S)Nr )r( r0 �
prev_childr+ )r r r r r6
s zTreeNode.prev_siblingc C s/ | } x" | j � d k r* | j � } q W| S)N)r0 )r �rootr r r �get_root s zTreeNode.get_root)r r r rD r r2 rM r( rP r+ rQ rS r0 rT r- rU r/ r6 rY r r r r rE � s rE c @ s� e Z d Z d Z d d d d d � Z d d d � Z d d � Z d
d � Z d d d
� Z d d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d S) �
ParentNodez"Maintain sort order for TreeNodes.Nc C s8 t j | | d | d | d | �d | _ i | _ d S)NrL r= r4 )rE r � _child_keys� _children)r rK rL r= r4 r r r r s " zParentNode.__init__Fc C s1 | j d k s | d k r* | j � | _ | j S)z,Return a possibly ordered list of child keysNT)r[ �load_child_keys)r rN r r r �get_child_keys s zParentNode.get_child_keysc C s t d � � d S)zPProvide ParentNode with an ordered list of child keys (virtual
function)z(virtual function. Implement in subclassN)r )r r r r r] % s zParentNode.load_child_keysc C s | j | � } | j � S)z8Return the widget for a given key. Create if necessary.)�get_child_noder2 )r r= Zchildr r r �get_child_widget* s zParentNode.get_child_widgetc C s< | | j k s | d k r1 | j | � | j | <| j | S)z<Return the child node for a given key. Create if necessary.T)r\ �load_child_node)r r= rN r r r r_ 0 s zParentNode.get_child_nodec C s t d � � d S)z6Load the child node for a given key (virtual function)z(virtual function. Implement in subclassN)r )r r= r r r ra 6 s zParentNode.load_child_nodec C s | | j | <d S)z^Set the child node for a given key. Useful for bottom-up, lazy
population of a tree..N)r\ )r r= r r r r �set_child_node: s zParentNode.set_child_nodec C sP | | j k r t d | � � | j j | � | j | <| j | j | � d S)Nz%s is already in use)r\ r �poprQ )r ZoldkeyZnewkeyr r r rR ? s zParentNode.change_child_keyc C se y | j � j | � SWnG t k
r` d d } t | | | j � t | j � � f � � Yn Xd S)Nz#Can't find key %s in ParentNode %s
zParentNode items: %s)r^ �index�
ValueErrorr r+ r, )r r= Z errorstringr r r rO E s
zParentNode.get_child_indexc C s` | j | � } | d k r d S| d 7} | j � } | t | � k rX | j | | � Sd Sd S)z=Return the next child node in index order from the given key.Nr )rO r^ �lenr_ )r r= rd �
child_keysr r r rV N s
zParentNode.next_childc C sZ | j | � } | d k r d S| j � } | d 8} | d k rR | j | | � Sd Sd S)zAReturn the previous child node in index order from the given key.Nr r )rO r^ r_ )r r= rd rg r r r rW ^ s
zParentNode.prev_childc C s | j � } | j | d � S)z+Return the first TreeNode in the directory.r )r^ r_ )r rg r r r r
m s zParentNode.get_first_childc C s | j � } | j | d � S)z*Return the last TreeNode in the directory.r ���)r^ r_ )r rg r r r rC r s zParentNode.get_last_childc C s t | j � � d k S)z!Does this node have any children?r )rf r^ )r r r r rB w s zParentNode.has_children)r r r rD r r^ r] r` r_ ra rb rR rO rV rW r
rC rB r r r r rZ s rZ c @ sR e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d S)
�
TreeWalkerzTListWalker-compatible class for displaying TreeWidgets
positions are TreeNodes.c C s
| | _ d S)z,start_from: TreeNode with the initial focus.N)r@ )r �
start_fromr r r r � s zTreeWalker.__init__c C s | j j � } | | j f S)N)r@ r2 )r r r r r � get_focus� s zTreeWalker.get_focusc C s | | _ | j � d S)N)r@ Z _modified)r r@ r r r � set_focus� s zTreeWalker.set_focusc C s< | j � } | j � } | d k r( d S| | j � f Sd S)N)NN)r2 r5 r' )r rj r �targetr r r �get_next� s
zTreeWalker.get_nextc C s< | j � } | j � } | d k r( d S| | j � f Sd S)N)NN)r2 r9 r' )r rj r rm r r r �get_prev� s
zTreeWalker.get_prevN) r r r rD r rk rl rn ro r r r r ri | s ri c @ s^ e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d S)� TreeListBoxzPA ListBox with special handling for navigation and
collapsing of TreeWidgetsc C s% | j j | | � } | j | | � S)N)Z_TreeListBox__superr; �unhandled_input)r r< r= r r r r; � s zTreeListBox.keypressc C sx | d k r | j | � nX | d k r8 | j | � n< | d k rT | j | � n | d k rp | j | � n | Sd S)z Handle macro-navigation keysr r �home�endN)�move_focus_to_parent�collapse_focus_parent�
focus_home� focus_end)r r< �inputr r r rq � s zTreeListBox.unhandled_inputc C sW | j j � \ } } | j | � | j j � \ } } | | k rS | j | d � d S)zCollapse parent directory.r N)�bodyrk rt r; )r r< r �posZpwidgetZpposr r r ru � s
z!TreeListBox.collapse_focus_parentc C s� | j j � \ } } | j � } | d k r1 d S| j | � \ } } } | \ } } }
} } | \ }
} xD | D]< \ } } } | | 8} | | k rq | j | | | � d Sqq W| j | | j � � d S)z(Move focus to parent of widget in focus.N)ry rk r0 Zcalculate_visible� change_focus)r r< r rz � parentposZmiddle�topZbottom�
row_offsetZ focus_widgetZ focus_posZ
focus_rowsZcursorZtrim_topZ
fill_aboveZrowsr r r rt � s
z TreeListBox.move_focus_to_parentc C s5 | j j � \ } } | j � } | j | | � d S)zMove focus to very top.N)ry rk rY r{ )r r< r rz �rootnoder r r rv � s zTreeListBox.focus_homec
C sl | \ } } | j j � \ } } | j � } | j � } | j � } | j � } | j | | | d � d S)zMove focus to far bottom.r N)ry rk rY r2 r7 r' r{ )
r r< ZmaxrowZmaxcolr rz r Z
rootwidgetZ
lastwidgetZlastnoder r r rw � s zTreeListBox.focus_endN)
r r r rD r; rq ru rt rv rw r r r r rp � s
rp )rD r Z
urwid.wimpr � RuntimeErrorr Z
WidgetWrapr �objectrE rZ Z
ListWalkerri ZListBoxrp r r r r �<module> s �Ue"
Mini Shell