o
K��^� � @ s� d dl mZmZmZmZmZmZmZm Z d dl
m Z m Z m
Z
d dlmZ d dlmZ G dd� de
�ZG dd� de
�ZG d d
� d
e�ZG d d � d �ZG d
d� de�ZG dd� de
�ZdS )� )�ACTIVATE�AttrWrap�CompositeCanvas�connect_signal�LineBox�Padding�
PopUpLauncher�Text)�Columns�ListBox�
WidgetWrap)�Color)� widget_widthc sD e Zd ZdgZdd� Zd
� fdd� Zdd� Zd d
� Zd d � Z� Z S )�ClickableThing�clickc C � dS �NT� ��selfr r �1/usr/share/subiquity/subiquitycore/ui/selector.py�
selectable( � zClickableThing.selectableFc s* t � �||�}|rt|�}| �|�|_|S �N)�super�renderr �get_cursor_coords�cursor)r �size�focus�c�� __class__r r r + s
zClickableThing.renderc C r )z�
Return the position of the cursor if visible. This method
is required for widgets that display a cursor.
)r r r )r r r r r r 3 s z ClickableThing.get_cursor_coordsc C r r r )r r �x�yr r r �move_cursor_to_coords: r z$ClickableThing.move_cursor_to_coordsc C s | j | tkr |S | �d� d S )Nr )� _command_mapr �_emit�r r �keyr r r �keypress= s zClickableThing.keypress)F)
�__name__�
__module__� __qualname__�signalsr r r r% r* �
__classcell__r r r! r r % s r c s4 e Zd ZdZ� fdd�Zdd� Z� fdd�Z� ZS )�_PopUpSelectDialogz.A list of PopUpButtons with a box around them.c
s� || _ g }t| j j�D ]O\}}|jr)t|j�}t|d| j|� ||kr&d}nd}n|j}d}tdt d�f|dt |�fg�}|jrGt
|dd�}nt
|d�}t || j j j
d �}|�|� q t|�} || j_t� �t�t| ��� d S )
Nr u ◂ � � � � menu_button�menu_button focus�
info_minor)�width)�parent� enumerate�_options�enabledr �labelr r r
r r � UrwidPadding�_paddingr7 �appendr � base_widget�focus_positionr �__init__r
�bodyr )
r r8 � cur_index�group�i�option�btn�rhs�row�list_boxr! r r rB F s0
�
z_PopUpSelectDialog.__init__c C s || j _| j �� d S r )r8 �index� close_pop_up)r rH rL r r r r c s z_PopUpSelectDialog.clickc s$ |dkr | j �� d S t� �||�S )N�esc)r8 rM r r* r( r! r r r* g s z_PopUpSelectDialog.keypress)r+ r, r- �__doc__rB r r* r/ r r r! r r0 C s
r0 c @ s e Zd ZdS )�
SelectorErrorN)r+ r, r- r r r r rP n s rP c @ s e Zd Zdd� ZdS )�Optionc C s t |t�s-t |t�r|j| _|j| _|j| _n[t |t�r&|| _d| _|| _nLtd|����t|�dkrA|d | _d| _|d | _n1t|�dkrW|d | _|d | _|d | _nt|�dkrm|d | _|d | _|d | _ntd|��t | jt�r�t t
| j��| _d S d S )NTzinvalid option r2 r r3 � zinvalid option %r) �
isinstance�tuplerQ r< r; �value�strrP �lenr �_�r �valr r r rB t s4
�zOption.__init__N)r+ r, r- rB r r r r rQ r s rQ c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )� _Launcherc s || _ t� �|� d S r )r8 r rB )r r8 �childr! r r rB � s z_Launcher.__init__c C �
| j �� S r )r8 �
create_pop_upr r r r r^ � �
z_Launcher.create_pop_upc C r] r )r8 �get_pop_up_parametersr r r r r` � r_ z _Launcher.get_pop_up_parameters)r+ r, r- rB r^ r` r/ r r r! r r[ � s r[ c s� e Zd ZdZdgZd#� fdd� Zdd� Zdd � Zd
d � Ze d d
� �Z
e
j dd
� �Z
e dd� �Z e j dd� �Z dd� Z
dd� Zdd� Ze dd� �Zej dd� �Zdd � Zd d � Zd d � Zd!d"� Z� ZS )$�Selectorz�A widget that allows the user to chose between options by popping
up a list of options.
(A bit like <select> in an HTML form).
�selectr c s� t td��| _tttdtd�f| jdtd�fgdd�dd��| _g }|D ] }|�t|�� q$|| _ | �
|� t � � t
| | j�� d S ) Nr1 r2 �[rR u ▾ ])� dividecharsr4 r5 )r r �_iconr= r r
r> r? rQ �options�
_set_indexr rB r[ )r �optsrL rf �optr! r r rB � s"
���
zSelector.__init__c C s | j jS r )r> r7 r r r r �get_natural_width� s zSelector.get_natural_widthc C s | j | tkr |S | �� d S r )r& r � open_pop_upr( r r r r* � s zSelector.keypressc C s | j | j| j_|| _d S r )r: r< re �_w�_indexrY r r r rg � s
zSelector._set_indexc C s | j S r )rm r r r r rL � s zSelector.indexc C s" | � d| j| j� | �|� d S )Nrb )r' r: rU rg rY r r r rL � s c C s | j d d � S r �r: r r r r rf � � zSelector.optionsc C s& || _ tdd� | j D ��d | j_d S )Nc S s g | ]}t |j��qS r )r r< )�.0�or r r �
<listcomp>� s z$Selector.options.<locals>.<listcomp>� )r: �maxr> r7 rY r r r rf � s � �c C �" | j D ] }|j|kr| S qd S r )r: r< )r r< ri r r r �option_by_label� �
��zSelector.option_by_labelc C ru r )r: rU )r rU ri r r r �option_by_value� rw zSelector.option_by_valuec C s
| j | S r rn )r rL r r r �option_by_index� r_ zSelector.option_by_indexc C s | j | j jS r )r: rm rU r r r r rU � ro zSelector.valuec C s8 t | j�D ]\}}|j|kr| �|� d S qtd|��)Nzcannot set value to %r)r9 r: rU rg �AttributeError)r rZ rF ri r r r rU � s
�
c C s t | | j�S r )r0 rL r r r r r^ � s zSelector.create_pop_upc C s$ d| j d | jjt| j�d d�S )Nr r2 r3 )�left�top�
overlay_width�overlay_height)rL r> r7 rW r: r r r r r` � s �z Selector.get_pop_up_parametersc C � | j �� d S r )rl rk r r r r rk � zSelector.open_pop_upc C r r )rl rM r r r r rM r� zSelector.close_pop_up)r )r+ r, r- rO r. rB rj r* rg �propertyrL �setterrf rv rx ry rU r^ r` rk rM r/ r r r! r ra � s4
ra N)�urwidr r r r r r r= r r �subiquitycore.ui.containerr
r r �subiquitycore.ui.utilsr
�subiquitycore.ui.widthr r r0 � ExceptionrP rQ r[ ra r r r r �<module> s ( +