3
)�2Y�% � @ sx d dl mZmZ d dlmZmZ d dlmZmZmZm Z d dlm
Z
d dl Z ye W n e
k
rl eZ Y nX dddd d
d gZG d d� de�Zed
deid�Zedjdd� d7D ��deid�Zeddeid�Zed�Zd8dd�Zd9d d�Zd d � Zd d � Zd!d"� Zd#d$� Z d%d&� Z d'd(� Z d:d)d*�Z d+d,� Z G d-d � d e!�Z"e"� Z#ddd.e#fd/d �Z$d0d
� Z%d1d2� Z&d3d4� Z'd5d6� Z(dS );� )�XPath� ElementBase)�
fromstring�XHTML_NAMESPACE)� _forms_xpath�_options_xpath�_nons�_transform_result)�defsN� FormNotFound� fill_form�fill_form_html�
insert_errors�insert_errors_html�DefaultErrorCreatorc @ s e Zd ZdZdS )r z*
Raised when no form can be found
N)�__name__�
__module__� __qualname__�__doc__� r r � /usr/lib64/python3.6/formfill.pyr s zKdescendant-or-self::form[name=$name]|descendant-or-self::x:form[name=$name]�x)Z
namespaces�|c C s g | ] }d | �qS )zdescendant-or-self::r )�.0Z_tagr r r �
<listcomp> s r �input�select�textarea�x:input�x:select�
x:textareaz-//label[@for=$for_id]|//x:label[@for=$for_id]z"descendant-or-self::*[@name=$name]c C s t | ||d�} t| |� d S )N)�form_id�
form_index)�
_find_form�
_fill_form)�el�valuesr! r" r r r r s c C s@ t | �}t| t�r t| �}n
tj| �}t||||d� t||�S )N)r! r" )�type�
isinstance�
basestringr �copy�deepcopyr r )�htmlr& r! r" � result_type�docr r r r
& s
c C s� i }t |d�r|j� }t| �}x�|D ]�}|jd�}|s8q$t|�rl|j|g �}t|ttf�s`|g}t||� q$||krxq$q$|j|d�}|d ||< || }t|ttf�r�y || }W q� t k
r� w$Y q�X n
|dkr�q$t
||� q$W d S )N�mixed�namer � ) �hasattrr/ � _input_xpath�get�_takes_multipler( �list�tuple�_fill_multiple�
IndexError� _fill_single)r% r& ZcountsZinputsr r0 �value�indexr r r r$ / s4
r$ c C s< t | j�dkr | jd�r dS | jdd�j� }|d kr8dS dS )
Nr ZmultipleTr' � �radio�checkboxF)r> r? )r �tagr4 �lower)r r' r r r r5 M s r5 c C s� | j dd�j� }|dkrj| j d�}|d krZ|s4d}n|d }t|t�rN|dk}t| |� q�t| ||k� nn|dkr�| j d�}t| ||k� nLt| j�d ks�t�x8t| �D ],}|j d�}|d kr�|j � }t
|||k� q�W d S )
Nr' r= r? r; Fr Zonr> r ) r4 rA r( r) �_checkr r@ �AssertionErrorr Z text_content�_select)r r; r' �v�resultZoptionr r r r8 V s(
r8 c C s( |r| j dd� nd| jkr$| jd= d S )NZcheckedr= )�set�attrib)r% Zcheckr r r rB r s
rB c C s( |r| j dd� nd| jkr$| jd= d S )NZselectedr= )rG rH )r% r r r r rD y s
rD c C s& t | j�dkr|| _n | jd|� d S )Nr r; )r r@ �textrG )r r; r r r r: � s r: c C s� |d kr0|d kr0t | �}x|D ]}|S W td��|d k r�| j|�}|d k rN|S t| |d�}|rf|d S tdtdjt| ��f ��|d k r�t | �}y|| S tk
r� td|t|�f ��Y nX d S )NzNo forms in page)r0 r z-No form with the name or id of %r (forms: %s)z, z3There is no form with the index %r (%i forms found)) r r �get_element_by_id�_form_name_xpath�id�join�_find_form_idsr9 �len)r% r! r" �forms�formr r r r# � s0
r# c c s� t | �}|sdV d S xtt|�D ]h\}}|jd�rf|jd�rXd|jd�|jd�f V q�|jd�V q |jd�r~|jd�V q d| V q W d S )Nz
(no forms)rL r0 z%s or %sz(unnamed form %s))r � enumerater4 )r% rP r< rQ r r r rN � s
rN c @ s4 e Zd ZdZdZdZdZdZdZdd� Z dd � Z
d
S ) r TZdivz
error-messagez error-blockZInvalidc K s< x6|j � D ]*\}}t| |�s(td| ��t| ||� q
W d S )Nz Unexpected keyword argument: %s)�itemsr2 � TypeError�setattr)�self�kwr0 r; r r r �__init__� s
z DefaultErrorCreator.__init__c C s |j | j�}| jr |jd| j� |rH| jrH|jd|jdd�d | j � |d ksX|dkr^| j}t|t�rt|j |� n"t|t
�s�t d| ��|p�| j|_ |r�| j
r�| jr�|j |_d |_ |jd|� n
|j |� nD|j� }|j|�}| jr�|j||� n |j|_d |_|j|d |� d S )N�classr= � z.Bad message; should be a string or element: %rr r1 )Z makeelement�error_container_tag�error_message_classrG �error_block_classr4 �default_messager( r �appendr) rC rI � block_inside�
insert_before�tail�insertZ getparentr< )rV r% �is_block�message�error_el�parent�posr r r �__call__� s2
z DefaultErrorCreator.__call__N) r r r ra r` r[ r\ r] r^ rX ri r r r r r � s �errorc
C sz t | ||d�} xf|j� D ]Z\}}|d kr*qxFt| ||�D ]6\}} t| ttd �tf�s`td| ��t|| ||� q8W qW d S )N)r! r" zBad message: %r) r# rS �_find_elements_for_namer( r) r' r rC �
_insert_error)
r% �errorsr! r" � error_class�
error_creatorr0 rj rf re r r r r � s
c K s> t | �}t| t�r t| �}n
tj| �}t||f|� t||�S )N)r' r( r) r r* r+ r r )r, r& rW r- r. r r r r � s
c C s� t | j�tjks t | j�dkr$d}nd}t | j�dkrD|rDt| |� | jd�r|t| | jd�d�}|r|x|D ]}t||� qjW || ||� d S )Nr FTrQ rL )Zfor_id)r r@ r
Z
empty_tags�
_add_classr4 �_label_for_xpath)r% rj rn ro rd ZlabelsZlabelr r r rl s
rl c C s6 | j d�r&| jd| j d�d | � n | jd|� d S )NrY rZ )r4 rG )r% �
class_namer r r rp s
rp c c s� |d kr| |fV d S |j d�rH| j|dd � �}|d k rD||fV d S t| |d�}|s\d S t|ttf�s||d |fV d S x*t||�D ] \}}|d kr�q�||fV q�W d S )N�#r1 )r0 r )�
startswithrJ � _name_xpathr( r6 r7 �zip)rQ r0 rj r% Zels�errr r r rk s$
rk )r r r r r r )NN)NN)NN))Z
lxml.etreer r Z lxml.htmlr r r r r r r
r* r) � NameError�str�__all__� LookupErrorr rK rM r3 rq ru r r
r$ r5 r8 rB rD r: r# rN �objectr Zdefault_error_creatorr r rl rp rk r r r r �<module> sN
.