EVOLUTION-NINJA
Edit File: formfill.cpython-36.pyc
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 .