0{Y6@s-dZddlmZmZmZddlZddlZddlZddlm Z ddl m Z ddlm Z dd lm Z dd lmZddlj Z d Zejd Ze jd jZGdddeZGdddeZddZddZGddde jZdS)aModule 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@seZdZddZdS) DocumentTypecCs||_||_||_dS)N)namepublicIdsystemId)selfr rrrG/tmp/pip-ok8sy_ze-build/pip/_vendor/html5lib/treebuilders/etree_lxml.py__init__"s  zDocumentType.__init__N)__name__ __module__ __qualname__rrrrrr !s r c@s@eZdZddZddZddZeeZdS)DocumentcCsd|_g|_dS)N) _elementTree _childNodes)rrrrr)s zDocument.__init__cCs|jjj|jdS)N)rgetrootZaddnext_element)relementrrr appendChild-szDocument.appendChildcCs|jS)N)r)rrrr_getChildNodes0szDocument._getChildNodesN)rrrrrrproperty childNodesrrrrr(s    rcssgd}tjdfdd|d|dk rfjdd|fndjS) Nrc sLt|dst|dr jd|jjr|jjpL|jjsbd|jj}n%d|jj|jj|jjf}jdd|d|fn|j}x"|jdk r|j}qWx|dk r ||d|j }qWqHt |t s+t |t rpt |t sRt jjdksRtjd d||fqHjd x|D]}||dqWn|jtkr jd d||jft|d rH|jrHjd d||jfqHn=t |tjs#ttjj|j}|dk r|jd }|jd}tj|}jdd||j|fn'jdd|j|jft|drg}x|jjD]\} } tj| }|dk rM|j \}} j| } tj|}d|| f} nj| } |j| | fqWx?t!|D].\} } jdd|d| | fqWn|jrjd d|d|jfn|d7}x|D]} | |qWt|d rH|jrHjd d|d|jfndS)Ntagrz #documentz zz|%s%s rz|%s"%s"z#document-fragmentz|%stailrz |%s<%s %s>z|%s<%s>attribz%s %sz |%s%s="%s")"hasattrappenddocinfo internalDTD public_idZ system_url root_namerZ getpreviousZgetnext isinstancestrbytessys version_infomajorAssertionErrorr! comment_typetextr#r Z_Elementetree_builders tag_regexpmatchgroupr prefixes fromXmlNamer$itemsgroupssorted) rindentdtd_strZ next_elementZnsmatchnsr!prefix attributesr valueZ attr_stringchild) infosetFilterrvserializeElementrrrF;st      " '  $    ) %  z(testSerializer..serializeElementz|%s"%s"r"r z )r InfosetFilterr&join)r finalTextr)rDrErFrtestSerializer6s F  rKcs^gd}fdd||dk rQjdd|fndjS) z4Serialize an element and its child nodes to a stringNcsqt|dsi|jjrV|jjr6|jj}nd|jj}j|n|jn|jtkrjd|j fn|j sjd|jfn?dj dd|j j D}jd|j|f|j rj|j nx|D]}|qWjd |jft|d rm|j rmj|j ndS) Nr!z z z<%s>r"cSs&g|]\}}d||fqS)z%s="%s"r).0r rBrrr s z6tostring..serializeElement..z<%s %s>zr#)r%r'r(doctyper*r&rr!r2r3r$rIr:r#)rr>attrrC)rErFrrrFs*     z"tostring..serializeElementz%s"r"rz )r&rI)rrJr)rErFrtostrings  rQcseZdZeZeZdZdZeZ e Z dddZ ddZ ddZd d Zd d Zd dZdddZdfddZddZS) TreeBuilderNFcstjtd|tj|_||_GfdddtGfdddj}Gfdddj }||_ j |_ t j j||dS)NfullTreecs7eZdZifddZfddZdS)z(TreeBuilder.__init__..Attributescs||_tj||xo|jD]a\}}t|tred|dj|df}nj|}||jjj|.Attributes.__init__csltj|||t|trFd|dj|df}nj|}||jjj|.Attributes.__setitem__N)rrrrrXr)rDrr Attributess  rYcseZdZfddZfddZfddZeeeZddZfd d Z eee Z d fd d Z fddZ d S)z%TreeBuilder.__init__..Elementcs;j|}jj||d|||_dS)N namespace) coerceElementElementr _attributes)rr rZ)rYbuilderrDrrrsz.TreeBuilder.__init__..Element.__init__cs4j||_|j|j|j|j_dS)N)r[_nameZ _getETreeTagZ _namespacerr!)rr )rDrr_setNamesz.TreeBuilder.__init__..Element._setNamecsj|jS)N)r9r_)r)rDrr_getNamesz.TreeBuilder.__init__..Element._getNamecSs|jS)N)r])rrrr_getAttributessz4TreeBuilder.__init__..Element._getAttributescs|||_dS)N)r])rrA)rYrr_setAttributessz4TreeBuilder.__init__..Element._setAttributesNcs)j|}jj|||dS)N)ZcoerceCharactersr\ insertText)rdata insertBefore)r^rDrrrdsz0TreeBuilder.__init__..Element.insertTextcsjj||dS)N)r\r)rrC)r^rrrsz1TreeBuilder.__init__..Element.appendChild) rrrrr`rarr rbrcrArdrr)rYr^rDrrr\s  r\csReZdZfddZfddZddZeeeZdS)z%TreeBuilder.__init__..Commentcs&j|}jj||dS)N) coerceCommentCommentr)rre)r^rDrrrsz.TreeBuilder.__init__..Comment.__init__csj|}||j_dS)N)rgrr3)rre)rDrr_setDatasz.TreeBuilder.__init__..Comment._setDatacSs |jjS)N)rr3)rrrr_getDatasz.TreeBuilder.__init__..Comment._getDataN)rrrrrirjrrer)r^rDrrrhs  rh)r4getETreeModuler r rHrDnamespaceHTMLElementsrTr\rh elementClass commentClassrrRr)rrlrSr\rhr)rYr^rDrrs %"  zTreeBuilder.__init__cCs2tjj||j|_g|_d|_dS)N)rrRresetinsertCommentInitial insertCommentinitial_commentsrN)rrrrros  zTreeBuilder.resetcCs t|S)N)rK)rrrrrrKszTreeBuilder.testSerializercCs$tr|jjS|jjjSdS)N)rSdocumentrr)rrrr getDocuments zTreeBuilder.getDocumentcCseg}|jdj}|jr2|j|jn|jt||jra|j|jn|S)Nr) openElementsrr3r&extendlistr#)rfragmentrrrr getFragments  zTreeBuilder.getFragmentcCs|d}|d}|d}|s@tjdtd|_nO|jj|}||krqtjdtn|j|||}||_dS)Nr rrz#lxml cannot represent empty doctypez%lxml cannot represent non-xml doctype)warningswarnrrNrDr[ doctypeClass)rtokenr rrZ coercedNamerNrrr insertDoctype!s     zTreeBuilder.insertDoctypecCs|jj|dS)N)rrr&)rreparentrrrrp1sz TreeBuilder.insertCommentInitialcs^||jkrA|jjjdjtkrAtjdtntt |j ||dS)Nrz@lxml cannot represent adjacent comments beyond the root elements) rsrrr!r2rzr{rsuperrRrq)rrer) __class__rrinsertCommentMain4szTreeBuilder.insertCommentMainc Cscd}|jr`|jjs!t|d|jj7}|jjdk sY|jjdk r*|d|jj|jjpwd7}|jjr|jj}|jddkr|jddkrtj dt |j dd }n|jddkr |d |7}q'|d |7}q*|d 7}n|d 7}|jj|dkr`tj dt q`n|d7}t j |}x+|jD] }|jt j|dqW|j|_|j|j_|d}|jd|j}|dkr|}nd||f}||_|j||} || _|jjj| |jj| |j|_dS)zCreate the document rootrPz r zGlxml cannot represent doctype with a different name to the root elementz$rerZz{%s}%s)rNr r1rrrDZ coercePubidfindrzr{rreplacer fromstringrrZ addpreviousrh documentClassrsZ getroottreergetdefaultNamespacer!rmrrr&rurrq) rr}ZdocStrsysidroot comment_tokenr rZZ etree_tagZ root_elementrrr insertRoot:sJ    *        zTreeBuilder.insertRoot)rrrrrr r|rmrn fragmentClassr implementationrrorKrtryr~rprrrr)rrrRs J    rR)__doc__ __future__rrrrzrer.rPrr rr r4r Z lxml.etreerScompiler5rhr!r2objectr rrKrQrRrrrr s$    S -