File: //usr/lib/python3/dist-packages/html5lib/treewalkers/__pycache__/etree_lxml.cpython-38.pyc
U
63)Z� � @ s� d dl mZmZmZ d dlmZ d dlmZ ddlm Z ddl
m Z ddl
m Z d d
� Z
G d d � d e�ZG d
d� de�ZG dd� de�ZG dd� de�ZG dd� de j�ZdS )� )�absolute_import�division�unicode_literals)� text_type)�etree� )�
tag_regexp� )�base)� _ihatexmlc C s* | d kr d S t | t�r| S | �dd�S d S )N�ascii�strict)�
isinstancer �decode)�s� r �A/usr/lib/python3/dist-packages/html5lib/treewalkers/etree_lxml.py�
ensure_str s
r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�Rootc C s� || _ g | _z:|jjrD| j�t| t|jj�t|jj�t|jj ��� W n t
k
rZ Y nX z |� � }W n t
k
r� |}Y nX |� � d k r�|� � }q�|d k r�| j�|� |�
� }q�d | _d | _d S �N)Z elementtree�childrenZdocinfoZ internalDTD�append�Doctyper Z root_name� public_idZ
system_url�AttributeErrorZgetrootZ getprevious�getnext�text�tail)�selfZet�noder r r �__init__ s,
�
z
Root.__init__c C s
| j | S r )r �r �keyr r r � __getitem__1 s zRoot.__getitem__c C s d S r r �r r r r r 4 s z Root.getnextc C s dS �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 Zd Zdd� Zdd� ZdS )r c C s( || _ || _|| _|| _d | _d | _d S r )� root_node�namer � system_idr r )r r* r+ r r, r r r r < s zDoctype.__init__c C s | j jd S r% )r* r r$ r r r r E s zDoctype.getnextN�r' r( r) r r r r r r r ; s r c @ s e Zd Zdd� Zdd� ZdS )� FragmentRootc s$ � fdd�|D �� _ d � _� _d S )Nc s g | ]}t � |��qS r )�FragmentWrapper)�.0Zchildr$ r r �
<listcomp>K s z)FragmentRoot.__init__.<locals>.<listcomp>)r r r )r r r r$ r r J s zFragmentRoot.__init__c C s d S r r r$ r r r r N s zFragmentRoot.getnextNr- r r r r r. I s r. c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )r/ c C sT || _ || _t| jd�r(t| jj�| _nd | _t| jd�rJt| jj�| _nd | _d S )Nr r )r* �obj�hasattrr r r )r Z
fragment_rootr2 r r r r S s zFragmentWrapper.__init__c C s t | j|�S r )�getattrr2 )r r+ r r r � __getattr___ s zFragmentWrapper.__getattr__c C s6 | j j}|�| �}|t|�d k r.||d S d S d S r% )r* r �index�len)r Zsiblings�idxr r r r b s
zFragmentWrapper.getnextc C s
| j | S r )r2 r! r r r r# j s zFragmentWrapper.__getitem__c C s
t | j�S r )�boolr2 r$ r r r �__bool__m s zFragmentWrapper.__bool__c C s d S r r r$ r r r � getparentp s zFragmentWrapper.getparentc C s
t | j�S r ��strr2 r$ r r r �__str__s s zFragmentWrapper.__str__c C s
t | j�S r r< r$ r r r � __unicode__v s zFragmentWrapper.__unicode__c C s
t | j�S r )r7 r2 r$ r r r r& y s zFragmentWrapper.__len__N) r' r( r) r r5 r r# r: r; r> r? r& r r r r r/ R s r/ c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �
TreeWalkerc C sJ t |t�r t|�| _t|�}nt� | _t|�}tj�| |� t �
� | _ d S r ) r �list�set�fragmentChildrenr. r r
�NonRecursiveTreeWalkerr r Z
InfosetFilter�filter)r Ztreer r r r ~ s
zTreeWalker.__init__c C s� t |t�r:|\}}|dks&td| ��tjtt||��fS t |t�rLtjfS t |t �rjtj
|j |j |j
fS t |t�r�t|d�s�tjt|j�fS |jtjkr�tjt|j�fS |jtjkr�tjt|j�dd� fS t�t|j��}|� r�|�� \}}nd }t|j�}i }t|j � � �D ]P\}}t|�}t|�}t�|�}|�rT|||� d�|� d�f< n ||d |f< �qtj || j �!|�|t"|�dk�p�|jfS d S )N�r r �%Text nodes are text or tail, found %s�tagr ���r r )#r �tuple�AssertionErrorr
ZTEXTr r4 r ZDOCUMENTr ZDOCTYPEr+ r r, r/ r3 r2 rH r �Comment�COMMENTr ZEntityZENTITYr �match�groupsrA Zattrib�items�groupZELEMENTrE Z fromXmlNamer7 ) r r r" rN � namespacerH Zattrsr+ �valuer r r �getNodeDetails� s>
�zTreeWalker.getNodeDetailsc C sB t |t�rtd��t|�s(|js(td��|jr6|dfS |d S d S )NzText nodes have no childrenzNode has no childrenr r )r rJ rK r7 r )r r r r r �
getFirstChild� s
zTreeWalker.getFirstChildc C sb t |t�rL|\}}|dks&td| ��|dkrDt|�r>|d S d S n|�� S |jrZ|dfS |�� S )NrF rG r r r )r rJ rK r7 r r �r r r" r r r �getNextSibling� s
zTreeWalker.getNextSiblingc C sJ t |t�r4|\}}|dks&td| ��|dkrB|S n|| jkrBd S |�� S )NrF rG r )r rJ rK rC r; rV r r r �
getParentNode� s
zTreeWalker.getParentNodeN)r' r( r) r rT rU rW rX r r r r r@ } s
) r@ N)Z
__future__r r r Zsixr Zlxmlr Ztreebuilders.etreer � r
r r �objectr r r. r/ rD r@ r r r r �<module> s & +