o
    8F`L                     @   s   d Z ddlmZmZ ddlZddlZddlmZ ddl	m
Z
mZmZ ejd dkZdZejZejd	ed
d G dd dejZG dd dejZe ZejZdddZdddZdddZeZdd ZdS )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 Zd ZdZdZejZdd Z fddZ	d fd	d
	Z
d fdd	Zd fdd	Zd fdd	Z fddZd fdd	Z  ZS )RestrictedElementzEA restricted Element class that filters out instances of some classes c                 c   s(    | j }|D ]}t||rq|V  qd S N)	blacklist
isinstance)selfiteratorr   childr   r   1/usr/lib/python3/dist-packages/defusedxml/lxml.py_filter(   s   
zRestrictedElement._filterc                    s   t t|  }| |S r   )superr   __iter__r   r   r   	__class__r   r   r   /   s   
zRestrictedElement.__iter__NFc                       t t| j||d}| |S )N)tagreversed)r   r   iterchildrenr   )r   r   r   r   r   r   r   r   3      
zRestrictedElement.iterchildrenc                        t t| j|d|i}| |S Nr   )r   r   iterr   r   r   Ztagsr   r   r   r   r#   7      
zRestrictedElement.iterc                    r!   r"   )r   r   iterdescendantsr   r$   r   r   r   r&   ;   r%   z!RestrictedElement.iterdescendantsc                    r   )N)r   	preceding)r   r   itersiblingsr   )r   r   r'   r   r   r   r   r(   ?   r    zRestrictedElement.itersiblingsc                    s   t t|  }t| |S r   )r   r   r   listr   r   r   r   r   getchildrenC   s   zRestrictedElement.getchildrenc                    s   t t| |}| |S r   )r   r   getiteratorr   )r   r   r   r   r   r   r+   G   s   
zRestrictedElement.getiterator)NFr   )__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 Zd ZdZddiZeZdd Zdd Zdd	 Z	d
S )GlobalParserTLSz0Thread local context for custom parser instancesZresolve_entitiesFc                 C   s<   t jdi | j}| j}| jd urt j|d}|| |S )N)elementr   )r1   Z	XMLParserparser_configelement_classZElementDefaultClassLookupZset_element_class_lookup)r   parserr6   lookupr   r   r   createDefaultParserW   s   

z#GlobalParserTLS.createDefaultParserc                 C   s
   || _ d S r   )_default_parserr   r7   r   r   r   setDefaultParser_   s   
z GlobalParserTLS.setDefaultParserc                 C   s*   t | dd }|d u r|  }| | |S )Nr:   )getattrr9   r<   r;   r   r   r   getDefaultParserb   s
   
z GlobalParserTLS.getDefaultParserN)
r,   r-   r.   r/   r5   r   r6   r9   r<   r>   r   r   r   r   r3   L   s    r3   FTc              	   C   s|   | j }|jr|rt|j|j|j|rtstd|r:|j|jfD ]}|du r(q!|	 D ]}t
|j|jddddq!dS dS )zCheck 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)docinfoZdoctyper   Z
system_urlZ	public_idLXML3r   ZinternalDTDZexternalDTDZiterentitiesr   nameZcontent)elementtree
forbid_dtdforbid_entitiesr?   ZdtdZentityr   r   r   check_docinfon   s   rE   c                 C   s.   |d u rt  }tj| ||d}t||| |S N)base_url)r>   r1   parserE   )sourcer7   rG   rC   rD   rB   r   r   r   rH      s
   rH   c                 C   s6   |d u rt  }tj| ||d}| }t||| |S rF   )r>   r1   
fromstringZgetroottreerE   )textr7   rG   rC   rD   ZrootelementrB   r   r   r   rJ      s   rJ   c                  O   s   t d)Nz*defused lxml.etree.iterparse not available)r   )argskwargsr   r   r   	iterparse   s   rN   )FT)NNFT)r/   Z
__future__r   r   Z	threadingwarningsZlxmlr   r1   commonr   r   r   ZLXML_VERSIONr@   
__origin__ZtostringwarnDeprecationWarningZElementBaser   Zlocalr3   Z_parser_tlsr>   rE   rH   rJ   ZXMLrN   r   r   r   r   <module>   s.   +


	