3
Pf @ 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 | jddS d S )Nasciistrict)
isinstancer decode)s r /usr/lib/python3.6/etree_lxml.py
ensure_str s
r c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
Rootc C s || _ g | _y:|jjrD| jjt| t|jjt|jjt|jj W n t
k
rZ Y nX y |j }W n t
k
r |}Y nX x|j d k r|j }qW x |d k r| jj| |j
}qW d | _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 )N)r )r keyr r r __getitem__1 s zRoot.__getitem__c C s d S )Nr )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 )N) root_nodenamer system_idr r )r r& r' r r( r r r r < s zDoctype.__init__c C s | j jd S )Nr )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).0Zchild)r 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 )Nr )r r r r r N s zFragmentRoot.getnextN)r# r$ r% r r 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_rootr- r r r r S s zFragmentWrapper.__init__c C s t | j|S )N)getattrr- )r r' r r r __getattr___ s zFragmentWrapper.__getattr__c C s6 | j j}|j| }|t|d k r.||d S d S d S )Nr )r& r indexlen)r Zsiblingsidxr r r r b s
zFragmentWrapper.getnextc C s
| j | S )N)r- )r r r r r r! j s zFragmentWrapper.__getitem__c C s
t | jS )N)boolr- )r r r r __bool__m s zFragmentWrapper.__bool__c C s d S )Nr )r r r r getparentp s zFragmentWrapper.getparentc C s
t | jS )N)strr- )r r r r __str__s s zFragmentWrapper.__str__c C s
t | jS )N)r7 r- )r r r r __unicode__v s zFragmentWrapper.__unicode__c C s
t | jS )N)r2 r- )r r r r r" y s zFragmentWrapper.__len__N) r# r$ r% r r0 r r! r5 r6 r8 r9 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|}tjj| | t j
| _ d S )N) r listsetfragmentChildrenr) r r
NonRecursiveTreeWalkerr r Z
InfosetFilterfilter)r Ztreer r r r ~ s
zTreeWalker.__init__c C s t |tr&|\}}tjtt||fS t |tr8tjfS t |trVtj |j
|j |j fS t |t
r|t|d r|tjt|jfS |jtjkrtjt|jfS |jtjkrtjt|jdd fS tjt|j}|r|j \}}nd }t|j}i }xbt|jj D ]P\}}t|}t|}tj|}|rB|||j d|j df< n ||d |f< q W tj || j j ||t!|dkpx|jfS d S )Ntagr r r )"r tupler
ZTEXTr r/ r ZDOCUMENTr ZDOCTYPEr' r r( r* r. r- r@ r CommentCOMMENTr ZEntityZENTITYr matchgroupsr; ZattribitemsgroupZELEMENTr? Z fromXmlNamer2 ) r r r rE namespacer@ Zattrsr' valuer r r getNodeDetails s8
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 |tr8|\}}|dkr0t|r*|d S d S n|j S |jrF|dfS |j S )Nr r r )r rB r2 r r )r r r r r r getNextSibling s
zTreeWalker.getNextSiblingc C s6 t |tr |\}}|dkr.|S n|| jkr.d S |j S )Nr )r rB r= r6 )r r r r r r
getParentNode s
zTreeWalker.getParentNodeN)r# r$ r% r rK rL rM rN 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* r> r: r r r r s & +