U
K��^5Q � @ sN d Z ddlZddlZddlZe�d�Zdd� Zdd� Zeej j
� � �D ]&\Z Z
e
dd ejfkrFe
ej e d
< qFej �� Zded < ded < G d
d� dej�ZG dd� dej�ZG dd� dej�ZG dd� d�ZG dd� dee�ZG dd� dee�ZG dd� de�ZeZeZG dd � d ej�Z d$d d �d d �Ze� d!�Z G d"d#� d#ej �Z e Z dS )%a" Extended versions of urwid containers.
This module provides versions of urwid's Columns, Pile and ListBox
containers. It adds support for tab-cycling, callbacks on
gaining/losing focus and scrollbars on listboxes.
1. Tab-cycling
Tab advances to the next selectable item and wraps when it gets to the
end of the screen. Shift tab goes the other way. In addition, an
unhandled "enter" key also advances to the next selectable item.
2. Callbacks when a widget gains/loses focus
Define lost_focus or gained_focus callbacks on widgets and they'll be
called when the widget gains or loses focus.
3. Scrollbars on ListBoxes.
All ListBoxes in subiquity gain a scrollbar when their contents don't
entirely fit on screen. The implementation assumes that there are not
too many elements in the ListBox.
� Nzsubiquitycore.ui.containerc C s, | d kr d S t | j|d �}|d k r(|� d S �N)�getattr� base_widget)�wZmethname�m� r �2/usr/share/subiquity/subiquitycore/ui/container.py� _maybe_calla s
r c C s, t | �D ] \}}||kr|�� r dS qdS �NTF)� enumerate�
selectable)Zwidgets� cur_focus�ir r r r �_has_other_selectablei s r �next selectable�prev selectable� no wrapZenterz
enter no wrapc @ s8 e Zd ZeZdd� Zdd� Zdd� Zdd� Zd d
� Z d S ) �TabCyclingPilec C s" | j D ]\}}|�� r dS qdS r
)�contentsr )�selfr �_r r r r | s zTabCyclingPile.selectablec C s> t | j�D ].\}\}}|�� r
| �|� t|d� dS q
dS �z6Select first selectable child (possibily recursively).�_select_first_selectableN�r r r � set_focusr �r r r �or r r r � s
z'TabCyclingPile._select_first_selectablec C sF t tt| j���D ].\}\}}|�� r| �|� t|d� dS qdS �z5Select last selectable child (possibily recursively).�_select_last_selectableN��reversed�listr r r r r r r r r r � s
z&TabCyclingPile._select_last_selectablec C s dd� | j D �S )Nc S s g | ] \}}|�qS r r )�.0r r r r r �
<listcomp>� s z+TabCyclingPile._widgets.<locals>.<listcomp>)� _contents�r r r r �_widgets� s zTabCyclingPile._widgetsc C s� | j s
|S |}|�d�s>| j| dkr>t| �� | j�r>|d7 }d}t|�dkr\| j|dd�}| j}| �� r�| � ||d|�}| j
� ||�}| j| dkr�|S | j| dk�r| jd }t | j
|d� |�D ].\}\} }
| �� r�| �|� t| d
� dS q�|�d��s | �� |S | j| d k�r�| j
d| j� } ttt | ���D ]2\}\} }
| �� �r@| �|� t| d � dS �q@|�d��s�| �� |S | j| d
k�r�tt|d dd��} d }
n tt|d t| j ���} d
}
|�s�| j|dd�}| D ]�}| j | d �� �s�q�| �|� || _t| j
|
� t| j
d��s6 dS || }| j| d
k�rdtt|d dd��}n tt|��}|D ]0}| � ||d|�}| j�|| j|��rt �q��qt dS |S )zjPass the keypress to the widget in focus.
Unhandled 'up' and 'down' keys may cause a focus change.r �r r N� T)�focus)� cursor upz cursor downr r r � r r r r* ���r �move_cursor_to_coords)r �endswith� _command_mapr r& �focus_position�lenZ
get_item_rowsr Z
get_item_sizer) �keypressr r$ r r r r r! r �rangeZ_update_pref_col_from_focus�hasattrZ
focus_itemr- Zpref_col)r �size�key�downkeyZ item_rowsr Ztsize�upkey�next_fpr r � positionsZ
candidates�meth�j�rowsZrowlist�rowr r r r2 � s� �
�
zTabCyclingPile.keypressN)
�__name__�
__module__� __qualname__�enter_advancing_command_mapr/ r r r r& r2 r r r r r y s r c s. e Zd Zd
� fdd� Zdd� Zdd � Z� ZS ) �OneSelectableColumnsr Nr+ c sL t � �|||||� d}| jD ]\}}|�� r |d7 }q |dkrHtd��d S )Nr r+ z3subiquity only supports one selectable in a Columns)�super�__init__r$ r � Exception) r Z widget_listZ dividecharsZ focus_columnZ min_widthZ box_columnsZ selectablesr r �� __class__r r rE � s �
�z OneSelectableColumns.__init__c C s> t | j�D ].\}\}}|�� r
| �|� t|d� dS q
dS r r r r r r r s
z-OneSelectableColumns._select_first_selectablec C sF t tt| j���D ].\}\}}|�� r| �|� t|d� dS qdS r r r r r r r
s
z,OneSelectableColumns._select_last_selectable)r Nr+ N)r? r@ rA rE r r �
__classcell__r r rG r rC � s � rC c sD e Zd ZeZ� fdd�Zdd� Zdd� Zdd� Z� fd d
�Z � Z
S ) �TabCyclingListBoxc s* t |dd �d krt�|�}t� �|� d S )N� get_focus)r �urwid�SimpleFocusListWalkerrD rE )r �bodyrG r r rE s
zTabCyclingListBox.__init__c C s | � |� | � |� | �� d S r )r � _invalidate)r r r r r �_set_focus_no_move s
z$TabCyclingListBox._set_focus_no_movec C s: t | j�D ]*\}}|�� r
| �|� t|d� dS q
dS r )r rN r rP r �r r r r r r r ) s
z*TabCyclingListBox._select_first_selectablec C sB t tt| j���D ]*\}}|�� r| �|� t|d� dS qdS r )r r! r rN r rP r rQ r r r r 1 s
z)TabCyclingListBox._select_last_selectablec sH |}|� d�s2| j| dkr2t| j| j�r2|d7 }t� �||�}||krL|}t| j�dkr^|S | j| dkr�| jd }t| j|d � |�D ]*\}}|� � r�| �
|� t |d� d S q�|� d�s�| � � |S | j| dk�r@| jd | j� }t
tt|���D ]*\}}|� � r�| �
|� t |d� d S q�|� d��s<| �� |S |S d S ) Nr r' r r r+ r r r )r. r/ r rN r0 rD r2 r1 r r r r r r r! r ) r r5 r6 r7 r8 r9 r r r: rG r r r2 9 s@ �
zTabCyclingListBox.keypress) r? r@ rA rB r/ rE rP r r r2 rI r r rG r rJ s rJ c s4 e Zd Z� fdd�Zdd� Zdd� Zdd� Z� ZS ) �FocusTrackingMixinc s t � j||� | j�| j� d S r )rD rE r$ �set_focus_changed_callback�_focus_changed�r �args�kwrG r r rE a s zFocusTrackingMixin.__init__c C s. z
| j }W n tk
r Y n X t|d� d S �N�
lost_focus)r) �
IndexErrorr )r r
r r r rY e s
z FocusTrackingMixin.lost_focusc C s t | jd� d S �N� gained_focus�r r) r% r r r r\ m s z FocusTrackingMixin.gained_focusc C sD z
| j }W n tk
r Y n X t|d� t| | d� | �� d S �NrY r\ )r) rZ r rO )r � new_focusr
r r r rT p s
z!FocusTrackingMixin._focus_changed)r? r@ rA rE rY r\ rT rI r r rG r rR _ s rR c @ s e Zd ZdS )�FocusTrackingPileN�r? r@ rA r r r r r` { s r` c @ s e Zd ZdS )�FocusTrackingColumnsNra r r r r rb s rb c s@ e Zd Z� fdd�Zdd� Zdd� Zdd� Z� fd d
�Z� ZS ) �FocusTrackingListBoxc s& t � j||� | j�| j� d| _d S )NF)rD rE rN rS rT �_in_set_focus_completerU rG r r rE � s z FocusTrackingListBox.__init__c C s. | j s*t| jd� t| j| d� | �� d S r^ )rd r r) rN rO )r r_ r r r rT � s z#FocusTrackingListBox._focus_changedc C s t | jd� d S rX r] r% r r r rY � s z FocusTrackingListBox.lost_focusc C s t | jd� d S r[ r] r% r r r r\ � s z!FocusTrackingListBox.gained_focusc s& d| _ zt� �||� W 5 d| _ X d S r
)rd rD �_set_focus_complete�r r5 r) rG r r re � s z(FocusTrackingListBox._set_focus_complete) r? r@ rA rE rT rY r\ re rI r r rG r rc � s
rc c sd e Zd Ze�d�Ze�d�Ze�d�Ze�d�Zdd�� fdd�
Z d d
� Z
d d � Z dd
d�Z � Z
S )�ScrollBarListBoxu ▴u █� u ▾F��
always_scrollc s` t dd| jfdd| jfdd| jfdd| jfdd| jfg�}t�|dd�| _|| _t � �
|� d S )NZfixedr+ �weight� scrollbarzscrollbar focus) �Pile�top�bg�box�botrL �AttrMap�bar�_always_scrollrD rE )r �lbrj ZpilerG r r rE � s
�zScrollBarListBox.__init__c C s | j r
dS t| j�||��dkS )NTr( )rt r1 �original_widget� ends_visiblerf r r r �_scroll� s zScrollBarListBox._scrollc C s2 | j }| �|d�s&|d d |d f}|�||�S )NTr r+ )rv rx r2 )r r5 r6 ru r r r r2 � s zScrollBarListBox.keypressc C s� | j }| �||�r |�||�S |\}}|�|d |f�\}}|�|d |f|�}d} d }
} |j�� \} }
|jD ]6}|�|d f�}|| kr�d} n | s�| |7 } |
|7 }
qpd|kr�d}n | | | }d|kr�d}n |
| | }| jj j
}||| k �r |dd�}n
|d|�}| j }|dk�r&t �
|d �}| j}|dk�rBt �
|d �}||dd�f| j|d|�f| j|f| j|d|�f||dd�fg| jj jd d �< |�|d |f|�}| j�d|f|�}t �||jd|d f|d ddfg�S d S )
Nr+ Fr Trn �bottomZgivenrk rl )rv rx �renderZget_focus_offset_insetrw rN rK r= rs r Zoptionsrn rL rr rq ro rp r Z
CanvasJoinr0 )r r5 r) ru ZmaxcolZmaxrow�offsetZinsetZvisibleZ
seen_focusZheightZheight_before_focusZ focus_widgetZ focus_posZwidgetr= rn ry r ZboxoptZ top_arrowZ bot_arrowZ lb_canvasZ
bar_canvasr r r rz � sX
�
�zScrollBarListBox.render)F)r? r@ rA rL Z SolidFillrn rp ro rq rE rx r2 rz rI r r rG r rg � s
rg Fri c C sF | d kr g } | t | dd � kr(d kr6n n
t�| �} tt| �|d�S )NrK ri )r rL rM rg rc )rN rj r r r �ListBox
s
�r| z_wrapped_widget.base_widgetc @ s< e Zd Zedd� �Zedd� �Zedd� �Zedd� �ZdS )�
OurWidgetWrapc C s
t | �jS r )� get_delegater r% r r r �<lambda>! � zOurWidgetWrap.<lambda>c C s
t | �jS r )r~ r r% r r r r # r� c C s
t | �jS r )r~ rY r% r r r r $ r� c C s
t | �jS r )r~ r\ r% r r r r % r� N)r? r@ rA �propertyr r rY r\ r r r r r} s �� r} )N)!�__doc__Zlogging�operatorrL Z getLogger�logr r r! Z command_mapZ_command�itemsr6 ZcommandZACTIVATE�copyrB rm r ZColumnsrC r| rJ rR r` rb rc ZWidgetDecorationrg �
attrgetterr~ Z
WidgetWrapr} r r r r �<module>' s4
| L h
|