B
    n&=[                 @   s   d dl mZmZmZ d dlmZ d dlmZ ddlm	Z	 ddl
mZ ddl
mZ d	d
 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jZdS )    )absolute_importdivisionunicode_literals)	text_type)etree   )
tag_regexp   )base)	_ihatexmlc             C   s*   | d krd S t | tr| S | ddS d S )Nasciistrict)
isinstancer   decode)s r   >lib/python3.7/site-packages/html5lib/treewalkers/etree_lxml.py
ensure_str   s
    
r   c               @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
Rootc             C   s   || _ g | _y:|jjrD| jt| t|jjt|jjt|jj	 W n t
k
rZ   Y nX y| }W n t
k
r   |}Y nX x| d k	r| }qW x |d k	r| j| | }qW d | _d | _d S )N)ZelementtreechildrenZdocinfoZinternalDTDappendDoctyper   Z	root_name	public_idZ
system_urlAttributeErrorZgetrootZgetpreviousgetnexttexttail)selfZetnoder   r   r   __init__   s*    




zRoot.__init__c             C   s
   | j | S )N)r   )r   keyr   r   r   __getitem__1   s    zRoot.__getitem__c             C   s   d S )Nr   )r   r   r   r   r   4   s    zRoot.getnextc             C   s   dS )Nr	   r   )r   r   r   r   __len__7   s    zRoot.__len__N)__name__
__module____qualname__r   r!   r   r"   r   r   r   r   r      s   r   c               @   s   e Zd Zdd Zdd ZdS )r   c             C   s(   || _ || _|| _|| _d | _d | _d S )N)	root_nodenamer   	system_idr   r   )r   r&   r'   r   r(   r   r   r   r   <   s    zDoctype.__init__c             C   s   | j jd S )Nr	   )r&   r   )r   r   r   r   r   E   s    zDoctype.getnextN)r#   r$   r%   r   r   r   r   r   r   r   ;   s   	r   c               @   s   e Zd Zdd Zdd ZdS )FragmentRootc                s$    fdd|D  _ d   _ _d S )Nc                s   g | ]}t  |qS r   )FragmentWrapper).0Zchild)r   r   r   
<listcomp>K   s    z)FragmentRoot.__init__.<locals>.<listcomp>)r   r   r   )r   r   r   )r   r   r   J   s    zFragmentRoot.__init__c             C   s   d S )Nr   )r   r   r   r   r   N   s    zFragmentRoot.getnextN)r#   r$   r%   r   r   r   r   r   r   r)   I   s   r)   c               @   sT   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd ZdS )r*   c             C   sT   || _ || _t| jdr(t| jj| _nd | _t| jdrJt| jj| _nd | _d S )Nr   r   )r&   objhasattrr   r   r   )r   Zfragment_rootr-   r   r   r   r   S   s    zFragmentWrapper.__init__c             C   s   t | j|S )N)getattrr-   )r   r'   r   r   r   __getattr___   s    zFragmentWrapper.__getattr__c             C   s6   | j j}|| }|t|d k r.||d  S d S d S )Nr	   )r&   r   indexlen)r   Zsiblingsidxr   r   r   r   b   s
    
zFragmentWrapper.getnextc             C   s
   | j | S )N)r-   )r   r    r   r   r   r!   j   s    zFragmentWrapper.__getitem__c             C   s
   t | jS )N)boolr-   )r   r   r   r   __bool__m   s    zFragmentWrapper.__bool__c             C   s   d S )Nr   )r   r   r   r   	getparentp   s    zFragmentWrapper.getparentc             C   s
   t | jS )N)strr-   )r   r   r   r   __str__s   s    zFragmentWrapper.__str__c             C   s
   t | jS )N)r7   r-   )r   r   r   r   __unicode__v   s    zFragmentWrapper.__unicode__c             C   s
   t | jS )N)r2   r-   )r   r   r   r   r"   y   s    zFragmentWrapper.__len__N)r#   r$   r%   r   r0   r   r!   r5   r6   r8   r9   r"   r   r   r   r   r*   R   s   r*   c               @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )
TreeWalkerc             C   sJ   t |trt|| _t|}nt | _t|}tj| | t	
 | _d S )N)r   listsetfragmentChildrenr)   r   r
   NonRecursiveTreeWalkerr   r   ZInfosetFilterfilter)r   Ztreer   r   r   r   ~   s    


zTreeWalker.__init__c       	      C   s  t |tr:|\}}|dks&td| tjtt||fS t |trLtjfS t |t	rjtj
|j|j|jfS t |trt|dstjt|jfS |jtjkrtjt|jfS |jtjkrtjt|jdd fS tt|j}| r| \}}nd }t|j}i }xbt|j D ]P\}}t|}t|}t|}|rV|||d|df< n||d |f< qW tj|| j !||t"|dkp|jfS d S )N)r   r   z%Text nodes are text or tail, found %stagr	   r   r   )#r   tupleAssertionErrorr
   ZTEXTr   r/   r   ZDOCUMENTr   ZDOCTYPEr'   r   r(   r*   r.   r-   r@   r   CommentCOMMENTr   ZEntityZENTITYr   matchgroupsr;   ZattribitemsgroupZELEMENTr?   ZfromXmlNamer2   )	r   r   r    rF   	namespacer@   Zattrsr'   valuer   r   r   getNodeDetails   s:    




zTreeWalker.getNodeDetailsc             C   sB   t |trtdt|s(|js(td|jr6|dfS |d S d S )NzText nodes have no childrenzNode has no childrenr   r   )r   rB   rC   r2   r   )r   r   r   r   r   getFirstChild   s
    zTreeWalker.getFirstChildc             C   sb   t |trL|\}}|dks&td| |dkrDt|r>|d S d S n| S |jrZ|dfS | S )N)r   r   z%Text nodes are text or tail, found %sr   r   r   )r   rB   rC   r2   r   r   )r   r   r    r   r   r   getNextSibling   s    
zTreeWalker.getNextSiblingc             C   sJ   t |tr4|\}}|dks&td| |dkrB|S n|| jkrBd S | S )N)r   r   z%Text nodes are text or tail, found %sr   )r   rB   rC   r=   r6   )r   r   r    r   r   r   getParentNode   s    

zTreeWalker.getParentNodeN)r#   r$   r%   r   rL   rM   rN   rO   r   r   r   r   r:   }   s
   )	r:   N)Z
__future__r   r   r   Zsixr   Zlxmlr   Ztreebuilders.etreer    r
   r   r   objectr   r   r)   r*   r>   r:   r   r   r   r   <module>   s   	&	+