Current File : //snap/core20/2318/lib/python3/dist-packages/urwid/__pycache__/treetools.cpython-38.pyc
U
��dZ�< � @ s� d dl mZmZ d dlZd dlmZ G dd� de�ZG dd� dej�Z G dd � d e
�Z G d
d � d e �Z G d d
� d
ej
�ZG dd� dej�ZdS )� )�division�print_functionN)�SelectableIconc @ s e Zd ZdS )�TreeWidgetErrorN)�__name__�
__module__� __qualname__� r r �1/usr/lib/python3/dist-packages/urwid/treetools.pyr & s r c @ s� e Zd ZdZdZedd�Zedd�Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$S )%�
TreeWidgetz9A widget representing something in a nested tree display.� �+r �-c C s8 || _ d | _t|d� | _d| _| �� }| 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 0 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
�
selectable8 s zTreeWidget.selectablec C sL | � � }| js4tjdd| j| jg| j f|gdd�}| �� }tj|d|d�S )NZfixed� )Z dividechars)�relative�d )�width�left) �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 | jg| j | jjjd< dS )z-Update display widget text for parent widgetsr N)r$ r% r �_wZ base_widgetZ widget_listr r r r
�update_expanded_iconH s
��z TreeWidget.update_expanded_iconc C s | j | �� �� S �N)r' �get_node� get_depthr r r r
r& N s zTreeWidget.get_indent_colsc C s | j d kr| �� | _ | j S r* )r �load_inner_widgetr r r r
r" Q s
zTreeWidget.get_inner_widgetc C s t �| �� �S r* )r# ZText�get_display_textr r r r
r- V s z TreeWidget.load_inner_widgetc C s | j S r* )r r r r r
r+ Y s zTreeWidget.get_nodec C s | � � �� d t| � � �� � S )Nz: )r+ �get_key�str� get_valuer r r r
r. \ s �zTreeWidget.get_display_textc C s~ | � � }|dk r|S | �� }|�� }|�� }|dkrf|dkrf|�� }|�� }|d8 }||�� ks,t�q,|dkrrdS |�� S dS )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 sp | j }|�� }|dk r8|�� }|�� }|dkr2|S |S n4|�� }|dkrT|dkrTdS |dkrd|�� }|�� S dS )z9Return the previous TreeWidget depth first from this one.Nr )r � prev_siblingr6 �
last_childr, r4 )r r7 ZprevnodeZ
prevwidget� lastchildr8 r r r
� prev_inorderw s zTreeWidget.prev_inorderc C sZ | j r
|S |dkr"d| _| �� n4|dkr:d| _| �� n | j�� rR| j�||�S |S dS )z2Handle expand & collapse requests (non-leaf nodes))r
�rightTr FN)r r r) r( r r �keypress�r �size�keyr r r
r? � s
zTreeWidget.keypressc C sH | j s|dks|dkrdS |dkrD|| �� krD| j | _| �� dS dS )Nz mouse pressr Fr T)r r& r r) )r rA ZeventZbutton�col�row�focusr r r
� mouse_event� s
zTreeWidget.mouse_eventc C s4 | j s | jsdS | j�� r,| j�� }|�� S dS dS )z Return first child if expanded.N)r r r � has_childrenr r6 )r Z firstnoder r r
r2 � s
zTreeWidget.first_childc C sJ | j s | jsdS | j�� r*| j�� �� }ndS |�� }|dkrB|S |S dS )z Return last child if expanded.N)r r r rG �get_last_childr6 r; )r r< Zlastdescendantr r r
r; � s
zTreeWidget.last_childN)r r r �__doc__r' r r$ r% r r r r) r& r" r- r+ r. r9 r= r? rF r2 r; r r r r
r * s&
r c @ s� e Zd ZdZd"dd�Zd#dd�Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � 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
Nc C s" || _ || _|| _|| _d | _d S r* )�_key�_parent�_value�_depth�_widget�r �value�parentrB r8 r r r
r � s
zTreeNode.__init__Fc C s" | j dks|dkr | �� | _ | j S )z! Return the widget for this node.NT)rO � load_widget�r �reloadr r r
r6 � s
zTreeNode.get_widgetc C s t | �S r* )r r r r r
rS � s zTreeNode.load_widgetc C s< | j d kr | jd kr d| _ n| j d kr6| j�� d | _ | j S )Nr r )rN rL r, r r r r
r, � s
zTreeNode.get_depthc C s. | � � dkrd S | �� }| �� }|�|�S d S �Nr )r, r/ r4 �get_child_index)r rB rR r r r
� get_index� s
zTreeNode.get_indexc C s | j S r* �rK r r r r
r/ � s zTreeNode.get_keyc C s
|| _ d S r* rY �r rB r r r
�set_key� s zTreeNode.set_keyc C s | � � �| j|� d S r* )r4 �change_child_keyrK rZ r r r
�
change_key� s zTreeNode.change_keyc C s& | j d kr | �� dkr | �� | _ | j S rV )rL r, � load_parentr r r r
r4 � s
zTreeNode.get_parentc C s t d��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�r r r r r
r^ � s zTreeNode.load_parentc C s | j S r* )rM r r r r
r1 s zTreeNode.get_valuec C s | � � dkS rV )r, r r r r
�is_root s zTreeNode.is_rootc C s& | � � dkr | �� �| �� �S d S d S rV )r, r4 �
next_childr/ r r r r
r3 s zTreeNode.next_siblingc C s& | � � dkr | �� �| �� �S d S d S rV )r, r4 �
prev_childr/ r r r r
r: s zTreeNode.prev_siblingc C s | }|� � d k r|� � }q|S r* )r4 )r �rootr r r
�get_root s
zTreeNode.get_root)NNN)F)r r r rI r r6 rS r, rX r/ r[ r] r4 r^ r1 ra r3 r: re r r r r
rJ � s
rJ c @ s� e Zd ZdZd dd�Zd!dd�Zdd � Zd
d � Zd"d d
�Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � ZdS )#�
ParentNodez"Maintain sort order for TreeNodes.Nc C s$ t j| ||||d� d | _i | _d S )N)rR rB r8 )rJ r � _child_keys� _childrenrP r r r
r s zParentNode.__init__Fc C s" | j dks|dkr | �� | _ | j S )z,Return a possibly ordered list of child keysNT)rg �load_child_keysrT r r r
�get_child_keys! s
zParentNode.get_child_keysc C s t d��dS )zPProvide ParentNode with an ordered list of child keys (virtual
function)r_ Nr` r r r r
ri ' s zParentNode.load_child_keysc C s | � |�}|�� S )z8Return the widget for a given key. Create if necessary.)�get_child_noder6 )r rB Zchildr r r
�get_child_widget, s
zParentNode.get_child_widgetc C s, || j ks|dkr"| �|�| j |< | j | S )z<Return the child node for a given key. Create if necessary.T)rh �load_child_node)r rB rU r r r
rk 2 s zParentNode.get_child_nodec C s t d��dS )z6Load the child node for a given key (virtual function)r_ Nr` rZ r r r
rm 8 s zParentNode.load_child_nodec C s || j |< dS )z]Set the child node for a given key. Useful for bottom-up, lazy
population of a tree.N)rh )r rB r r r r
�set_child_node< s zParentNode.set_child_nodec C s< || j krtd| ��| j �|�| j |< | j | �|� d S )Nz%s is already in use)rh r �popr[ )r ZoldkeyZnewkeyr r r
r\ A s
zParentNode.change_child_keyc
C sL z| � � �|�W S tk
rF d}t||| �� t| � � �f ��Y nX d S )Nz7Can't find key %s in ParentNode %s
ParentNode items: %s)rj �index�
ValueErrorr r/ r0 )r rB Z errorstringr r r
rW G s
�zParentNode.get_child_indexc C sH | � |�}|dkrdS |d7 }| �� }|t|�k r@| �|| �S dS dS )z=Return the next child node in index order from the given key.Nr )rW rj �lenrk �r rB rp �
child_keysr r r
rb P s
zParentNode.next_childc C sD | � |�}|dkrdS | �� }|d8 }|dkr<| �|| �S dS dS )zAReturn the previous child node in index order from the given key.Nr r )rW rj rk rs r r r
rc ` s
zParentNode.prev_childc C s | � � }| �|d �S )z+Return the first TreeNode in the directory.r �rj rk �r rt r r r
r o s zParentNode.get_first_childc C s | � � }| �|d �S )z*Return the last TreeNode in the directory.���ru rv r r r
rH t s zParentNode.get_last_childc C s t | �� �dkS )z!Does this node have any children?r )rr rj r r r r
rG y s zParentNode.has_children)NNN)F)F)r r r rI r rj ri rl rk rm rn r\ rW rb rc r rH rG r r r r
rf s
rf c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�
TreeWalkerzTListWalker-compatible class for displaying TreeWidgets
positions are TreeNodes.c C s
|| _ dS )z,start_from: TreeNode with the initial focus.N)rE )r �
start_fromr r r
r � s zTreeWalker.__init__c C s | j �� }|| j fS r* )rE r6 )r r r r r
� get_focus� s
zTreeWalker.get_focusc C s || _ | �� d S r* )rE Z _modified)r rE r r r
� set_focus� s zTreeWalker.set_focusc C s, |� � }|�� }|d kr dS ||�� fS d S �N)NN)r6 r9 r+ �r ry r �targetr r r
�get_next� s
zTreeWalker.get_nextc C s, |� � }|�� }|d kr dS ||�� fS d S r| )r6 r= r+ r} r r r
�get_prev� s
zTreeWalker.get_prevN) r r r rI r rz r{ r r� r r r r
rx ~ s rx c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� TreeListBoxzPA ListBox with special handling for navigation and
collapsing of TreeWidgetsc C s | j �||�}| �||�S r* )Z_TreeListBox__superr? �unhandled_inputr@ r r r
r? � s zTreeListBox.keypressc C sX |dkr| � |� n@|dkr(| �|� n,|dkr<| �|� n|dkrP| �|� n|S dS )z Handle macro-navigation keysr! r �home�endN)�move_focus_to_parent�collapse_focus_parent�
focus_home� focus_end)r rA �inputr r r
r� � s zTreeListBox.unhandled_inputc C s> | j �� \}}| �|� | j �� \}}||kr:| �|d� dS )zCollapse parent directory.r N)�bodyrz r� r? )r rA r �posZpwidgetZpposr r r
r� � s
z!TreeListBox.collapse_focus_parentc C s� | j �� \}}|�� }|dkr"dS | �|�\}}}|\}} }
} } |\}
}|D ].\}}}||8 }||krL| �|||� dS qL| �||�� � dS )z(Move focus to parent of widget in focus.N)r� rz r4 Zcalculate_visible� change_focus)r rA r r� � parentposZmiddle�topZbottom�
row_offsetZ focus_widgetZ focus_posZ
focus_rowsZcursorZtrim_topZ
fill_aboveZrowsr r r
r� � s z TreeListBox.move_focus_to_parentc C s& | j �� \}}|�� }| �||� dS )zMove focus to very top.N)r� rz re r� )r rA r r� �rootnoder r r
r� � s zTreeListBox.focus_homec
C sL |\}}| j �� \}}|�� }|�� }|�� }|�� } | �|| |d � dS )zMove focus to far bottom.r N)r� rz re r6 r; r+ r� )
r rA ZmaxrowZmaxcolr r� r� Z
rootwidgetZ
lastwidgetZlastnoder r r
r� � s zTreeListBox.focus_endN)
r r r rI r? r� r� r� r� r� r r r r
r� � s
r� )Z
__future__r r r# Z
urwid.wimpr � RuntimeErrorr Z
WidgetWrapr �objectrJ rf Z
ListWalkerrx ZListBoxr� r r r r
�<module> s Ue"
Mini Shell