Your IP : 18.118.184.76
B
�|/\�G � @ sp d Z ddlmZ ddlZddlZddlZyddlmZ W n e k
rX ddl
mZ Y nX ddddd d
d gZ ye Z
W n ek
r� eefZ
Y nX ejd d kZe�d�Ze�d�Ze�d�ZejZd
d� Zdd� Zejddd�Zdd� Z e� d�Z e� d�Z G dd� de�Z G dd � d e �Z!d#dd
�Z"d$dd �Z#G d d � d e$�Z%d d � Z&d d!iZ'e(d"k�rlddlZe�)� dS )%a�
lxml-based doctest output comparison.
Note: normally, you should just import the `lxml.usedoctest` and
`lxml.html.usedoctest` modules from within a doctest, instead of this
one::
>>> import lxml.usedoctest # for XML output
>>> import lxml.html.usedoctest # for HTML output
To use this module directly, you must call ``lxmldoctest.install()``,
which will cause doctest to use this in all subsequent calls.
This changes the way output is checked and comparisons are made for
XML or HTML-like content.
XML or HTML content is noticed because the example starts with ``<``
(it's HTML if it starts with ``<html``). You can also use the
``PARSE_HTML`` and ``PARSE_XML`` flags to force parsing.
Some rough wildcard-like things are allowed. Whitespace is generally
ignored (except in attributes). In text (attributes and text in the
body) you can use ``...`` as a wildcard. In an example it also
matches any trailing tags in the element, though it does not match
leading tags. You may create a tag ``<any>`` or include an ``any``
attribute in the tag. An ``any`` tag matches any tag, while the
attribute matches any and all attributes.
When a match fails, the reformatted example and gotten text is
displayed (indented), and a rough diff-like output is given. Anything
marked with ``+`` is in the output but wasn't supposed to be, and
similarly ``-`` means its in the example but wasn't in the output.
You can disable parsing on one line with ``# doctest:+NOPARSE_MARKUP``
� )�etreeN)�escape�
PARSE_HTML� PARSE_XML�NOPARSE_MARKUP�LXMLOutputChecker�LHTMLOutputChecker�install� temp_install� c C s | d kr d S | � � S d S )N)�strip)�v� r �F/opt/alt/python37/lib64/python3.7/site-packages/lxml/doctestcompare.pyr ? s r c C s t �d| �S )N� )�_norm_whitespace_re�sub)r
r r r �norm_whitespaceE s r FT)ZrecoverZremove_blank_textc C s t �| t�S )N)r Z
fromstring� _html_parser)�htmlr r r �html_fromstringJ s r z^<[^>]+ (at|object) z[ \t\n][ \t\n]+c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z d'dd�Z d(dd�Z
d)dd�Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd*d$d%�Zd&S )+r )
�paramZimgZarea�brZbasefont�input�base�meta�link�colc C s t jS )N)r �XML)�selfr r r �get_default_parserW s z$LXMLOutputChecker.get_default_parserc C s� t | dd �}|d k r | j}|} ntj}| �|||�}|sF|| |||�S y ||�}W n tjk
rh dS X y ||�}W n tjk
r� dS X | �||�S )N�_temp_override_selfF)�getattr� _temp_call_super_check_output�
OutputChecker� check_output�
get_parserr �XMLSyntaxError� compare_docs) r �want�got� optionflagsZalt_selfZ super_method�parser�want_doc�got_docr r r r% Z s$ z LXMLOutputChecker.check_outputc C st d }t |@ rd S t|@ r t}nRt|@ r.tj}nB|�� �� �d�rT|�� �d�rTt}n | � |�rp| � |�rp| �
� }|S )Nz<html) r r r r r r r �lower�
startswith�_looks_like_markupr )r r) r* r+ r, r r r r& o s
z LXMLOutputChecker.get_parserc C s |� � }|�d�o t�|� S )N�<)r r0 �_repr_re�search)r �sr r r r1 s
z$LXMLOutputChecker._looks_like_markupc
C s | � |j|j�sdS | �|j|jd�s*dS | �|j|jd�s@dS d|jkr�t|j�� �}t|j�� �}||krrdS x*|D ]"}| �|j| |j| d�sxdS qxW |jdks�t|��rt |�}t |�}xN|s�|�r|r�|s�dS |�
d�}|�
d�} | � || �s�dS |s�|jdkr�P q�W dS )NFT�anyz...r ) � tag_compare�tag� text_compare�text�tail�attrib�sorted�keys�len�list�popr( )
r r) r* Z want_keysZgot_keys�key�
want_children� got_childrenZ
want_firstZ got_firstr r r r( � s6
z LXMLOutputChecker.compare_docsc C s^ |pd}|pd}|r,t |��� }t |��� }dt�|� }|�dd�}t�||�rVdS dS d S )N� z^%s$z\.\.\.z.*TF)r r �rer �replacer4 )r r) r* r r r r r9 � s z LXMLOutputChecker.text_comparec C sj |dkr dS t |t�r t |t�s(||kS |p.d}|p6d}|�d�r^|�d�d |�d�d kS ||kS d S )Nr6 TrE z{...}�}���)�
isinstance� _basestringr0 �split)r r) r* r r r r7 � s
z LXMLOutputChecker.tag_comparec
C s |j }| �|||�}g }|d k r�y ||�}W n0 tjk
r\ t�� d }|�d| � Y nX y ||�} W n0 tjk
r� t�� d }|�d| � Y nX |d ks�|r�t�| |||�}
|r�|�|
� d� |�S |
S |t
k} d| � || d�d| � | | d�d| � || | d�g} d� | �S ) N� zIn example: %szIn actual output: %s�
z Expected:� zGot:zDiff:)
r) r& r r' �sys�exc_info�appendr$ �output_difference�joinr �
format_doc� collect_diff)
r Zexampler* r+ r) r, �errorsr- �er. �valuer Z
diff_partsr r r rS � s:
z#LXMLOutputChecker.output_differenceTc C s. |sdS |j | jkrdS |js&t|�r*dS dS )NFT)r8 �
empty_tagsr: r? )r �elr r r r �html_empty_tag� s z LXMLOutputChecker.html_empty_tagrE c C s� g }t |�s�|�d| � |�|� |�| �|�� | �||�slt|j�r\|�| �|j�� |�| �|�� t|j�r�|�| �|j�� |�d� d� |�S |�d| � |�|� |�| �|�� | �||��sb|�d� t|j��r|�d| � |�| �|j�� |�d� x&|D ] }|�| �
|||d �� �qW |�d| � |�| �|�� |�d� t|j��r�|�d| � |�| �|j�� |�d� d� |�S )Nr rN rE rO ) r? rR �
format_tagr\ r r: � format_text�format_end_tagr; rT rU )r �docr �indent�prefix�partsr[ r r r rU � s@
z LXMLOutputChecker.format_docc C s" |d kr dS |r|� � }t|d�S )NrE rM )r � html_escape)r r: r r r r r^ s
z LXMLOutputChecker.format_textc C sl g }t |tj�rdS x4t|j�� �D ]"\}}|�d|| �|d�f � q$W |sXd|j S d|jd� |�f S )Nz<!--z%s="%s"Fz<%s>z<%s %s>r )
rJ r � CommentBaser= r< �itemsrR r^ r8 rT )r r[ �attrs�namerY r r r r] s
z LXMLOutputChecker.format_tagc C s t |tj�rdS d|j S )Nz-->z</%s>)rJ r re r8 )r r[ r r r r_ s z LXMLOutputChecker.format_end_tagc C s g }t |�s�t |�s�|�d| � |�| �||�� | �||�sh|�| �|j|j�� |�| �||�� |�| �|j|j�� |�d� d�|�S |�d| � |�| �||�� |�d� t |j�s�t |j�r�|�d| � |�| �|j|j�� |�d� t
|�}t
|�}x�|�s |�r�|�sF|�| � |� d�||d d�� �q|�sp|�| � |� d�||d d�� �q|�| �
|� d�|� d�||d �� �qW |�d| � |�| �||�� |�d� t |j��s�t |j��r |�d| � |�| �|j|j�� |�d� d�|�S )Nr rN rE r rO �+�-)r? rR �collect_diff_tagr\ �collect_diff_textr: �collect_diff_end_tagr; rT r r@ rU rA rV )r r) r* r ra rc rC rD r r r rV ! sH
$
z LXMLOutputChecker.collect_diffc C s2 | � |j|j�s"d|j|jf }n|j}g }|jdkp>d|jk}x�t|j�� �D ]p\}}||jkr�|s�|�d|| �|d�f � qP||jkr�| �|j| |d�}n | �|d�}|�d||f � qPW |�s x@t|j�� �D ].\}}||jkr�q�|�d|| �|d�f � q�W |�r&d|d�|�f }nd | }|S )
Nz %s (got: %s)r6 z+%s="%s"Fz%s="%s"z-%s="%s"z<%s %s>r z<%s>) r7 r8 r< r= rf rR r^ rl rT ) r r) r* r8 rg r6 rh rY r: r r r rk G s*
z"LXMLOutputChecker.collect_diff_tagc C s, |j |j kr d|j |j f }n|j }d| S )Nz %s (got: %s)z</%s>)r8 )r r) r* r8 r r r rm b s z&LXMLOutputChecker.collect_diff_end_tagc C s: | � |||�r"|sdS | �||�S d||f }| �||�S )NrE z %s (got: %s))r9 r^ )r r) r* r r: r r r rl i s z#LXMLOutputChecker.collect_diff_textN)T)rE )T)T)�__name__�
__module__� __qualname__rZ r r% r& r1 r( r9 r7 rS r\ rU r^ r] r_ rV rk rm rl r r r r r Q s$
#
&c @ s e Zd Zdd� ZdS )r c C s t S )N)r )r r r r r r s z%LHTMLOutputChecker.get_default_parserN)rn ro rp r r r r r r q s c C s | r t t_ntt_dS )z�
Install doctestcompare for all future doctests.
If html is true, then by default the HTML parser will be used;
otherwise the XML parser is used.
N)r �doctestr$ r )r r r r r u s c C sz | r
t }nt}t� }|jd }|� }|j}||_trJ|jd j}|jj}n|jd j}|jj}t t
_ t ||||||� dS )z�
Use this *inside* a doctest to enable this checker for this
doctest only.
If html is true, then by default the HTML parser will be used;
otherwise the XML parser is used.
r �checkN) r r �_find_doctest_frame�f_localsZ_checker� _IS_PYTHON_3�__func__r% Zim_funcr rq �_RestoreChecker) r �
del_moduleZChecker�frame�dt_self�checker� old_checker�
check_funcZchecker_check_funcr r r r
� s"
c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )rw c C sD || _ || _| j| j_|| j_|| _|| _|| _| �� | � � d S )N)
rz r{ �
call_superr# r! r} �
clone_funcrx �
install_clone�install_dt_self)r rz r| Z new_checkerr} r rx r r r �__init__� s
z_RestoreChecker.__init__c C sJ t r&| jj| _| jj| _| jj| j_n | jj| _| jj| _| jj| j_d S )N)ru r} �__code__� func_code� __globals__Z func_globalsr )r r r r r� � s
z _RestoreChecker.install_clonec C s t r| j| j_n
| j| j_d S )N)ru r� r} r� )r r r r �uninstall_clone� s z _RestoreChecker.uninstall_clonec C s | j j| _| | j _d S )N)rz � _DocTestRunner__record_outcome� prev_func)r r r r r� � s
z _RestoreChecker.install_dt_selfc C s | j | j_d S )N)r� rz r� )r r r r �uninstall_dt_self� s z!_RestoreChecker.uninstall_dt_selfc C sL | j rHdd l}|j| j = d| j krH| j �dd�\}}|j| }t||� d S )Nr �.rM )rx rP �modules�rsplit�delattr)r rP �package�moduleZ package_modr r r �uninstall_module� s
z _RestoreChecker.uninstall_modulec O s4 | � � | �� | j`| j`| j||�}| �� |S )N)r� r� r{ r! r# r� r� )r �args�kw�resultr r r �__call__� s z_RestoreChecker.__call__c O s$ | � � z | j||�S | �� X d S )N)r� r} r� )r r� r� r r r r~ � s z_RestoreChecker.call_superN) rn ro rp r� r� r� r� r� r� r� r~ r r r r rw � s rw c C s@ dd l } | �d�}x |r2|j}d|kr*|S |j}qW td��d S )Nr rM ZBOOMzBCould not find doctest (only use this function *inside* a doctest))rP � _getframert �f_back� LookupError)rP ry �lr r r rs � s
rs Zbasicai
>>> temp_install()
>>> print """<xml a="1" b="2">stuff</xml>"""
<xml b="2" a="1">...</xml>
>>> print """<xml xmlns="http://example.com"><tag attr="bar" /></xml>"""
<xml xmlns="...">
<tag attr="..." />
</xml>
>>> print """<xml>blahblahblah<foo /></xml>""" # doctest: +NOPARSE_MARKUP, +ELLIPSIS
<xml>...foo /></xml>
�__main__)F)FN)*�__doc__�lxmlr rP rF rq r r rd � ImportError�cgi�__all__�
basestringrK � NameError�str�bytes� version_inforu Zregister_optionflagr r r r$ r r Z
HTMLParserr r �compiler3 r r r r r
�objectrw rs Z__test__rn Ztestmodr r r r �<module>$ sL
"
'6