a
lj9 @ s d Z ddlmZmZmZ ddlZddlZddlZzddlm Z W n e
y^ ddl m Z Y n0 ddl m
Z
ddlmZ dd l mZ dd
l mZ dd l mZ ddlmZ dd lmZmZ d
ZedZedjZ G dd de Z G dd de Z dd Z dd Z!G dd de
j"Z"dS )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)MutableMapping )base )DataLossWarning) constants)etree) _ihatexml)PY3 binary_typeTz
{([^}]*)}(.*)Zasdc @ s e Zd Zdd ZdS ) DocumentTypec C s || _ || _|| _d S N)namepublicIdsystemId)selfr r r r P/usr/lib/python3.9/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 0 s zDocument.__init__c C s. | j }| j D ]}q||j d S r )r getrootZ itersiblingsZaddnext_element)r elementlastr r r appendChild4 s
zDocument.appendChildc C s | j S r )r r r r r _getChildNodes; 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 ur| }q|d ur||d | }qnnt
| t st
| t rt
| t st
jd dksJ d
d| | f n$d | D ]}||d qn| jtkrd d| | jf t | d
r(| jr(d
d| | jf nt
| tjsJ tj| j}|d ur|d}|d}tj| }dd| | |f n dd| | jf t | drg }| j D ]d\} }
t| }|d urj| \}} | } tj| }d|| f } n
| } || |
f q t |D ]&\} }
dd|d | |
f q| jrڈd
d|d | jf |d7 }| D ]} | | qt | d
r(| jr(d
d|d | jf d S )Ntagr z #document
zz|%s%s r 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
isinstancestrbytessys version_infor' comment_typetextr* r Z_Elementetree_builders
tag_regexpmatchgroupr
prefixes fromXmlNamer+ itemsgroupssorted)
r! indentdtd_strZ next_elementZnsmatchnsr' prefix
attributesr valueZ attr_stringchild
infosetFilterrvserializeElementr r rK E s|
z(testSerializer..serializeElement
)r )r
InfosetFilterjoinr! r rH r testSerializerA s
F
rP 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 rF r r r
s z6tostring..serializeElement..z<%s %s>zr* )r, r. r/ doctyper0 r- r r' r6 r7 r+ rN r> r* )r! rB attrrG rJ rK r r rK s,
z"tostring..serializeElement )rN rO r rU r tostring s rW 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 sP e Zd Zdd Z fddZdd Zdd Zd d
Zd d Zd
d Z dd Z
dS )z(TreeBuilder.__init__..Attributesc S s
|| _ d S r )r r r! r r r r s z1TreeBuilder.__init__..Attributes.__init__c s4 t |tr&d|d |d f }n
|}|S )N{%s}%sr r )r1 tupleZcoerceAttribute)r keyr rI r r
_coerceKey s
z3TreeBuilder.__init__..Attributes._coerceKeyc S s0 | j j j| | }ts,t|tr,|d}|S )Nascii)r r+ r_ r
r1 r decoder r] rF r r r __getitem__ s
z4TreeBuilder.__init__..Attributes.__getitem__c S s || j j j| |< d S r r r+ r_ rb r r r __setitem__ s z4TreeBuilder.__init__..Attributes.__setitem__c S s | j j j| |= d S r rd )r r] r r r __delitem__ s z4TreeBuilder.__init__..Attributes.__delitem__c S s t | jjjS r )iterr r+ r r r r __iter__ s z1TreeBuilder.__init__..Attributes.__iter__c S s t | jjjS r )lenr r+ r r r r __len__ s z0TreeBuilder.__init__..Attributes.__len__c S s | j j j S r )r r+ clearr r r r rk s z.TreeBuilder.__init__..Attributes.clearN) r r r r r_ rc re rf rh rj rk r r^ r r
Attributes s rl c sp e Zd Z fddZfddZfddZeeeZdd Zd d
Z eee Z
dfd d
Z dd Z d S )z%TreeBuilder.__init__..Elementc s* |}jj| ||d | | _d S )N) namespace)
coerceElementElementr _attributes)r r rm rl builderrI r r r s
z.TreeBuilder.__init__..Element.__init__c s$ || _| | j| j| j_d S r )rn _nameZ _getETreeTagZ
_namespacer r' )r r r^ r r _setName s z.TreeBuilder.__init__..Element._setNamec s | jS r )r= rs r r^ r r _getName s z.TreeBuilder.__init__..Element._getNamec S s | j S r )rp r r r r _getAttributes s z4TreeBuilder.__init__..Element._getAttributesc S s | j }| || d S r )rE rk update)r rF rE r r r _setAttributes s z4TreeBuilder.__init__..Element._setAttributesNc s |} j| || d S r )ZcoerceCharactersro
insertText)r dataZ insertBeforerr rI r r ry s
z0TreeBuilder.__init__..Element.insertTextc S s0 t | | j| j}| jjr,|jj| jj |S r )typer rm r r+ rw rZ r r r cloneNode s z/TreeBuilder.__init__..Element.cloneNode)N)
r r r r rt ru r% r rv rx rE ry r} r rq r r ro s
ro c s8 e Zd Z fddZfddZdd ZeeeZdS )z%TreeBuilder.__init__..Commentc s |} j| | d S r )
coerceCommentCommentr r rz r{ r r r s
z.TreeBuilder.__init__..Comment.__init__c s |}|| j_d S r )r~ r r7 r r^ r r _setData s
z.TreeBuilder.__init__..Comment._setDatac S s | j jS r )r r7 r r r r _getData s z.TreeBuilder.__init__..Comment._getDataN)r r r r r r r% rz r r{ r r r
s r )r8 ZgetETreeModuler r rM rI namespaceHTMLElementsr ro r elementClass commentClassr rX r )r r rY ro r r rq r r s $zTreeBuilder.__init__c C s$ t j| | j| _g | _d | _d S r )r rX resetinsertCommentInitial
insertCommentinitial_commentsrS r r r r r s zTreeBuilder.resetc C s t |S r )rP rZ r r r rP # s zTreeBuilder.testSerializerc C s t r | jjS | jj S d S r )rY documentr r r r r r getDocument& s zTreeBuilder.getDocumentc C sF g }| j d j}|jr"||j |t| |jrB||j |S )Nr ) openElementsr r7 r- extendlistr* )r fragmentr! 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 rS rI rn doctypeClass)r tokenr r r Z coercedNamerS r r r
insertDoctype6 s zTreeBuilder.insertDoctypec C s6 |d u s|| j u sJ | j jd u s&J | j| d S r )r r r r- r rz parentr r r r F 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) r r r r' r6 r r r superrX r r __class__r r insertCommentMainK s
z TreeBuilder.insertCommentMainc C s d}| j r| j jsJ |d| j j 7 }| j jd us>| j jd ur|d| j| j jpRd 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 u rj|} n d||f } | |_| ||}
||
_| jj|
| j|
| j | _ d S )NrV z r zGlxml cannot represent doctype with a different name to the root elementz$rz rm r[ ) rS r r r rI Z coercePubidfindr r r replacer Z
fromstringr r Z addpreviousr
documentClassr Z getroottreer getZdefaultNamespacer' r r r- r r r ) r r ZdocStrZsysidroot
comment_tokenZcommentr rm Z etree_tagZ root_elementr r r
insertRootQ sP
zTreeBuilder.insertRoot)F)N)N)r r r r r r r r r Z
fragmentClassr implementationr r rP r r r r r r
__classcell__r r r r rX s
\
rX )#__doc__Z
__future__r r r r rer4 collections.abcr ImportError collectionsrV r r
r r r8 r Z
lxml.etreeZpip._vendor.sixr
r rY compiler9 r r' r6 objectr r rP rW rX r r r r s.
O)