3
(h                 @   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 | jddS d S )Nasciistrict)
isinstancer   decode)s r   E/tmp/pip-install-q3hcpn_q/html5lib/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jt| t|jjt|jjt|jj	 W n t
k
rZ   Y nX y|j }W n t
k
r   |}Y nX x|j d k	r|j }qW x |d k	r| jj| |j }qW d | _d | _d S )N)ZelementtreechildrenZdocinfoZinternalDTDappendDoctyper   Z	root_name	public_idZ
system_urlAttributeErrorgetrootZgetpreviousgetnexttexttail)self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).0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_rootr0   r   r   r   r!   S   s    zFragmentWrapper.__init__c             C   s   t | j|S )N)getattrr0   )r   r)   r   r   r   __getattr___   s    zFragmentWrapper.__getattr__c             C   s6   | j 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)r0   )r   r"   r   r   r   r#   j   s    zFragmentWrapper.__getitem__c             C   s
   t | jS )N)boolr0   )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)strr0   )r   r   r   r   __str__s   s    zFragmentWrapper.__str__c             C   s
   t | jS )N)r:   r0   )r   r   r   r   __unicode__v   s    zFragmentWrapper.__unicode__c             C   s
   t | jS )N)r5   r0   )r   r   r   r   r$   y   s    zFragmentWrapper.__len__N)r%   r&   r'   r!   r3   r   r#   r8   r9   r;   r<   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j| | t	j
 | _d S )N)r   listsetfragmentChildrenr+   r   r
   NonRecursiveTreeWalkerr!   r   InfosetFilterfilter)r   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 rt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jt|j}| r|j \}}nd }t|j}i }xbt|jj D ]P\}}t|}t|}tj|}|rX|||jd|jdf< n||d |f< qW tj|| j j!||t"|dkp|jfS d S )
Nr   r   z%Text nodes are text or tail, found %stagr	   r   r   )r   r   )#r   tupleAssertionErrorr
   ZTEXTr   r2   r   ZDOCUMENTr   ZDOCTYPEr)   r   r*   r,   r1   r0   rE   r   CommentCOMMENTr   EntityZENTITYr   matchgroupsr>   attribitemsgroupZELEMENTrC   fromXmlNamer5   )	r   r    r"   rL   	namespacerE   attrsr)   valuer   r   r   getNodeDetails   s:    




zTreeWalker.getNodeDetailsc             C   sD   t |t stdt|s*|js*td|jr8|dfS |d S d S )NzText nodes have no childrenzNode has no childrenr   r   )r   rG   rH   r5   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|j S |jrZ|dfS |j S )Nr   r   z%Text nodes are text or tail, found %sr   )r   r   )r   rG   rH   r5   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 |j S )Nr   r   z%Text nodes are text or tail, found %s)r   r   )r   rG   rH   r@   r9   )r   r    r"   r   r   r   getParentNode   s    

zTreeWalker.getParentNodeN)r%   r&   r'   r!   rU   rV   rW   rX   r   r   r   r   r=   }   s
   )	r=   N)
__future__r   r   r   sixr   lxmlr   Ztreebuilders.etreer    r
   r   r   objectr   r   r+   r,   rA   r=   r   r   r   r   <module>   s   	&	+