o
K��^5Q � @ sN d Z ddlZddlZddlZe�d�Zdd� Zdd� Zeej j
� � �D ]\Z Z
e
dd ejfv r6e
ej e d
< q#ej �� 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 s0 | d u rd S t | j|d �}|d ur|� d S d S �N)�getattr� base_widget)�w�methname�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)�widgets� cur_focus�ir r r r �_has_other_selectablei s
�r �next selectable�prev selectable� no wrap�enterz
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 �> t | j�D ]\}\}}|�� r | �|� t|d� dS qdS �z6Select first selectable child (possibily recursively).�_select_first_selectableN�r r r
� set_focusr
�r r r �or r r r � �
��z'TabCyclingPile._select_first_selectablec C �F t tt| j���D ]\}\}}|�� r | �|� t|d� dS q dS �z5Select last selectable child (possibily recursively).�_select_last_selectableN��reversed�listr r r
r r
r r r r r$ � �
��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� � zTabCyclingPile._widgetsc C s� | j s|S |}|�d�s | j| dv r t| �� | j�r |d7 }d}t|�dkr.| j|dd�}| j}| �� rM| � ||d|�}| j
� ||�}| j| dvrM|S | j| dkr�| jd }t | j
|d� |�D ]\}\} }
| �� rz| �|� t| d
� dS qc|�d�s�| �� |S | j| d kr�| j
d| j� } ttt | ���D ]\}\} }
| �� r�| �|� t| d � dS q�|�d�s�| �� |S | j| d
kr�tt|d dd��} d }
ntt|d t| j ���} d
}
|s�| j|dd�}| D ]_}| j | d �� s�q�| �|� || _t| j
|
� t| j
d��s dS || }| j| d
k�r(tt|d dd��}ntt|��}|D ]}| � ||d|�}| j�|| j|��rH dS �q0 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$ r2 ���r �move_cursor_to_coords)r �endswith� _command_mapr r- �focus_position�len�
get_item_rowsr
�
get_item_sizer1 �keypressr r+ r r
r r& r' r$ �range�_update_pref_col_from_focus�hasattr�
focus_itemr5 �pref_col)r �size�key�downkey� item_rowsr �tsize�upkey�next_fpr r � positions�
candidates�meth�j�rows�rowlist�rowr r r r< � s� �
�
�
��zTabCyclingPile.keypressN)
�__name__�
__module__� __qualname__�enter_advancing_command_mapr7 r
r r$ r- r< r r r r r y s r c s2 e Zd Z d
� fdd� Zdd� Zdd � Z� ZS ) �OneSelectableColumnsr Nr3 c sL t � �|||||� d}| jD ] \}}|�� r|d7 }q|dkr$td��d S )Nr r3 z3subiquity only supports one selectable in a Columns)�super�__init__r+ r
� Exception) r � widget_list� dividechars� focus_column� min_width� box_columns� selectablesr r �� __class__r r rV � s ����z OneSelectableColumns.__init__c C r r r r r r r r r! z-OneSelectableColumns._select_first_selectablec C r" r# r% r r r r r$
r( z,OneSelectableColumns._select_last_selectable)r Nr3 N)rP rQ rR rV r r$ �
__classcell__r r r^ r rT � s � rT c sD e Zd ZeZ� fdd�Zdd� Zdd� Zdd� Z� fd d
�Z � Z
S ) �TabCyclingListBoxc s* t |dd �d u r
t�|�}t� �|� d S )N� get_focus)r �urwid�SimpleFocusListWalkerrU rV )r �bodyr^ r r rV 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 qdS r )r re r
rg r
�r r r r r r r ) s
��z*TabCyclingListBox._select_first_selectablec C sB t tt| j���D ]\}}|�� r | �|� t|d� dS q dS r# )r& r' r re r
rg r
rh r r r r$ 1 s
��z)TabCyclingListBox._select_last_selectablec s@ |}|� d�s| j| dv rt| j| j�r|d7 }t� �||�}||kr&|}t| j�dkr/|S | j| dkrf| jd }t| j|d � |�D ]\}}|� � rZ| �
|� t |d� d S qE|� d�sd| � � |S | j| dkr�| jd | j� }t
tt|���D ]\}}|� � r�| �
|� t |d� d S q}|� d�s�| �� |S |S ) Nr r/ r r r3 r r r$ )r6 r7 r re r8 rU r< r9 r r
r r
r r& r' r$ ) r rB rC rD rG rH r r rI r^ r r r< 9 sD �
�
�
zTabCyclingListBox.keypress) rP rQ rR rS r7 rV rg r r$ r< r` r r r^ r ra s ra c s4 e Zd Z� fdd�Zdd� Zdd� Zdd� Z� ZS ) �FocusTrackingMixinc s$ t � j|i |�� | j�| j� d S r )rU rV r+ �set_focus_changed_callback�_focus_changed�r �args�kwr^ r r rV a s zFocusTrackingMixin.__init__c C s. z| j }W n
ty Y d S w t|d� d S �N�
lost_focus)r1 �
IndexErrorr
)r r r r r rp e s
�z FocusTrackingMixin.lost_focusc C � t | jd� d S �N� gained_focus�r
r1 r, r r r rt m r. z FocusTrackingMixin.gained_focusc C sB z| j }W n ty Y nw t|d� t| | d� | �� d S �Nrp rt )r1 rq r
rf )r � new_focusr r r r rk p s
�
z!FocusTrackingMixin._focus_changed)rP rQ rR rV rp rt rk r` r r r^ r ri _ s
ri c @ � e Zd ZdS )�FocusTrackingPileN�rP rQ rR r r r r ry { � ry c @ rx )�FocusTrackingColumnsNrz r r r r r| r{ r| c s@ e Zd Z� fdd�Zdd� Zdd� Zdd� Z� fd d
�Z� ZS ) �FocusTrackingListBoxc s* t � j|i |�� | j�| j� d| _d S )NF)rU rV re rj rk �_in_set_focus_completerl r^ r r rV � s
z FocusTrackingListBox.__init__c C s2 | j st| jd� t| j| d� | �� d S d S rv )r~ r
r1 re rf )r rw r r r rk � s
�z#FocusTrackingListBox._focus_changedc C rr ro ru r, r r r rp � r. z FocusTrackingListBox.lost_focusc C rr rs ru r, r r r rt � r. z!FocusTrackingListBox.gained_focusc s* d| _ z
t� �||� W d| _ d S d| _ w r )r~ rU �_set_focus_complete�r rB r1 r^ r r r � s z(FocusTrackingListBox._set_focus_complete) rP rQ rR rV rk rp rt r r` r r r^ r r} � s r} 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 )N�fixedr3 �weight� scrollbarzscrollbar focus) �Pile�top�bg�box�botrc �AttrMap�bar�_always_scrollrU rV )r �lbr� �piler^ r r rV � s
�zScrollBarListBox.__init__c C s | j rdS t| j�||��dkS )NTr0 )r� r9 �original_widget� ends_visibler� r r r �_scroll� s zScrollBarListBox._scrollc C s2 | j }| �|d�s|d d |d f}|�||�S )NTr r3 )r� r� r< )r rB rC r� r r r r< � s zScrollBarListBox.keypressc C s� | j }| �||�r|�||�S |\}}|�|d |f�\}}|�|d |f|�}d} d }
} |j�� \} }
|jD ]}|�|d f�}|| u rId} n| sO| |7 } |
|7 }
q8d|v r[d}n| | | }d|v rhd}n|
| | }| jj j
}||| k r|dd�}n|d|�}| j }|dkr�t �
|d �}| j}|dkr�t �
|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 )
Nr3 Fr Tr� �bottom�givenr� r� )r� r� �render�get_focus_offset_insetr� re rb rM r� r �optionsr� rc r� r� r� r� r �
CanvasJoinr8 )r rB r1 r� �maxcol�maxrow�offset�inset�visible�
seen_focus�height�height_before_focus� focus_widget� focus_pos�widgetrM r� r� r �boxopt� top_arrow� bot_arrow� lb_canvas�
bar_canvasr r r r� � sX
�
�zScrollBarListBox.render)F)rP rQ rR rc � SolidFillr� r� r� r� rV r� r< r� r` r r r^ r r� � s
r� Fr� c C sF | d u rg } | t | dd � u rd u rn nt�| �} tt| �|d�S )Nrb r� )r rc rd r� r} )re r� r r r �ListBox
s
�r� z_wrapped_widget.base_widgetc @ s< e Zd Zedd� �Zedd� �Zedd� �Zedd� �ZdS )�
OurWidgetWrapc C �
t | �jS r )� get_delegater r, r r r �<lambda>! �
zOurWidgetWrap.<lambda>c C r� r )r� r$ r, r r r r� # r� c C r� r )r� rp r, r r r r� $ r� c C r� r )r� rt r, r r r r� % r� N)rP rQ rR �propertyr r$ rp rt r r r r r� s �� r� r )!�__doc__�logging�operatorrc � getLogger�logr
r r' � command_map�_command�itemsrC �command�ACTIVATE�copyrS r� r �ColumnsrT r� ra ri ry r| r} �WidgetDecorationr� �
attrgetterr� �
WidgetWrapr� r r r r �<module> s8 &
�
| L h