3
T �\Y � @ s� d Z ddlmZmZ ddlZddlZddlmZ ddl m
Z
m Z m Z ej
d dkZdZejZejd ed
d � G d d
� d
ej�ZG dd� dej�Ze� ZejZddd�Zddd�Zd dd�ZeZ dd� Z dS ) zkDEPRECATED Example code for lxml.etree protection
The code has NO protection against decompression bombs.
� )�print_function�absolute_importN)�etree� )� DTDForbidden�EntitiesForbidden�NotSupportedError� z
lxml.etreezOdefusedxml.lxml is no longer supported and will be removed in a future release.� )�category�
stacklevelc s� e Zd ZdZf ZejZdd� Z� fdd�Z d� fdd � Z
d� fd
d � Z d� fd d
� Z d� fdd� Z
� fdd�Zd� fdd� Z� ZS )�RestrictedElementzJA restricted Element class that filters out instances of some classes
c c s* | j }x |D ]}t||�r q |V q W d S )N)� blacklist�
isinstance)�self�iteratorr Zchild� r �/usr/lib/python3.6/lxml.py�_filter) s
zRestrictedElement._filterc s t t| �j� }| j|�S )N)�superr
�__iter__r )r r )� __class__r r r 0 s zRestrictedElement.__iter__NFc s t t| �j||d�}| j|�S )N)�tag�reversed)r r
� iterchildrenr )r r r r )r r r r 4 s z RestrictedElement.iterchildrenc s t t| �j|d|i�}| j|�S )Nr )r r
�iterr )r r �tagsr )r r r r 8 s zRestrictedElement.iterc s t t| �j|d|i�}| j|�S )Nr )r r
�iterdescendantsr )r r r r )r r r r < s z!RestrictedElement.iterdescendantsc s t t| �j||d�}| j|�S )N)r � preceding)r r
� itersiblingsr )r r r r )r r r r @ s z RestrictedElement.itersiblingsc s t t| �j� }t| j|��S )N)r r
r �listr )r r )r r r � getchildrenD s z RestrictedElement.getchildrenc s t t| �j|�}| j|�S )N)r r
� getiteratorr )r r r )r r r r" H s z RestrictedElement.getiterator)NF)N)N)NF)N)�__name__�
__module__� __qualname__�__doc__� __slots__�_etreeZ_Entityr r r r r r r r! r" �
__classcell__r r )r r r
! s r
c @ s4 e Zd ZdZddiZeZdd� Zdd� Zdd � Z d
S ) �GlobalParserTLSz5Thread local context for custom parser instances
Zresolve_entitiesFc C s8 t jf | j�}| j}| jd k r4t j|d�}|j|� |S )N)�element)r( Z XMLParser�
parser_config�
element_classZElementDefaultClassLookupZset_element_class_lookup)r �parserr- �lookupr r r �createDefaultParserY s
z#GlobalParserTLS.createDefaultParserc C s
|| _ d S )N)�_default_parser)r r. r r r �setDefaultParsera s z GlobalParserTLS.setDefaultParserc C s* t | dd �}|d kr&| j� }| j|� |S )Nr1 )�getattrr0 r2 )r r. r r r �getDefaultParserd s
z GlobalParserTLS.getDefaultParserN)
r# r$ r% r& r, r
r- r0 r2 r4 r r r r r* M s r* FTc C s� | j }|jr4|r"t|j|j|j��|r4t r4td��|r�xF|j|jfD ]6}|dkrTqFx&|j � D ]}t
|j |j dddd��q^W qFW dS )z�Check docinfo of an element tree for DTD and entity declarations
The check for entity declarations needs lxml 3 or newer. lxml 2.x does
not support dtd.iterentities().
z3Unable to check for entity declarations in lxml 2.xN)
�docinfoZdoctyper Z
system_urlZ public_id�LXML3r Z internalDTDZ externalDTDZ iterentitiesr �nameZcontent)� elementtree�
forbid_dtd�forbid_entitiesr5 ZdtdZentityr r r �
check_docinfop s
r; c C s. |d krt � }tj| ||d�}t|||� |S )N)�base_url)r4 r( �parser; )�sourcer. r< r9 r: r8 r r r r= � s
r= c C s6 |d krt � }tj| ||d�}|j� }t|||� |S )N)r< )r4 r( �
fromstringZ getroottreer; )�textr. r< r9 r: Z rootelementr8 r r r r? � s r? c O s t d��d S )Nz*defused lxml.etree.iterparse not available)r )�args�kwargsr r r � iterparse� s rC )FT)NNFT)NNFT) r& Z
__future__r r Z threading�warningsZlxmlr r( �commonr r r Z LXML_VERSIONr6 Z
__origin__Ztostring�warn�DeprecationWarningZ ElementBaser
Zlocalr* Z _parser_tlsr4 r; r= r? ZXMLrC r r r r �<module> s* ,