U
    HaL!                     @   sv   d Z ddlmZ ddlmZ edZdd ZG dd dZG d	d
 d
eZG dd de	Z
G dd deZdd ZdS )zCompare two HTML documents.    )
HTMLParser)_lazy_re_compilez[\t\n\f\r ]+c                 C   s   t d| S )N )ASCII_WHITESPACEsub)string r   4/tmp/pip-unpacked-wheel-3jxiddxt/django/test/html.pynormalize_whitespace   s    r
   c                   @   sf   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdddZdd Z	dd Z
dd Zdd Zdd ZdS )Elementc                 C   s   || _ t|| _g | _d S N)namesorted
attributeschildren)selfr   r   r   r   r	   __init__   s    
zElement.__init__c                 C   s   t |trTt|}| jrt | jd tr| jd  |7  < t| jd | jd< d S n.| jrt | jd tr| jd  r| j  |r| j| d S N)
isinstancestrr
   r   isspacepopappendr   elementr   r   r	   r      s    

zElement.appendc                    sZ    fdd  | j  t| j D ]4\}}t|trB| | j |< q t|dr |  q d S )Nc                    s>   | r:t | d tr:| d  | d< | d s:|    | } | S r   )r   r   rstripr   r   rstrip_last_elementr   r	   r   (   s    z-Element.finalize.<locals>.rstrip_last_elementfinalize)r   	enumerater   r   striphasattrr    )r   ichildr   r   r	   r    '   s    


zElement.finalizec                 C   s   t |dr| j|jkrdS t| jt|jkr2dS | j|jkrtt| jD ]N}| j| \}}|j| \}}|d krx|}|d kr|}||ks||krL dS qL| j|jkS )Nr   F)r#   r   lenr   ranger   )r   r   r$   attrvalueZ
other_attrZother_valuer   r   r	   __eq__7   s    zElement.__eq__c                 C   s   t | jf| jS r   )hashr   r   r   r   r   r	   __hash__K   s    zElement.__hash__Tc                 C   s   t |ts| |krdS t |tr0| j|jkr0dS d}d}| jD ]}t |trzt |tr|rj|||7 }q||kr dS q>||j||d7 }|s|r|  S t |tr>|jr>|j| }||kr|d7 }|t|jkr|d7 }d}q>d}q>|S )N   r   count)r   r   RootElementr   r0   _countr&   )r   r   r0   r$   Zelem_child_idxr%   Z
elem_childr   r   r	   r2   N   s2    



zElement._countc                 C   s   | j |dddkS )NFr/   r   r2   r   r   r   r	   __contains__u   s    zElement.__contains__c                 C   s   | j |ddS )NTr/   r3   r   r   r   r	   r0   x   s    zElement.countc                 C   s
   | j | S r   r   )r   keyr   r   r	   __getitem__{   s    zElement.__getitem__c                 C   s   d| j  }| jD ]*\}}|r.|d||f 7 }q|d| 7 }q| jrt|d7 }|ddd | jD 7 }|d| j  7 }n|d	7 }|S )
Nz<%sz %s="%s"z %sz>
 c                 s   s   | ]}t |V  qd S r   r   .0cr   r   r	   	<genexpr>   s     z"Element.__str__.<locals>.<genexpr>z
</%s>>)r   r   r   join)r   outputr5   r)   r   r   r	   __str__~   s    
zElement.__str__c                 C   s   t | S r   r8   r,   r   r   r	   __repr__   s    zElement.__repr__N)T)__name__
__module____qualname__r   r   r    r*   r-   r2   r4   r0   r6   r@   rA   r   r   r   r	   r      s   
'r   c                       s$   e Zd Z fddZdd Z  ZS )r1   c                    s   t  d d d S )Nr   )superr   r,   	__class__r   r	   r      s    zRootElement.__init__c                 C   s   d dd | jD S )Nr7   c                 s   s   | ]}t |V  qd S r   r8   r9   r   r   r	   r<      s     z&RootElement.__str__.<locals>.<genexpr>)r>   r   r,   r   r   r	   r@      s    zRootElement.__str__)rB   rC   rD   r   r@   __classcell__r   r   rF   r	   r1      s   r1   c                   @   s   e Zd ZdS )HTMLParseErrorN)rB   rC   rD   r   r   r   r	   rI      s   rI   c                       s~   e Zd Zddddddddd	d
ddddddhZ fddZdd Zd"ddZedd Zdd Z	dd Z
dd Zd d! Z  ZS )#ParserareabasebrcolembedhrimginputlinkmetaparamsourcetrackwbrframeZspacerc                    s"   t    t | _g | _i | _d S r   )rE   r   r1   root	open_tagselement_positionsr,   rF   r   r	   r      s    
zParser.__init__c                 C   s   t ||  d S r   )rI   getpos)r   msgr   r   r	   error   s    zParser.errorNc                 C   s@   |s|r| j | }|d kr"|  }t|dr8|j|jf}d| S )NlinenozLine %d, Column %d)r\   r]   r#   r`   offset)r   positionr   r   r   r	   format_position   s    

zParser.format_positionc                 C   s   | j r| j d S | jS d S r   )r[   rZ   r,   r   r   r	   current   s    
zParser.currentc                 C   s$   |  || || jkr | | d S r   )handle_starttagSELF_CLOSING_TAGShandle_endtag)r   tagattrsr   r   r	   handle_startendtag   s    
zParser.handle_startendtagc                 C   sL   dd |D }t ||}| j| || jkr:| j| |  | j|< d S )Nc              
   S   sB   g | ]:\}}|d kr6|d tdd t|D fn||fqS )classr   c                 s   s   | ]}|r|V  qd S r   r   )r:   r)   r   r   r	   r<      s      z4Parser.handle_starttag.<locals>.<listcomp>.<genexpr>)r>   r   r   split)r:   r   r)   r   r   r	   
<listcomp>   s   $z*Parser.handle_starttag.<locals>.<listcomp>)r   rd   r   rf   r[   r]   r\   )r   rh   ri   r   r   r   r	   re      s    

zParser.handle_starttagc                 C   s\   | j s| d||  f  | j  }|j|krX| j sL| d||  f  | j  }q&d S )NzUnexpected end tag `%s` (%s))r[   r_   rc   r   r   )r   rh   r   r   r   r	   rg      s     

 zParser.handle_endtagc                 C   s   | j | d S r   )rd   r   )r   datar   r   r	   handle_data   s    zParser.handle_data)NN)rB   rC   rD   rf   r   r_   rc   propertyrd   rj   re   rg   ro   rH   r   r   rF   r	   rJ      s4                
	
rJ   c                 C   sR   t  }||  |  |j}|  t|jdkrNt|jd tsN|jd }|S )z
    Take a string that contains HTML and turn it into a Python object structure
    that can be easily compared against other HTML on semantic equivalence.
    Syntactical differences like which quotation is used on arguments will be
    ignored.
    r.   r   )	rJ   feedcloserZ   r    r&   r   r   r   )htmlparserdocumentr   r   r	   
parse_html   s    

rv   N)__doc__html.parserr   Zdjango.utils.regex_helperr   r   r
   r   r1   	ExceptionrI   rJ   rv   r   r   r   r	   <module>   s    E