a
lj @ s d dl mZmZmZ d dlmZ d dlmZ d dlm Z ddl
m Z ddl m
Z
dd l mZ d
d ZG d d
d
eZG dd deZG dd deZG dd deZG dd de
jZdS ) )absolute_importdivisionunicode_literals) text_type) OrderedDict)etree )
tag_regexp )base) _ihatexmlc C s* | d u r d S t | tr| S | ddS d S )Nasciistrict)
isinstancer decode)s r O/usr/lib/python3.9/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
yX Y n0 z | }W n t
y| |}Y n0 | d ur| }q~|d ur| j| |
}qd | _d | _d S N)Z elementtreechildrenZdocinfoZ internalDTDappendDoctyper Z root_name public_idZ
system_urlAttributeErrorZgetrootZ getpreviousgetnexttexttail)selfetnoder r r __init__ s,
z
Root.__init__c C s
| j | S r )r r keyr r r __getitem__3 s zRoot.__getitem__c C s d S r r r r r r r 6 s z Root.getnextc C s dS Nr
r r& r r r __len__9 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 G 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).0childr& r r
M z)FragmentRoot.__init__..)r r r )r r r r& r r" L s zFragmentRoot.__init__c C s d S r r r& r r r r P s zFragmentRoot.getnextNr/ r r r r r0 K s r0 c @ sT e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dS )r1 c C sT || _ || _t| jdr(t| jj| _nd | _t| jdrJt| jj| _nd | _d S )Nr r )r, objhasattrr r r )r Z
fragment_rootr6 r r r r" U s zFragmentWrapper.__init__c C s t | j|S r )getattrr6 )r r- r r r __getattr__a 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 d s
zFragmentWrapper.getnextc C s
| j | S r )r6 r# r r r r% l s zFragmentWrapper.__getitem__c C s
t | jS r )boolr6 r& r r r __bool__o s zFragmentWrapper.__bool__c C s d S r r r& r r r getparentr s zFragmentWrapper.getparentc C s
t | jS r strr6 r& r r r __str__u s zFragmentWrapper.__str__c C s
t | jS r r@ r& r r r __unicode__x s zFragmentWrapper.__unicode__c C s
t | jS r )r; r6 r& r r r r( { s zFragmentWrapper.__len__N) r) r* r+ r" r9 r r% r> r? rB rC r( r r r r r1 T s r1 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 listsetfragmentChildrenr0 r r NonRecursiveTreeWalkerr" r Z
InfosetFilterfilter)r Ztreer r r r" s
zTreeWalker.__init__c C s t |tr:|\}}|dv s&J d| tjtt||fS t |trLtjfS t |trjtj |j
|j |j fS t |t
rt|dstjt|jfS |jtjkrtjt|jfS |jtjkrtjt|jdd fS tt|j}| r| \}}nd }t|j}t }t|j D ]P\}}t|}t|}t|}|rV||| 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 tupler ZTEXTr r8 r ZDOCUMENTr ZDOCTYPEr- r r. r1 r7 r6 rL r CommentCOMMENTr ZEntityZENTITYr matchgroupsr rE ZattribitemsgroupZELEMENTrI Z fromXmlNamer; ) r r! r$ rQ namespacerL attrsr- valuer r r getNodeDetails s<
zTreeWalker.getNodeDetailsc C sB t |trJ dt|s(|js(J d|jr6|dfS |d S d S )NzText nodes have no childrenzNode has no childrenr r )r rN r; r )r r! r r r
getFirstChild s
zTreeWalker.getFirstChildc C sb t |trL|\}}|dv s&J d| |dkrDt|r>|d S d S n| S |jrZ|dfS | S )NrJ rK r r r )r rN r; r r r r! r$ r r r getNextSibling s
zTreeWalker.getNextSiblingc C sJ t |tr4|\}}|dv s&J d| |dkrB|S n|| jv rBd S | S )NrJ rK r )r rN rG r? rZ r r r
getParentNode s
zTreeWalker.getParentNodeN)r) r* r+ r" rX rY r[ r\ r r r r rD s
) rD N)Z
__future__r r r Zpip._vendor.sixr collectionsr Zlxmlr Ztreebuilders.etreer r r r objectr r r0 r1 rH rD r r r r s & +