U
.e*7 @ s d Z ddlmZmZmZ ddlZddlZddlZddlm Z ddl
m Z ddlm
Z
dd lm Z
dd
lmZ ddlm Z d Zed Ze d
jZG dd deZG dd deZdd Zdd ZG dd de jZdS )a Module for supporting the lxml.etree library. The idea here is to use as much
of the native library as possible, without using fragile hacks like custom element
names that break between releases. The downside of this is that we cannot represent
all possible trees; specifically the following are known to cause problems:
Text or comments as siblings of the root element
Docypes with no name
When any of these things occur, we emit a DataLossWarning
)absolute_importdivisionunicode_literalsN )base )DataLossWarning) constants)etree) _ihatexmlTz
{([^}]*)}(.*)Zasdc @ s e Zd Zdd ZdS ) DocumentTypec C s || _ || _|| _d S N)namepublicIdsystemId)selfr r r r P/usr/lib/python3.8/site-packages/pip/_vendor/html5lib/treebuilders/etree_lxml.py__init__# s zDocumentType.__init__N)__name__
__module__ __qualname__r r r r r r " s r c @ s, e Zd Zdd Zdd Zdd ZeeZdS )Documentc C s d | _ g | _d S r
) _elementTree _childNodesr r r r r * s zDocument.__init__c C s | j |j d S r
)r getrootZaddnext_elementr elementr r r appendChild. s zDocument.appendChildc C s | j S r
)r r r r r _getChildNodes1 s zDocument._getChildNodesN)r r r r r r! propertyZ
childNodesr r r r r ) s r c s6 g t jdd d fdd | d dS )NTZpreventDoubleDashCommentsr c
s t | ds t | drd | jjrz| jjsF| jjsFd| jj }nd| jj| jj| jjf }dd|d |f | }| d k r| }q|d k r||d | }qnNt
| t st
| t rd d| | f n"d
| D ]}||d qn| j
tkr`d d| | jf t | d r| jrd d| | jf ntj| j
}|d k r|d
}|d}tj| }dd| | |f n dd| | j
f t | drg }| j D ]d\} }
t| }|d k r8| \}} | } tj| }d|| f } n
| } || |
f qt|D ]&\} }
dd|d | |
f q\| jrd d|d | jf |d7 }| D ]} | | qt | d r| jrd d|d | jf d S )Ntagr z #document
zz|%s%s r z|%s"%s"z#document-fragmentz|%stailr z
|%s<%s %s>z|%s<%s>attribz%s %sz
|%s%s="%s") hasattrappenddocinfo internalDTDZ public_idZ
system_url root_namer Z getpreviousZgetnext
isinstancestrbytesr$ comment_typetextr' etree_builders
tag_regexpmatchgroupr prefixes fromXmlNamer( itemsgroupssorted)
r indentdtd_strZ next_elementZnsmatchnsr$ prefix
attributesr valueZ attr_stringchild
infosetFilterrvserializeElementr r rF ; sx
z(testSerializer..serializeElement
)r )r
InfosetFilterjoinr r rC r testSerializer7 s
F
rK c s$ g fdd| d S )z4Serialize an element and its child nodes to a stringc s t | dsH| jjr:| jjr$| jj}n d| jj } | | n| jtkrf d| j f n~| j
s d| jf n.d dd | j
D } d| j|f | j r | j | D ] }| qĈ d | jf t | d
r| j
r | j
d S ) Nr$ r% z z<%s>r& c S s g | ]\}}d ||f qS )z%s="%s"r ).0r rA r r r
s z6tostring..serializeElement..z<%s %s>zr' )r) r+ r, doctyper- r* r r$ r1 r2 r( rI r9 r' )r r= attrrB rE rF r r rF s,
z"tostring..serializeElement )rI rJ r rP r tostring s rR c sz e Zd ZeZeZdZdZeZ e
Z dddZ dd Z
dd Zd d
Zd d Zd
d ZdddZd fdd Zdd Z ZS ) TreeBuilderNFc s t jt|dtjdd | _|| _G fdddt G fdddj}G fdd d j }|| _
|| _ t j
| | d S )
N)fullTreeTr# c s& e Zd Zd fdd Z fddZdS )z(TreeBuilder.__init__..AttributesNc sr |d kr i }|| _ t| | | D ]F\}}t|trTd|d |d f }n
|}|| j j j|< q&d S N{%s}%sr r )r dictr r9 r. tuplecoerceAttributer( )r r rA keyr rD r r r s
z1TreeBuilder.__init__..Attributes.__init__c sP t | || t|tr4d|d |d f }n
|}|| jjj|< d S rU )rW __setitem__r. rX rY r r( )r rZ rA r r[ r r r\ s
z4TreeBuilder.__init__..Attributes.__setitem__)N)r r r r r\ r r[ r r
Attributes s r] c sx e Zd Z fddZfddZfddZeeeZdd Z fd d
Z eee Z
dfd d
Z fddZ d S )z%TreeBuilder.__init__..Elementc s* |}jj| ||d | | _d S )N) namespace)
coerceElementElementr _attributes)r r r^ r] builderrD r r r s
z.TreeBuilder.__init__..Element.__init__c s$ || _| | j| j| j_d S r
)r_ _nameZ _getETreeTagZ
_namespacer r$ )r r r[ r r _setName s
z.TreeBuilder.__init__..Element._setNamec s | jS r
)r8 rd r r[ r r _getName s z.TreeBuilder.__init__..Element._getNamec S s | j S r
ra r r r r _getAttributes s z4TreeBuilder.__init__..Element._getAttributesc s | || _ d S r
rg )r r@ )r] r r _setAttributes s z4TreeBuilder.__init__..Element._setAttributesNc s |} j| || d S r
)ZcoerceCharactersr`
insertText)r dataZ insertBeforerc rD r r rj s
z0TreeBuilder.__init__..Element.insertTextc s j | | d S r
)r` r )r rB )rc r r r s z1TreeBuilder.__init__..Element.appendChild)N)
r r r r re rf r" r rh ri r@ rj r r rb r r r` s
r` c s8 e Zd Z fddZfddZdd ZeeeZdS )z%TreeBuilder.__init__..Commentc s |} j| | d S r
)
coerceCommentCommentr r rk rl r r r s
z.TreeBuilder.__init__..Comment.__init__c s |}|| j_d S r
)rm r r2 ro r[ r r _setData s
z.TreeBuilder.__init__..Comment._setDatac S s | j jS r
)r r2 r r r r _getData s z.TreeBuilder.__init__..Comment._getDataN)r r r r rp rq r" rk r rl r r rn s rn )r3 ZgetETreeModuler
r rH rD namespaceHTMLElementsrW r` rn elementClass commentClassr rS r )r rr rT r` rn r rb r r s zTreeBuilder.__init__c C s$ t j| | j| _g | _d | _d S r
)r rS resetinsertCommentInitial
insertCommentinitial_commentsrN r r r r ru s zTreeBuilder.resetc C s t |S r
)rK r r r r rK s zTreeBuilder.testSerializerc C s t r | jjS | jj S d S r
)rT documentr r r r r r getDocument s zTreeBuilder.getDocumentc C sF g }| j d j}|jr"||j |t| |jrB||j |S )Nr ) openElementsr r2 r* extendlistr' )r Zfragmentr r r r getFragment s zTreeBuilder.getFragmentc C sh |d }|d }|d }|s0t dt d | _n4| j|}||krPt dt | |||}|| _d S )Nr r r z#lxml cannot represent empty doctypez%lxml cannot represent non-xml doctype)warningswarnr rN rD r_ doctypeClass)r tokenr r r Z coercedNamerN r r r
insertDoctype s zTreeBuilder.insertDoctypec C s | j | d S r
)rx r* r rk parentr r r rv , s z TreeBuilder.insertCommentInitialc sB || j kr,| j j d jtkr,tdt tt |
|| d S )Nz@lxml cannot represent adjacent comments beyond the root elements) ry r r r$ r1 r r r superrS rw r __class__r r insertCommentMain1 s
z TreeBuilder.insertCommentMainc C s d}| j r|d| j j 7 }| j jd k s2| j jd k r|d| j| j jpFd 7 }| j jr| j j}|ddkr|ddkrtdt |
dd}|ddkr|d | 7 }q|d
| 7 }n|d 7 }|d 7 }| j j|d
krtdt |d7 }t |}| j
D ] }| |d }||j q | | _| | j_|d
}|d| j}|d kr^|} n d||f } | |_| ||}
||
_| jj|
| j|
| j | _ d S )NrQ z r zGlxml cannot represent doctype with a different name to the root elementz$rk r^ rV ) rN r r r rD Z coercePubidfindr r r replacer
Z
fromstringrx rt Z addpreviousr
documentClassry Z getroottreer getZdefaultNamespacer$ rs r r* r{ r rw ) r r ZdocStrZsysidroot
comment_tokenZcommentr r^ Z etree_tagZ root_elementr r r
insertRoot7 sN
zTreeBuilder.insertRoot)F)N)N)r r r r r r r rs rt Z
fragmentClassr
implementationr ru rK rz r~ r rv r r
__classcell__r r r r rS s
L
rS ) __doc__Z
__future__r r r r resysrQ r r r r
r3 r Z
lxml.etreerT compiler4 rn r$ r1 objectr r rK rR rS r r r r s$
O)