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 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 | jdd�S d S )N�ascii�strict)�
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|jj�t|jj�t|jj ��� W n t
k
rZ Y nX y |j � }W n t
k
r� |}Y nX x|j � d k r�|j � }q�W x |d k r�| jj|� |j
� }q�W 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 )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_node�namer � 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| jd�r(t| jj�| _nd | _t| jd�rJt| jj�| _nd | _d S )Nr r )r& �obj�hasattrr 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 �index�len)r Zsiblings�idxr 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 | j�S )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 | j�S )N)�strr- )r r r r �__str__s s zFragmentWrapper.__str__c C s
t | j�S )N)r7 r- )r r r r � __unicode__v s zFragmentWrapper.__unicode__c C s
t | j�S )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 |t�r t|�| _t|�}nt� | _t|�}tjj| |� t j
� | _ d S )N) 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&|\}}tjtt||��fS t |t�r8tjfS t |t�rVtj |j
|j |j fS t |t
�r|t|d� r|tjt|j�fS |jtjkr�tjt|j�fS |jtjkr�tjt|j�dd� fS tjt|j��}|r�|j� \}}nd }t|j�}i }xbt|jj � �D ]P\}}t|�}t|�}tj|�}|�rB|||j d�|j d�f< n ||d |f< �q W tj || j j |�|t!|�dk�px|jfS d S )N�tagr r r �����)"r �tupler
ZTEXTr r/ r ZDOCUMENTr ZDOCTYPEr' r r( r* r. r- r@ r �Comment�COMMENTr ZEntityZENTITYr �match�groupsr; Zattrib�items�groupZELEMENTr? 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 |t�r8|\}}|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 |t�r |\}}|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 & +