U
K��^�: � @ sJ d dl Z d dlZd dlmZ d dlmZmZmZmZm Z m
Z m Z m
Z
mZ d dlmZmZ d dlmZmZ d dlmZmZmZmZ d dlmZ d dl m Z m Z m Z d d l m!Z!m"Z"m#Z#m$Z$ d d
l%m&Z& e�'d �Z(e)� Z*e)� Z+G d d
� d
ed�e�Z,G dd� de�Z-G dd� de j.�Z/G dd� de)�Z0de dd�iZ1G dd� de)�Z2dd� Z3e3e�Z4e3e�Z5e3e�Z6e3e�Z7G d d � d e �Z8G d d � d e/�Z9G d d!� d!ee0�Z:e3e:�Z;G d"d#� d#e/�Z<G d$d%� d%e
�Z=G d&d'� d'e/�Z>G d(d)� d)e�Z?G d*d+� d+e/�Z@G d,d-� d-e �ZAG d.d/� d/e)eAd0�ZBG d1d2� d2e�ZCG d3d4� d4e/�ZDG d5d6� d6eB�ZEdS )7� N)�urlparse) �CheckBox�connect_signal�delegate_to_widget_mixin� emit_signal� MetaSignals�Padding� RadioButton�Text�WidgetDecoration)�
cancel_btn�done_btn)�Pile�
WidgetWrap)�PasswordEditor�
IntegerEditor� StringEditor� EmailEditor)�Selector)�ColSpec� TablePile�TableRow)� button_pile�Color�disabled�screen)� widget_widthzsubiquitycore.ui.formc @ s2 e Zd ZdZdd� Zedd� �Zejdd� �ZdS )�
ToggleableTc C s || _ d| _d| _d S )NFT)�original�_enabled�enabled)�selfr � r" �-/usr/share/subiquity/subiquitycore/ui/form.py�__init__N s zToggleable.__init__c C s | j S �N�r �r! r" r" r# r S s zToggleable.enabledc C s4 |r| j s| j| _n|s*| j r*t| j�| _|| _ d S r% )r r Zoriginal_widgetr �r! �valr" r" r# r W s
N)�__name__�
__module__� __qualname__Zhas_original_widthr$ �propertyr �setterr" r" r" r# r I s
r Z_original_widgetc s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�
_Validatorc s || _ t� �|� d S r% )�field�superr$ )r! r0 �w�� __class__r" r# r$ b s z_Validator.__init__c C s
t | j�S r% )r �_wr' r" r" r# �get_natural_widthf s z _Validator.get_natural_widthc C s4 d| j _t| jjdd �}|d k r&|� | j �� d S )NF�
lost_focus)r0 �
showing_extra�getattrr5 � base_widget�validate)r! Zlfr" r" r# r7 i s
z_Validator.lost_focus)r* r+ r, r$ r6 r7 �
__classcell__r" r" r3 r# r/ ` s r/ c @ s8 e Zd ZdZdZdZd
dd�Zejdd� �Z dd � Z
dS ) � FormFieldr TNc C s& || _ || _tj| _t jd7 _d S )N� )�caption�helpr= �
next_index�index)r! r? r@ r" r" r# r$ w s zFormField.__init__c C s d S r% r" �r! �formr" r" r# � _make_widget} s zFormField._make_widgetc C s | � |�}t| ||�S r% )rE �BoundFormField)r! rD �widgetr" r" r# �bind� s
zFormField.bind)NN) r* r+ r, rA �takes_default_style�
caption_firstr$ �abc�abstractmethodrE rH r" r" r" r# r= q s
r= c @ s e Zd ZdZdd� ZdS )�WantsToKnowFormFieldzA marker class.c C s
|| _ d S r% )�bff)r! rN r" r" r# �set_bound_form_field� s z)WantsToKnowFormField.set_bound_form_fieldN)r* r+ r, �__doc__rO r" r" r" r# rM � s rM r> F)Zpackc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d d
�Zdd� Z e
dd� �Z e j dd� �Z e
dd� �Z
e
j dd� �Z
e
dd� �Zej dd� �Ze
dd� �Zej dd� �Zd S ) rF c C sh || _ || _|| _d| _d| _d | _d| _| �� dt|dg �krPt |d| j
� t |t �rd|�
| � d S )NFTZchange�signals)r0 rD rG �in_errorr �_helpr8 � _build_tabler9 r �_change�
isinstancerM rO )r! r0 rD rG r" r" r# r$ � s
zBoundFormField.__init__c C s | j }| jjrt�|�}| jtk r0t| j�| _n
td�| _| jj t
krddt | |�fg}d| jfg}n`tt | jj ��| _
| jjr�d| j
_| j
t | |�g}n t | t|dt|�d��| j
g}td�| jg}|g}| jtk r�|�|� dd� |D �| _t| jdtd�| _d S )N� � �right)�align�widthc S s g | ]}t t|���qS r" )r r )�.0�rowr" r" r# �
<listcomp>� s z/BoundFormField._build_table.<locals>.<listcomp>)ZspacingZcolspecs)rG r0 rI r Z string_inputr@ �NO_HELPr
�
under_textr? �
NO_CAPTIONr/ �_� caption_textrJ rZ � UrwidPaddingr �append�_rowsr �
form_colspecs�_table)r! rG Z first_rowZ
second_row�rowsr" r" r# rT � s:
���
zBoundFormField._build_tablec C s* t | jd| jj d �}|d k r&||�}|S )NZclean_)r9 rD r0 �name)r! �valueZcleanerr" r" r# �clean� s zBoundFormField.cleanc C sb | j s | jr^d| _|| _| �� }| `|d k r0d S d| _ | jsT| jtk rT| j�| j� | j� � d S )NF)
rR r8 �tmpval� _validater@ r_ r` �set_textrD � validated)r! �senderZnew_val�rr" r" r# rU � s zBoundFormField._changec
C sl | j s
d S z
| j W n. tk
rB } zt|� W Y �S d }~X Y nX t| jd| jj d �}|d k rh|� S d S )NZ validate_)r rk �
ValueError�strr9 rD r0 rj )r! �eZ validatorr" r" r# rn � s
zBoundFormField._validateTc C s\ | � � }|d kr6d| _| jsN| jtk rN| j�| j� nd| _|rN| �d|f� | j� � d S )NFTZ
info_error)
rn rR r8 r@ r_ r` ro �
show_extrarD rp )r! �
show_errorrr r" r" r# r; � s zBoundFormField.validatec C s d| _ | j�|� d S )NT)r8 r` ro )r! Z extra_markupr" r" r# rv � s zBoundFormField.show_extrac C s | � t| d| jj��S )Nrm )rl r9 rG rk r' r" r" r# rk � s zBoundFormField.valuec C s || j _d S r% )rG rk r( r" r" r# rk s c C sH | j d k r| j S | jjd k r@t| jjt�r6t| jj�S | jjS ndS d S �NrW )rS r0 r@ rV rt rb r' r" r" r# r@ s
zBoundFormField.helpc C s" |d kr d}|| _ | j�|� d S rx )rS r` ro r( r" r" r# r@ s c C s | j jS r% )rc �textr' r" r" r# r? s zBoundFormField.captionc C s | j �|� d S r% )rc ro r( r" r" r# r? s c C s | j S r% r& r' r" r" r# r ! s zBoundFormField.enabledc C s || _ | jD ]
}||_q d S r% )r rf r )r! r) r] r" r" r# r % s
N)T)r* r+ r, r$ rT rl rU rn r; rv r- rk r. r@ r? r r" r" r" r# rF � s. %
rF c s G � fdd�dt �}|S )Nc s e Zd Z� fdd�ZdS )zsimple_field.<locals>.Fieldc s � � S r% r" rC �� widget_makerr" r# rE . s z(simple_field.<locals>.Field._make_widgetN)r* r+ r, rE r" rz r" r# �Field- s r| )r= )r{ r| r" rz r# � simple_field, s r} c @ s* e Zd ZdZedd� �Zejdd� �ZdS )�RadioButtonEditor� c C s | j S r% ��stater' r" r" r# rk = s zRadioButtonEditor.valuec C s
|| _ d S r% r� r( r" r" r# rk A s N�r* r+ r, Zreserve_columnsr- rk r. r" r" r" r# r~ 9 s
r~ c s. e Zd ZdZdZd� fdd� Zdd� Z� ZS )�RadioButtonFieldFNc s. |d kr g }|� | � || _t� �||� d S r% )re �groupr1 r$ )r! r� r? r@ r3 r" r# r$ K s
zRadioButtonField.__init__c C s2 |j D ] }|j| jkr|jj} q(qg }t|d�S rx )�_fieldsr0 r� rG r~ )r! rD �bfr� r" r" r# rE R s
z RadioButtonField._make_widget)NN)r* r+ r, rJ rI r$ rE r< r" r" r3 r# r� F s r� c s8 e Zd Zeddg�f� fdd� Zejjdd� �Z� ZS )� URLEditorZhttpZhttpsc s || _ t� �� d S r% )�allowed_schemesr1 r$ )r! r� r3 r" r# r$ ] s zURLEditor.__init__c C s� | � � }|dkr|S t|�}|j| jkr�g }t| j�D ]}|�|� q6t|�dkrvd�|d d� �td� |d }n.t|�dkr�|d td� |d }n|d }t td �j
|d
���|S ) NrW rX z, ���z, or r z or r> z#This field must be a {schemes} URL.)�schemes) Z
get_edit_textr Zschemer� �sortedre �len�joinrb rs �format)r! �vZparsedr� �sr" r" r# rk a s& $ ��zURLEditor.value) r* r+ r, � frozensetr$ r rk �getterr< r" r" r3 r# r� \ s r� c s0 e Zd ZdZddg f� fdd� Zdd� Z� ZS )� ChoiceFieldFNc s t � �||� || _d S r% )r1 r$ �choices)r! r? r@ r� r3 r" r# r$ ~ s zChoiceField.__init__c C s
t | j�S r% )r r� rC r" r" r# rE � s zChoiceField._make_widget�r* r+ r, rI r$ rE r< r" r" r3 r# r� z s r� c @ s& e Zd Zedd� �Zejdd� �ZdS )�ReadOnlyWidgetc C s | j S r% )ry r' r" r" r# rk � s zReadOnlyWidget.valuec C s | � |� d S r% )ro r( r" r" r# rk � s N)r* r+ r, r- rk r. r" r" r" r# r� � s
r� c @ s e Zd ZdZdd� ZdS )�
ReadOnlyFieldFc C s t d�S rx )r� rC r" r" r# rE � s zReadOnlyField._make_widgetN)r* r+ r, rI rE r" r" r" r# r� � s r� c @ s* e Zd ZdZedd� �Zejdd� �ZdS )�CheckBoxEditorr c C s | j S r% r� r' r" r" r# rk � s zCheckBoxEditor.valuec C s
|| _ d S r% r� r( r" r" r# rk � s Nr� r" r" r" r# r� � s
r� c @ s e Zd ZdZdZdd� ZdS )� BooleanFieldFc C s t d�S rx )r� rC r" r" r# rE � s zBooleanField._make_widgetN)r* r+ r, rJ rI rE r" r" r" r# r� � s r� c s e Zd Z� fdd�Z� ZS )�MetaFormc sn t � �|||� g }|�� D ]6\}}t|t�r ||_|jd krH|d |_|�|� q |jdd� d� || _ d S )N�:c S s | j S r% )rB )�fr" r" r# �<lambda>� � z#MetaForm.__init__.<locals>.<lambda>)�key)
r1 r$ �itemsrV r= rj r? re �sort�_unbound_fields)r! rj �basesZattrsr� �kr� r3 r" r# r$ � s
zMetaForm.__init__�r* r+ r, r$ r< r" r" r3 r# r� � s r� c @ sr e Zd ZddgZed�Zed�Zi fdd�Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z ddd�Z
dd� Zdd� ZdS )�Form�submit�cancelZDoneZCancelc C s� t tt| j�| jd��| _t tt| j�| jd��| _t| j| jg�| _ g | _
| j D ]@}|� | �}t
| |jj|� | j
�|� |j|krR||j |_qR| j
D ]}|jdd� q�| �� d S )N)Zon_pressF)rw )r r
rb �ok_label� _click_doner � cancel_label�
_click_cancelr �buttonsr� r� rH �setattrr0 rj re rk r; rp )r! �initialr0 r� r" r" r# r$ � s" �
�
z
Form.__init__c C s, | j D ] }|jj|kr||jj |j_qd S r% )r� r0 rj rk )r! �datar� r" r" r# �
enter_data� s
zForm.enter_datac C s t | d| � d S )Nr� �r �r! rq r" r" r# r� � s zForm._click_donec C s t | d| � d S )Nr� r� r� r" r" r# r� � s zForm._click_cancelc C s8 g }| j D ]}|jj|kr
|�|� q
|| j d d �< d S r% )r� r0 rj re )r! Z
field_nameZ
new_fieldsr� r" r" r# � remove_field� s
zForm.remove_fieldc C sd t | j�dkrg S | jd j}|g}| jdd � D ],}|�td�� |j}|�|� |�|� q2|S )Nr r> rW )r� r� rh re r
rH )r! Zt0ri r0 �tr" r" r# �as_rows� s
z Form.as_rowsTNFc C s t | �� | j|||d�S )N)�
focus_buttons�excerpt� narrow_rows)r r� r� )r! r� r� r� r" r" r# � as_screen� s �zForm.as_screenc C sZ d}| j D ]}|jr
d} q q
|rBd| jjjd d _d| jj_nd| jjjd d _d S )NFTr r> )r� rR r� r: �contentsr Zfocus_position)r! rR r� r" r" r# rp � s
zForm.validatedc C s( i }| j D ]}|jr
|j||jj< q
|S r% )r� r rk r0 rj )r! r� r0 r" r" r# �as_data s
z Form.as_data)TNF)r* r+ r, rQ rb r� r� r$ r� r� r� r� r� r� rp r� r" r" r" r# r� � s
r� )� metaclassc s6 e Zd Z� fdd�Zedd� �Zejdd� �Z� ZS )�
SubFormWidgetc s || _ t� �t|�� �� d S r% )rD r1 r$ r r� rC r3 r" r# r$ s zSubFormWidget.__init__c C s
| j �� S r% )rD r� r' r" r" r# rk s zSubFormWidget.valuec C s$ |� � D ]\}}|t| j|�_qd S r% )r� r9 rD rk )r! r� r� r� r" r" r# rk s )r* r+ r, r$ r- rk r. r< r" r" r3 r# r� s
r� c s* e Zd ZdZd� fdd� Zdd� Z� ZS )� SubFormFieldFNc s t � j||d� || _d S )N)r? r@ )r1 r$ �form_cls)r! r� r? r@ r3 r" r# r$ " s zSubFormField.__init__c C s | � |�}t|�S r% )r� r� rC r" r" r# rE & s
zSubFormField._make_widget)NNr� r" r" r3 r# r� s r� c s e Zd Z� fdd�Z� ZS )�SubFormc s || _ t� jf |� d S r% )�parentr1 r$ )r! r� �kwr3 r" r# r$ - s zSubForm.__init__r� r" r" r3 r# r� + s r� )FrK ZloggingZ urllib.parser Zurwidr r r r r r rd r r
r Zsubiquitycore.ui.buttonsr r
Zsubiquitycore.ui.containerr r Z subiquitycore.ui.interactiver r r r Zsubiquitycore.ui.selectorr Zsubiquitycore.ui.tabler r r Zsubiquitycore.ui.utilsr r r r Zsubiquitycore.ui.widthr Z getLogger�log�objectra r_ r r/ �ABCr= rM rg rF r} Z StringFieldZ
PasswordFieldZ IntegerFieldZ
EmailFieldr~ r� r� ZURLFieldr� r� r� r� r� r� r� r� r� r� r" r" r" r# �<module> sR ,
�
P
|