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 eZG d
d deZG dd deZG dd deZG dd de jZdS ) )absolute_importdivisionunicode_literals) text_type)etree )
tag_regexp )base) _ihatexmlc C s* | d kr d S t | tr| S | ddS d S )Nasciistrict)
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| jt| t|jjt|jjt|jj W n t
k
rZ Y nX z | }W n t
k
r |}Y nX | d k r| }q|d k r| j| |
}qd | _d | _d S N)Z elementtreechildrenZdocinfoZ internalDTDappendDoctyper Z root_name public_idZ
system_urlAttributeErrorZgetrootZ getpreviousgetnexttexttail)selfZetnoder 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_nodenamer 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.getnextNr' 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| jdr(t| jj| _nd | _t| jdrJt| jj| _nd | _d S )Nr r )r* objhasattrr 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 indexlen)r Zsiblingsidxr 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 | jS 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 | jS r strr2 r$ r r r __str__s s zFragmentWrapper.__str__c C s
t | jS r r< r$ r r r __unicode__v s zFragmentWrapper.__unicode__c C s
t | jS 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 |tr t|| _t|}nt | _t|}tj| | t
| _ d S r ) r listsetfragmentChildrenr. r r
NonRecursiveTreeWalkerr r Z
InfosetFilterfilter)r Ztreer r r r ~ s
zTreeWalker.__init__c C s t |tr:|\}}|dks&td| tjtt||fS t |trLtjfS t |t rjtj
|j |j |j
fS t |trt|dstjt|jfS |jtjkrtjt|jfS |jtjkrtjt|jdd fS tt|j}| r| \}}nd }t|j}i }t|j D ]P\}}t|}t|}t|}|rT||| d| df< n ||d |f< qtj || j !||t"|dkp|jfS d S )Nr r %Text nodes are text or tail, found %stagr r r )#r tupleAssertionErrorr
ZTEXTr r4 r ZDOCUMENTr ZDOCTYPEr+ r r, r/ r3 r2 rH r CommentCOMMENTr ZEntityZENTITYr matchgroupsrA ZattribitemsgroupZELEMENTrE Z fromXmlNamer7 ) r r r" rN namespacerH Zattrsr+ valuer r r getNodeDetails s>
zTreeWalker.getNodeDetailsc C sB t |trtdt|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 |trL|\}}|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 |tr4|\}}|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 Zpip._vendor.sixr Zlxmlr Ztreebuilders.etreer r
r r objectr r r. r/ rD r@ r r r r s & +