Edit File: etree_lxml.cpython-36.pyc
3 )�]b� � @ 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 � <listcomp>K s z)FragmentRoot.__init__.<locals>.<listcomp>)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:|\}}|dks&td| ��tjtt||��fS t |t�rLtjfS t |t �rjtj |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|j j � �D ]P\}}t|�}t|�}tj|�}|�rX|||j d�|j d�f< n ||d |f< �qW tj || j j!|�|t"|�dk�p�|jfS d S ) Nr r z%Text nodes are text or tail, found %s�tagr r r )r r ���)#r �tuple�AssertionErrorr 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 rF � namespacer@ Zattrsr' �valuer r r �getNodeDetails� s: zTreeWalker.getNodeDetailsc C sD t |t� std��t|�s*|js*td��|jr8|dfS |d S d S )NzText nodes have no childrenzNode has no childrenr r )r rB rC r2 r )r r r r r � getFirstChild� s zTreeWalker.getFirstChildc C sb t |t�rL|\}}|dks&td| ��|dkrDt|�r>|d S d S n|j� S |jrZ|dfS |j� S )Nr r z%Text nodes are text or tail, found %sr )r r )r rB rC r2 r r )r r r r r r �getNextSibling� s zTreeWalker.getNextSiblingc C sJ t |t�r4|\}}|dks&td| ��|dkrB|S n|| jkrBd S |j� S )Nr r z%Text nodes are text or tail, found %s)r r )r rB rC r= r6 )r r r r r r � getParentNode� s zTreeWalker.getParentNodeN)r# r$ r% r rL rM rN rO 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 �<module> s & +