U
��.e� � @ 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 �O/usr/lib/python3.8/site-packages/pip/_vendor/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 �
K s z)FragmentRoot.__init__..)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 sz t |t�r&|\}}tjtt||��fS t |t�r8tjfS t |t�rVtj |j
|j |j fS t |t
�rzt|d�sztjt|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�|�}|�r>|||� d�|� d�f< n ||d |f< � q�tj || j � |�|t!|�dk�pr|jfS d S )N�tagr �����r r )"r �tupler
ZTEXTr r4 r ZDOCUMENTr ZDOCTYPEr+ r r, r/ r3 r2 rF r �Comment�COMMENTr ZEntityZENTITYr �match�groupsrA Zattrib�items�groupZELEMENTrE Z fromXmlNamer7 ) r r r" rK � namespacerF Zattrsr+ �valuer r r �getNodeDetails� s<
�zTreeWalker.getNodeDetailsc C s |j r|dfS |d S d S )Nr r )r )r r r r r �
getFirstChild� s zTreeWalker.getFirstChildc C sN t |t�r8|\}}|dkr0t|�r*|d S d S n|�� S |jrF|dfS |�� S )Nr r r )r rH r7 r r �r r r" r r r �getNextSibling� s
zTreeWalker.getNextSiblingc C s6 t |t�r |\}}|dkr.|S n|| jkr.d S |�� S )Nr )r rH rC r; rS r r r �
getParentNode� s
zTreeWalker.getParentNodeN)r' r( r) r rQ rR rT rU r r r r r@ } s
) r@ N)Z
__future__r r r Zpip._vendor.sixr Zlxmlr Ztreebuilders.etreer � r
r r �objectr r r. r/ rD r@ r r r r � s & +