U
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 s dS �NT� ��selfr r �1/usr/share/subiquity/subiquitycore/ui/selector.py�
selectable( s zClickableThing.selectableFc s* t � �||�}|r&t|�}| �|�|_|S �N)�super�renderr �get_cursor_coordsZcursor)r �sizeZfocus�c�� __class__r r r + s
zClickableThing.renderc C s dS )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 s dS r r )r r �x�yr r r �move_cursor_to_coords: s 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 ]�\}}|jrRt|j�}t|d| j|� ||krLd}q\d}n
|j}d}tdt d�f|dt |�fg�}|jr�t
|dd�}n
t
|d�}t || j j j
d �}|�|� qt|�} || j_t� �t�t| ��� d S )
Nr u ◂ � � � � menu_button�menu_button focusZ
info_minor)�width)�parent� enumerate�_options�enabledr �labelr r r
r r � UrwidPadding�_paddingr2 �appendr Z base_widgetZfocus_positionr �__init__r
Zbodyr )
r r3 Z cur_index�group�iZoption�btnZrhs�rowZlist_boxr r r r; F s0
�
z_PopUpSelectDialog.__init__c C s || j _| j �� d S r )r3 �index� close_pop_up)r r> r@ r r r r c s z_PopUpSelectDialog.clickc s& |dkr| j �� nt� �||�S d S )NZesc)r3 rA r r& r$ r r r r& g s z_PopUpSelectDialog.keypress)r' r( r) �__doc__r; r r& r+ r r r r r, C s r, c @ s e Zd ZdS )�
SelectorErrorN)r' r( r) r r r r rC n s rC c @ s e Zd Zdd� ZdS )�Optionc C s t |t�sXt |t�r.|j| _|j| _|j| _q�t |t�rL|| _d| _|| _q�td|��n�t|�dkr�|d | _d| _|d | _nbt|�dkr�|d | _|d | _|d | _n6t|�dkr�|d | _|d | _|d | _n
td|��t | jt�r�t t
| j��| _d S )NTzinvalid option %rr. r r/ � ) �
isinstance�tuplerD r7 r6 �value�strrC �lenr �_�r �valr r r r; t s2
zOption.__init__N)r' r( r) r; r r r r rD r s rD c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )� _Launcherc s || _ t� �|� d S r )r3 r r; )r r3 Zchildr r r r; � s z_Launcher.__init__c C s
| j �� S r )r3 �
create_pop_upr r r r rO � s z_Launcher.create_pop_upc C s
| j �� S r )r3 �get_pop_up_parametersr r r r rP � s z _Launcher.get_pop_up_parameters)r' r( r) r; rO rP r+ r r r r rN � s rN 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|�� qH|| _ | �
|� t � � t
| | j�� d S ) Nr- r. �[rE u ▾ ])Z dividecharsr0 r1 )r r �_iconr8 r r
r9 r: rD �options�
_set_indexr r; rN )r Zoptsr@ rU �optr r r r; � s$
�� �
zSelector.__init__c C s | j jS r )r9 r2 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 )r5 r7 rT �_w�_indexrL r r r rV � s zSelector._set_indexc C s | j S r )r[ r r r r r@ � s zSelector.indexc C s" | � d| j| j� | �|� d S )NrR )r# r5 rH rV rL r r r r@ � s c C s | j d d � S r �r5 r r r r rU � s zSelector.optionsc C s& || _ tdd� | j D ��d | j_d S )Nc S s g | ]}t |j��qS r )r r7 )�.0�or r r �
<listcomp>� s z$Selector.options.<locals>.<listcomp>� )r5 �maxr9 r2 rL r r r rU � s ��c C s" | j D ]}|j|kr| S qd S r )r5 r7 )r r7 rW r r r �option_by_label� s
zSelector.option_by_labelc C s" | j D ]}|j|kr| S qd S r )r5 rH )r rH rW r r r �option_by_value� s
zSelector.option_by_valuec C s
| j | S r r\ )r r@ r r r �option_by_index� s zSelector.option_by_indexc C s | j | j jS r )r5 r[ rH r r r r rH � s zSelector.valuec C s< t | j�D ]"\}}|j|kr
| �|� d S q
td|��d S )Nzcannot set value to %r)r4 r5 rH rV �AttributeError)r rM r= rW r r r rH � s
c C s t | | j�S r )r, r@ r r r r rO � s zSelector.create_pop_upc C s$ d| j d | jjt| j�d d�S )Nr r. r/ )�left�topZ
overlay_widthZoverlay_height)r@ r9 r2 rJ r5 r r r r rP � s �z Selector.get_pop_up_parametersc C s | j �� d S r )rZ rY r r r r rY s zSelector.open_pop_upc C s | j �� d S r )rZ rA r r r r rA s zSelector.close_pop_up)r )r' r( r) rB r* r; rX r& rV �propertyr@ �setterrU rb rc rd rH rO rP rY rA r+ r r r r rQ � s2
rQ N)Zurwidr r r r r r r8 r r Zsubiquitycore.ui.containerr
r r Zsubiquitycore.ui.utilsr
Zsubiquitycore.ui.widthr r r, � ExceptionrC rD rN rQ r r r r �<module> s ( + |