B
� f�? � @ s0 d dl Z d dlT d dlmZmZ d dlmZ d dlmZ dZ ye j
� e j
� e
�e �ZW n ek
rn e ZY nX e j
�e�r�eZ ne j
�e �s�ede f ��e fdd �Zd d
d �ZG d d
� d
�ZG dd� d�ZG dd� de�ZG dd� d�Zdd� Zedk�r,d dlmZ edddd� d d l m Z e e� dS ) � N)�*)�Frame� Scrollbar)�idleConf)�
zoomheightZIconsz can't find icon directory (%r)c C s� t � }ddl}|�tj�| d��}|�� g }d }}x�|D ]�}tj�tj�|��d }t||d�} |� | � t
|| ddd�}
|
j ||d� t
||d �}
|
j |d |d� |d }|d
kr<|d }d}q 1 if one scrolls fast. Flicking the wheel
generates up to maybe 20 events with n up to 10 or more 1.
Macs use wheel down (delta = 1*n) to scroll up, so positive
delta means to scroll up on both systems.
X-11 sends Control-Button-4,5 events instead.
The widget parameter is needed so browser label bindings can pass
the underlying canvas.
This function depends on widget.yview to not be overridden by
a subclass.
r � ������ NZunits�break) Z EventTypeZ
MouseWheelZdeltaZ ButtonPressZnum�type�widget�yviewZSCROLL)�eventr) Zup�linesr! r! r" � wheel_event; s r- c @ s� e Zd Zdd� Zdd� Zdd� Zd*dd �Zd+d
d �Zd d
� Zdd� Z d,dd�Z
d-dd�Z d.dd�Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd/d"d#�Zd0d$d%�Zd1d&d'�Zd2d(d)�ZdS )3�TreeNodec C s: || _ || _|| _d| _d| _g | _d | _| _i | _d S )N� collapsedF) �canvas�parent�item�state�selected�children�x�y�
iconimages)�selfr0 r1 r2 r! r! r" �__init__V s zTreeNode.__init__c C s6 x*| j d d � D ]}| j �|� |�� qW d | _d S )N)r5 �remove�destroyr1 )r9 �cr! r! r" r< ` s zTreeNode.destroyc C sf y
| j | S tk
r Y nX tj�|�\}}|p6d}tj�t|| �}t| j|d�}|| j |< |S )Nz.gif)r r ) r8 �KeyErrorr r r r �ICONDIRr r0 )r9 r r Zext�fullnamer
r! r! r" � geticonimagef s
zTreeNode.geticonimageNc C s: | j r
d S | �� d| _ | j�| j� | �� | �� d S )NT)r4 � deselectallr0 �delete�image_id�drawicon�drawtext)r9 r+ r! r! r" �selectr s zTreeNode.selectc C s2 | j s
d S d| _ | j�| j� | �� | �� d S )NF)r4 r0 rC rD rE rF )r9 r+ r! r! r" �deselect{ s zTreeNode.deselectc C s | j r| j �� n| �� d S )N)r1 rB � deselecttree)r9 r! r! r" rB � s zTreeNode.deselectallc C s* | j r| �� x| jD ] }|�� qW d S )N)r4 rH r5 rI )r9 �childr! r! r" rI � s zTreeNode.deselecttreec C s* | j dkr| �� n| �� | j�� dS )N�expandedr' )r3 �collapse�expandr2 �
OnDoubleClick)r9 r+ r! r! r" �flip� s
z
TreeNode.flipc C s2 | j �� sd S | jdkr.d| _| �� | �� d S )NrK )r2 �
_IsExpandabler3 �update�view)r9 r+ r! r! r" rM � s
zTreeNode.expandc C s | j dkrd| _ | �� d S )Nr/ )r3 rQ )r9 r+ r! r! r" rL � s
zTreeNode.collapsec C s� | j d }| �� j d }|| }| j�d�}| j�� }| j�|�}||krV||krVd S | j�| jd �\}}} }
||kr�||kr�|| | } n|} t| �|
} | j�| � d S )Nr � r � scrollregion)r7 �lastvisiblechildr0 ZcanvasyZ winfo_heightZ_getints�floatZ yview_moveto) r9 �topZbottomZheightZ visible_topZvisible_heightZvisible_bottom�x0�y0�x1�y1Zfractionr! r! r" rR � s
z
TreeNode.viewc C s& | j r | jdkr | j d �� S | S d S )NrK �����)r5 r3 rU )r9 r! r! r" rU � s zTreeNode.lastvisiblechildc C s� | j r| j �� nj| jd }d| jd< | j�� | j�t� | �dd� | j�t�\}}}}| jjdd||fd� || jd< d S )NZcursorZwatch� r r )rT )r1 rQ r0 rC ZALL�draw�bbox� configure)r9 Z oldcursorrX rY rZ r[ r! r! r" rQ � s
zTreeNode.updatec C sx d}|| | _ | _| �� | �� | jdkr4|| S | jsz| j�� }|sP|d S x(|D ] }| �| j | |�}| j�
|� qVW |d }|| }d} x�| jD ]�}|} | j j |d |d ||d dd� |� ||�}|j�
� r�|jdkr�d }
|j} n
d
}
|j} | �|
�} | j j|d | d | d �}
| j �|
d | � | j �|
d
dd� � q�W | j j |d |d |d | d dd�}
| j �|
� |S )N� rK rS r � r] Zgray50)�fillZ minusnodeZplusnode)r
z<1>z
c S s d S )Nr! )r6 r! r! r" �� � z TreeNode.draw..r )r6 r7 rE rF r3 r5 r2 � _GetSubList� __class__r0 r Z create_liner^ rP rL rM rA � create_image�tag_bindZ tag_lower)r9 r6 r7 Zdy�sublistr2 rJ ZcxZcyZcylastZiconname�callbackr
�idr! r! r" r^ � sD
"
z
TreeNode.drawc C s~ | j r | j�� p | j�� p d}n| j�� p,d}| �|�}| jj| j| jd|d�}|| _ | j�
|d| j � | j�
|d| j � d S )NZ
openfolderZfolder�nw)�anchorr
z<1>z
)
r4 r2 �GetSelectedIconName� GetIconNamerA r0 rh r6 r7 rD ri rG rO )r9 Z imagenamer
rl r! r! r" rE � s
zTreeNode.drawiconc
s� � j d d }� jd }� j�� }|r�� jj||d|d�}� j�|d� j� � j�|d� j� � j� |�\}}}}t
|d�d }� j� � p�d
} y
� j W n t
k
r� Y n
X � �� y
� j W n* t
k
r� t� j| d d d d
�� _Y nX t�� }
� j�r� j�t�|
d�� n� j�t�|
d�� � jj||d� jd�}� j�d� j� � j�d� j� � j�d� fdd�� � j�d� fdd�� � j�d� fdd�� |� _d S )Nra r r$ rm )rn r z<1>z
�� r z r r )r r ZpadxZpadyZhiliteZnormal)rn Zwindowz c s t | � j�S )N)r- r0 )�e)r9 r! r" rd re z#TreeNode.drawtext..z
c s t | � j�S )N)r- r0 )rr )r9 r! r" rd re z
c s t | � j�S )N)r- r0 )rr )r9 r! r" rd ! re )r6 r7 r2 � GetLabelTextr0 Z create_textri rG rO r_ �max�GetText�entry�AttributeError� edit_finishr r r Z CurrentThemer4 r` Z GetHighlightZ
create_window�bind�select_or_editZtext_id) r9 ZtextxZtextyZ labeltextrl rX rY rZ r[ r Zthemer! )r9 r" rF � s@
zTreeNode.drawtextc C s* | j r | j�� r | �|� n
| �|� d S )N)r4 r2 �
IsEditable�editrG )r9 r+ r! r! r" rz $ s zTreeNode.select_or_editc C sr t | jdddd�| _| j�d| jd � | j�dt� | jjdd� | j�� | j�d| j � | j�d| j
� d S ) Nr r )r �highlightthickness�widthr r& )Zipadxzz) ZEntryr rv �insertZselection_rangeZEND�pack� focus_setry rx � edit_cancel)r9 r+ r! r! r" r| * s
z
TreeNode.editc C s| y| j }| ` W n tk
r" d S X |�� }|�� |rR|| j�� krR| j�|� | j�� }|| jd< | �� | j �
� d S )Nr ) rv rw �getr< r2 ru �SetTextr rF r0 r� )r9 r+ rv r r! r! r" rx 3 s
zTreeNode.edit_finishc C sB y| j }| ` W n tk
r" d S X |�� | �� | j�� d S )N)rv rw r< rF r0 r� )r9 r+ rv r! r! r" r� B s zTreeNode.edit_cancel)N)N)N)N)N)N)N)N)N)�__name__�
__module__� __qualname__r: r< rA rG rH rB rI rO rM rL rR rU rQ r^ rE rF rz r| rx r� r! r! r! r" r. T s(
*
%
r. c @ st e Zd ZdZdd� Zdd� Zdd� ZdZd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� ZdS )�TreeItemz�Abstract class representing tree items.
Methods should typically be overridden, otherwise a default action
is used.
c C s dS )z)Constructor. Do whatever you need to do.Nr! )r9 r! r! r" r: V s zTreeItem.__init__c C s dS )z Return text string to display.Nr! )r9 r! r! r" ru Y s zTreeItem.GetTextc C s dS )z>Return label text string to display in front of text (if any).Nr! )r9 r! r! r" rs \ s zTreeItem.GetLabelTextNc C s | j dkr| �� | _ | j S )z%Do not override! Called by TreeNode.N)�
expandable� IsExpandable)r9 r! r! r" rP a s
zTreeItem._IsExpandablec C s dS )z"Return whether there are subitems.r r! )r9 r! r! r" r� g s zTreeItem.IsExpandablec C s" | � � s g S | �� }|s d| _|S )z%Do not override! Called by TreeNode.r )r� �
GetSubListr� )r9 rj r! r! r" rf k s zTreeItem._GetSubListc C s dS )z-Return whether the item's text may be edited.Nr! )r9 r! r! r" r{ t s zTreeItem.IsEditablec C s dS )z+Change the item's text (if it is editable).Nr! )r9 r r! r! r" r� w s zTreeItem.SetTextc C s dS )z-Return name of icon to be displayed normally.Nr! )r9 r! r! r" rp z s zTreeItem.GetIconNamec C s dS )z2Return name of icon to be displayed when selected.Nr! )r9 r! r! r" ro } s z TreeItem.GetSelectedIconNamec C s dS )z%Return list of items forming sublist.Nr! )r9 r! r! r" r� � s zTreeItem.GetSubListc C s dS )z%Called on a double-click on the item.Nr! )r9 r! r! r" rN � s zTreeItem.OnDoubleClick)r� r� r� �__doc__r: ru rs r� rP r� rf r{ r� rp ro r� rN r! r! r! r" r� M s r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )� FileTreeItemz4Example TreeItem subclass -- browse the file system.c C s
|| _ d S )N)r )r9 r r! r! r" r: � s zFileTreeItem.__init__c C s t j�| j�p| jS )N)r r r )r9 r! r! r" ru � s zFileTreeItem.GetTextc C s t j�| j�dkS )N� )r r r )r9 r! r! r" r{ � s zFileTreeItem.IsEditablec C sl t j�| j�}t j�||�}t j�|�t j�| j�kr:d S yt �| j|� || _W n tk
rf Y nX d S )N)r r �dirnamer �rename�OSError)r9 r �newpathr! r! r" r� � s
zFileTreeItem.SetTextc C s | � � s dS d S )N�python)r� )r9 r! r! r" rp � s zFileTreeItem.GetIconNamec C s t j�| j�S )N)r r �isdir)r9 r! r! r" r� � s zFileTreeItem.IsExpandablec C sj yt �| j�}W n tk
r$ g S X |jt jjd� g }x*|D ]"}tt j�| j|��}|�|� q@W |S )N)�key) r �listdirr r� r �normcaser� r r )r9 �namesrj r r2 r! r! r" r� � s
zFileTreeItem.GetSubListN) r� r� r� r� r: ru r{ r� rp r� r� r! r! r! r" r� � s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�ScrolledCanvasc K sh d|krd|d< || _ t|�| _| jjddd� | jjddd� t| jf|�| _| jjdddd� t| jdd �| _ | j jddd
d� t| jd d d
�| _
| j
jdddd� | j j | jd< | jj | j d< | j
j | jd< | jj
| j
d< | j�d| j� | j�d| j� | j�d| j� | j�d| j� | j�dt� | j�dt� | j�dt� | j�d| j� | j�� d S )NZyscrollincrementrS r r )ZweightZnsew)r r
Zsticky�vbar)r Znse�hbarZ
horizontal)r ZorientZewsZyscrollcommandZcommandZxscrollcommandz z
zz
z z
z
z )r r �frameZ rowconfigureZcolumnconfigureZCanvasr0 r r r� r� �setr* Zxviewry �page_up� page_down�unit_up� unit_downr- � zoom_heightr� )r9 r Zoptsr! r! r" r: � s2
zScrolledCanvas.__init__c C s | j �dd� dS )Nr\ �pager' )r0 � yview_scroll)r9 r+ r! r! r" r� � s zScrolledCanvas.page_upc C s | j �dd� dS )Nr r� r' )r0 r� )r9 r+ r! r! r" r� � s zScrolledCanvas.page_downc C s | j �dd� dS )Nr\ �unitr' )r0 r� )r9 r+ r! r! r" r� � s zScrolledCanvas.unit_upc C s | j �dd� dS )Nr r� r' )r0 r� )r9 r+ r! r! r" r� � s zScrolledCanvas.unit_downc C s t �| j� dS )Nr' )r r� r )r9 r+ r! r! r" r� � s zScrolledCanvas.zoom_heightN) r� r� r� r: r� r� r� r� r� r! r! r! r" r� � s r� c C s� t | �}tt| �� �d�dd � �\}}|�d|d |d f � t|dddd�}|jjdd td
� t t
�}t |j d |�}|�
� d S ) N�+r z+%d+%d�2 � Zwhiter )Zbgr} Z takefocusZboth)rM rc Zside)ZToplevel�map�intZgeometry�splitr� r� r� ZLEFTr� r? r. r0 rM )r1 rW r6 r7 Zscr2 Znoder! r! r" � _tree_widget� s r� �__main__)�mainzidlelib.idle_test.test_treer F)� verbosity�exit)�run)N) r ZtkinterZ tkinter.ttkr r Zidlelib.configr Zidlelibr r? r r r� �__file__Z_icondir� NameErrorr� � RuntimeErrorr# r- r. r� r� r� r� r� Zunittestr� Zidlelib.idle_test.htestr� r! r! r! r" � s4
z<..