U
    ¼Ê¦iX=  ã                   @   s¦  d Z dZddlT ddlmZmZ ddlmZ ddlmZmZ ddlm	Z	 ddlm
Z
mZ dd	lZejd	d
… dkr|edd	ƒ‚[zddlmZ W n ek
r¬   edd	ƒ‚Y nX eedƒsÂedd	ƒ‚ddlmZmZmZ ejZejZzdd	lZW n ek
�r   dd„ ZY nX dd	lZejZ[[G dd„ dƒZG dd„ dejƒZG dd„ dej ejƒZ!dd„ Z"e#dk�r¢dd	l$Z%e"ƒ Z&e& 'e%j(j )¡ ¡ e& *e%j( +¡ ¡ e& ,d¡ d	S )z]
SAX driver for the pyexpat C module.  This driver works with
pyexpat.__version__ == '2.22'.
z0.20é    )Ú*)Úfeature_validationÚfeature_namespaces)Úfeature_namespace_prefixes)Úfeature_external_gesÚfeature_external_pes)Úfeature_string_interning)Úproperty_xml_stringÚproperty_interning_dictNé   Zjavazexpat not available in Java)Úexpatzexpat not supportedÚParserCreate)Ú	xmlreaderÚsaxutilsÚhandlerc                 C   s   | S ©N© )Úor   r   ú)/usr/lib/python3.8/xml/sax/expatreader.pyÚ_mkproxy'   s    r   c                   @   s   e Zd ZdS )Ú_ClosedParserN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   .   s   r   c                   @   s8   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ ZdS )ÚExpatLocatorz¼Locator for use with the ExpatParser class.

    This uses a weak reference to the parser object to avoid creating
    a circular reference between the parser and the content handler.
    c                 C   s   t |ƒ| _d S r   )r   Ú_ref©ÚselfÚparserr   r   r   Ú__init__9   s    zExpatLocator.__init__c                 C   s   | j }|jd krd S |jjS r   )r   Ú_parserÚErrorColumnNumberr   r   r   r   ÚgetColumnNumber<   s    
zExpatLocator.getColumnNumberc                 C   s   | j }|jd krdS |jjS ©Né   )r   r    ÚErrorLineNumberr   r   r   r   ÚgetLineNumberB   s    
zExpatLocator.getLineNumberc                 C   s   | j }|d krd S |j ¡ S r   )r   Ú_sourceÚgetPublicIdr   r   r   r   r(   H   s    zExpatLocator.getPublicIdc                 C   s   | j }|d krd S |j ¡ S r   )r   r'   ÚgetSystemIdr   r   r   r   r)   N   s    zExpatLocator.getSystemIdN)	r   r   r   Ú__doc__r   r"   r&   r(   r)   r   r   r   r   r   3   s   r   c                   @   s  e Zd ZdZdCdd„Zdd„ Zdd	„ Zd
d„ Zdd„ Zdd„ Z	dd„ Z
dd„ ZdDdd„Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zd d!„ Zd"d#„ Zd$d%„ Zd&d'„ Zd(d)„ Zd*d+„ Zd,d-„ Zd.d/„ Zd0d1„ Zd2d3„ Zd4d5„ Zd6d7„ Zd8d9„ Zd:d;„ Zd<d=„ Z d>d?„ Z!d@dA„ Z"dBS )EÚExpatParserz$SAX driver for the pyexpat C module.r   éìÿ  c                 C   sF   t j | |¡ t  ¡ | _d | _|| _d | _d| _g | _	d| _
d | _d S )Nr   )r   ÚIncrementalParserr   ZInputSourcer'   r    Ú_namespacesÚ_lex_handler_propÚ_parsingÚ_entity_stackÚ_external_gesÚ
_interning)r   ZnamespaceHandlingÚbufsizer   r   r   r   Z   s    
zExpatParser.__init__c                 C   sV   t  |¡}|| _z*|  ¡  | j t| ƒ¡ tj 	| |¡ W n   |  
¡  ‚ Y nX dS )z3Parse an XML document from a URL or an InputSource.N)r   Úprepare_input_sourcer'   ÚresetÚ_cont_handlerZsetDocumentLocatorr   r   r-   ÚparseÚ_close_source©r   Úsourcer   r   r   r8   g   s    
zExpatParser.parsec                 C   s    |  ¡ d k	r| j |  ¡ ¡ d S r   )r)   r    ZSetBaser:   r   r   r   ÚprepareParserw   s    zExpatParser.prepareParserc                 C   s    t j | |¡ | jr|  ¡  d S r   )r   r-   ÚsetContentHandlerr0   Ú_reset_cont_handler)r   r   r   r   r   r=   }   s    zExpatParser.setContentHandlerc                 C   sP   |t kr| jS |tkr | jd k	S |tttfkr2dS |tkr@| jS t	d| ƒ‚d S )Nr   úFeature '%s' not recognized)
r   r.   r   r3   r   r   r   r   r2   ÚSAXNotRecognizedException©r   Únamer   r   r   Ú
getFeature‚   s    
ÿzExpatParser.getFeaturec                 C   s¦   | j rtdƒ‚|tkr|| _n„|tkr.|| _nt|tkrT|rL| jd krRi | _q¢d | _nN|tkrj|r¢tdƒ‚n8|t	kr€|r¢tdƒ‚n"|t
kr–|r¢tdƒ‚ntd| ƒ‚d S )Nz!Cannot set features while parsingz!expat does not support validationz/expat does not read external parameter entitiesz(expat does not report namespace prefixesr?   )r0   ÚSAXNotSupportedExceptionr   r.   r   r2   r   r3   r   r   r   r@   )r   rB   Ústater   r   r   Ú
setFeatureŽ   s:    
ÿÿÿÿzExpatParser.setFeaturec                 C   sd   |t jkr| jS |tkr| jS |tkrT| jrLt| jdƒrB| j ¡ S t	dƒ‚nt
dƒ‚t	d| ƒ‚d S )NÚGetInputContextz=This version of expat does not support getting the XML stringz.XML string cannot be returned when not parsingúProperty '%s' not recognized)r   Úproperty_lexical_handlerr/   r
   r3   r	   r    ÚhasattrrG   r@   rD   rA   r   r   r   ÚgetProperty¬   s    

ÿÿzExpatParser.getPropertyc                 C   sV   |t jkr || _| jrR|  ¡  n2|tkr0|| _n"|tkrFtd| ƒ‚nt	d| ƒ‚d S )NzProperty '%s' cannot be setrH   )
r   rI   r/   r0   Ú_reset_lex_handler_propr
   r3   r	   rD   r@   )r   rB   Úvaluer   r   r   ÚsetProperty¾   s    

ÿÿzExpatParser.setPropertyc              
   C   sz   | j s|  ¡  d| _ | j ¡  z| j ||¡ W nD tjk
rt } z$tt 	|j
¡|| ƒ}| j |¡ W 5 d }~X Y nX d S r#   )r0   r6   r7   ZstartDocumentr    ZParser   ÚerrorZSAXParseExceptionZErrorStringÚcodeZ_err_handlerZ
fatalError)r   ÚdataÚisFinalÚeÚexcr   r   r   ÚfeedÎ   s    
zExpatParser.feedc                 C   sB   | j }z| ¡ }|d k	r | ¡  W 5 | ¡ }|d k	r<| ¡  X d S r   )r'   ZgetByteStreamÚcloseZgetCharacterStream)r   r;   Úfiler   r   r   r9   ß   s    zExpatParser._close_sourcec                 C   sˆ   | j s| jd kst| jtƒr d S z(| jddd� | j	 
¡  d| _d | _W 5 d| _| jd k	rztƒ }| jj|_| jj|_|| _|  ¡  X d S )Nr   Ú r$   )rR   )r1   r    Ú
isinstancer   r0   r!   r%   r9   rU   r7   ZendDocumentr   r   r   r   rV   ê   s     
ÿ




zExpatParser.closec                 C   s   | j j| j_| j j| j_d S r   )r7   ÚprocessingInstructionr    ZProcessingInstructionHandlerÚ
charactersZCharacterDataHandler©r   r   r   r   r>   ÿ   s    ÿzExpatParser._reset_cont_handlerc                 C   s`   | j }| j}|d kr4d |_d |_d |_d |_d |_n(|j|_|j|_|j	|_| j
|_|j|_d S r   )r/   r    ZCommentHandlerZStartCdataSectionHandlerZEndCdataSectionHandlerZStartDoctypeDeclHandlerZEndDoctypeDeclHandlerZcommentZ
startCDATAZendCDATAÚstart_doctype_declZendDTD)r   Zlexr   r   r   r   rL     s    z#ExpatParser._reset_lex_handler_propc                 C   sú   | j r>tj| j ¡ d| jd�| _d| j_| j| j_	| j
| j_n,tj| j ¡ | jd�| _| j| j_	| j| j_|  ¡  | j| j_| j| j_| j| j_| j| j_d | _| jr®|  ¡  | j| j_z| j| j_W n tk
rÚ   Y nX | j tj ¡ d| _!g | _"d S )Nú )Úinternr$   r   )#r.   r   r   r'   ZgetEncodingr3   r    Znamespace_prefixesÚstart_element_nsZStartElementHandlerÚend_element_nsZEndElementHandlerÚstart_elementÚend_elementr>   Úunparsed_entity_declZUnparsedEntityDeclHandlerÚnotation_declZNotationDeclHandlerÚstart_namespace_declZStartNamespaceDeclHandlerÚend_namespace_declZEndNamespaceDeclHandlerZ_decl_handler_propr/   rL   Úexternal_entity_refZExternalEntityRefHandlerÚskipped_entity_handlerZSkippedEntityHandlerÚAttributeErrorZSetParamEntityParsingZ*XML_PARAM_ENTITY_PARSING_UNLESS_STANDALONEr0   r1   r\   r   r   r   r6     s<    ÿ
ÿ






ÿzExpatParser.resetc                 C   s   | j d krd S | j jS r   )r    r!   r\   r   r   r   r"   ;  s    
zExpatParser.getColumnNumberc                 C   s   | j d krdS | j jS r#   )r    r%   r\   r   r   r   r&   @  s    
zExpatParser.getLineNumberc                 C   s
   | j  ¡ S r   )r'   r(   r\   r   r   r   r(   E  s    zExpatParser.getPublicIdc                 C   s
   | j  ¡ S r   )r'   r)   r\   r   r   r   r)   H  s    zExpatParser.getSystemIdc                 C   s   | j  |t|ƒ¡ d S r   )r7   ZstartElementÚAttributesImpl)r   rB   Úattrsr   r   r   rb   L  s    zExpatParser.start_elementc                 C   s   | j  |¡ d S r   )r7   Z
endElementrA   r   r   r   rc   O  s    zExpatParser.end_elementc                 C   sì   |  ¡ }t|ƒdkrd |f}n&t|ƒdkr<|d |d f}nt|ƒ}i }i }| ¡ D ]|\}}|  ¡ }t|ƒ}	|	dkr‚|}
d |f}n>|	dkr°d|d |d f }
|d |d f}n|d }
t|ƒ}|||< |
||< qT| j |d t||ƒ¡ d S )Nr$   é   r   z%s:%sé   )ÚsplitÚlenÚtupleÚitemsr7   ZstartElementNSÚAttributesNSImpl)r   rB   rl   ÚpairZnewattrsZqnamesZanamerM   ÚpartsZlengthZqnameZapairr   r   r   r`   R  s0    



ÿzExpatParser.start_element_nsc                 C   sV   |  ¡ }t|ƒdkrd |f}n&t|ƒdkr<|d |d f}nt|ƒ}| j |d ¡ d S )Nr$   rm   r   )ro   rp   rq   r7   ZendElementNS)r   rB   rt   r   r   r   ra   t  s    
zExpatParser.end_element_nsc                 C   s   | j  ||¡ d S r   )r7   rZ   )r   ÚtargetrQ   r   r   r   Úprocessing_instruction€  s    z"ExpatParser.processing_instructionc                 C   s   | j  |¡ d S r   )r7   r[   )r   rQ   r   r   r   Úcharacter_data„  s    zExpatParser.character_datac                 C   s   | j  ||¡ d S r   )r7   ZstartPrefixMapping)r   ÚprefixZurir   r   r   rf   ‡  s    z ExpatParser.start_namespace_declc                 C   s   | j  |¡ d S r   )r7   ZendPrefixMapping)r   ry   r   r   r   rg   Š  s    zExpatParser.end_namespace_declc                 C   s   | j  |||¡ d S r   )r/   ZstartDTD)r   rB   ÚsysidÚpubidZhas_internal_subsetr   r   r   r]   �  s    zExpatParser.start_doctype_declc                 C   s   | j  ||||¡ d S r   )Ú_dtd_handlerZunparsedEntityDecl)r   rB   Úbaserz   r{   Znotation_namer   r   r   rd   �  s    z ExpatParser.unparsed_entity_declc                 C   s   | j  |||¡ d S r   )r|   ZnotationDecl)r   rB   r}   rz   r{   r   r   r   re   “  s    zExpatParser.notation_declc                 C   s–   | j s
dS | j ||¡}t || j ¡ p*d¡}| j | j	| jf¡ | j	 
|¡| _	|| _ztj | |¡ W n   Y dS X | jd \| _	| _| jd= dS )Nr$   rX   r   éÿÿÿÿ)r2   Z_ent_handlerZresolveEntityr   r5   r'   r)   r1   Úappendr    ZExternalEntityParserCreater   r-   r8   )r   Úcontextr}   rz   r{   r;   r   r   r   rh   –  s"    
þzExpatParser.external_entity_refc                 C   s   |rd| }| j  |¡ d S )Nú%)r7   ZskippedEntity)r   rB   Zis_per   r   r   ri   ¬  s    z"ExpatParser.skipped_entity_handlerN)r   r,   )r   )#r   r   r   r*   r   r8   r<   r=   rC   rF   rK   rN   rU   r9   rV   r>   rL   r6   r"   r&   r(   r)   rb   rc   r`   ra   rw   rx   rf   rg   r]   rd   re   rh   ri   r   r   r   r   r+   W   s@   

'"r+   c                  O   s
   t | |ŽS r   )r+   )ÚargsÚkwargsr   r   r   Úcreate_parser´  s    r„   Ú__main__z:http://www.ibiblio.org/xml/examples/shakespeare/hamlet.xml)-r*   ÚversionZxml.sax._exceptionsZxml.sax.handlerr   r   r   r   r   r   r	   r
   ÚsysÚplatformZSAXReaderNotAvailableZxml.parsersr   ÚImportErrorrJ   Zxml.saxr   r   r   rk   rs   Ú_weakrefr   ÚweakrefÚproxyr   ZLocatorr   r-   r+   r„   r   Zxml.sax.saxutilsZxmlÚpr=   ZsaxZXMLGeneratorZsetErrorHandlerZErrorHandlerr8   r   r   r   r   Ú<module>   sN   


$  _
