B îq\è(ã@shdZddlZddlZyddlmZWn$ek rHdd„Zdd„ZYnXejZej ZGd d „d ƒZ dS) zZ Contains a class that makes it simple to stream out well-formed and nicely-indented XML. éNé)Ú _iterparsercCs(| dd¡}| dd¡}| dd¡}|S)z< Escapes &, < and > in an XML CDATA string. ú&z&úz>)Úreplace)Ús©r ú7lib/python3.7/site-packages/astropy/utils/xml/writer.pyÚxml_escape_cdatas   r cCs@| dd¡}| dd¡}| dd¡}| dd¡}| d d ¡}|S) zE Escapes &, ', ", < and > in an XML attribute value. rz&ú'z'ú"z"rz<rz>)r)rr r r Ú xml_escapes      rc@s¨eZdZdZdd„Zd#dd„Zifdd „Zejd$d d „ƒZ ejifd d„ƒZ dd„Z dd„Z d%dd„Z dd„Zddifdd„Zdd„Zdd„Zd&dd „Zed!d"„ƒZdS)'Ú XMLWriteraN A class to write well-formed and nicely indented XML. Use like this:: w = XMLWriter(fh) with w.tag('html'): with w.tag('body'): w.data('This is the content') Which produces:: This is the content cCsB|j|_t|dƒr|j|_d|_g|_g|_d|_t|_t|_dS)zY Parameters ---------- file : writable file-like object. Úflushrz@ N) ÚwriteÚhasattrrÚ_openÚ_tagsÚ_dataÚ _indentationr r)ÚselfÚfiler r r Ú__init__:s zXMLWriter.__init__TFcCs¤|jr&|r| d¡n | d¡d|_|jr d |j¡}|rŠ| d¡}tj|||d}| d¡| | |¡¡| d¡| | ¡¡n| | |¡¡g|_dS) z) Flush internal buffers. z> rrÚr)Zinitial_indentZsubsequent_indentÚ N)rrrÚjoinÚget_indentation_spacesÚtextwrapZfillr )rÚindentÚwrapÚdatar r r Ú_flushKs&      zXMLWriter._flushcKs®| ¡g|_|j |¡| | d¡¡| d |¡¡|sB|rž| ¡}| |¡t |  ¡ƒ}|  ¡x4|D],\}}|dk rn|  |¡}| d ||¡¡qnWd|_ t|jƒS)am Opens a new element. Attributes can be given as keyword arguments, or as a string/string dictionary. The method returns an opaque identifier that can be passed to the :meth:`close` method, to close all open elements up to and including this one. Parameters ---------- tag : str The element name attrib : dict of str -> str Attribute dictionary. Alternatively, attributes can be given as keyword arguments. Returns ------- id : int Returns an element identifier. éÿÿÿÿz<{}Nz {}="{}"r)r"rrÚappendrrÚformatÚcopyÚupdateÚlistÚitemsÚsortrrÚlen)rÚtagÚattribÚextraÚkÚvr r r Ústartes     zXMLWriter.startÚ escape_xmlc+sŠ|j}|dkrVy ddl‰Wntk r6tdƒ‚YnXˆdkrDi‰‡‡fdd„|_n$|dkrjdd„|_n|d krztd ƒ‚dV||_dS) aYContext manager to control how XML data tags are cleaned (escaped) to remove potentially unsafe characters or constructs. The default (``method='escape_xml'``) applies brute-force escaping of certain key XML characters like ``<``, ``>``, and ``&`` to ensure that the output is not valid XML. In order to explicitly allow certain XML tags (e.g. link reference or emphasis tags), use ``method='bleach_clean'``. This sanitizes the data string using the ``clean`` function of the `http://bleach.readthedocs.io/en/latest/clean.html `_ package. Any additional keyword arguments will be passed directly to the ``clean`` function. Finally, use ``method='none'`` to disable any sanitization. This should be used sparingly. Example:: w = writer.XMLWriter(ListWriter(lines)) with w.xml_cleaning_method('bleach_clean'): w.start('td') w.data('google.com') w.end() Parameters ---------- method : str Cleaning method. Allowed values are "escape_xml", "bleach_clean", and "none". **clean_kwargs : keyword args Additional keyword args that are passed to the bleach.clean() function. Z bleach_cleanrNzTbleach package is required when HTML escaping is disabled. Use "pip install bleach".csˆj|fˆŽS)N)Zclean)Úx)ÚbleachÚ clean_kwargsr r ÚÃsz/XMLWriter.xml_cleaning_method..ZnonecSs|S)Nr )r3r r r r6Åsr2zEallowed values of method are "escape_xml", "bleach_clean", and "none")r r4Ú ImportErrorÚ ValueError)rÚmethodr5Zcurrent_xml_escape_cdatar )r4r5r Úxml_cleaning_method‘s%  zXMLWriter.xml_cleaning_methodcks$|j||f|ŽdV| |¡dS)aU A convenience method for creating wrapper elements using the ``with`` statement. Examples -------- >>> with writer.tag('foo'): # doctest: +SKIP ... writer.element('bar') ... # is implicitly closed here ... Parameters are the same as to `start`. N)r1Úend)rr,r-r.r r r r,Ísz XMLWriter.tagcCs0| ¡| | ¡¡| d | |¡¡¡dS)z¢ Adds a comment to the output stream. Parameters ---------- comment : str Comment text, as a Unicode string. z N)r"rrr%r )rÚcommentr r r r<ás zXMLWriter.commentcCs|j |¡dS)z¦ Adds character data to the output stream. Parameters ---------- text : str Character data, as a Unicode string. N)rr$)rÚtextr r r r!îs zXMLWriter.dataNcCsª|r>|jstd |¡ƒ‚||jdkrLtd |jd|¡ƒ‚n|jsLtdƒ‚|j ¡}|jrj| ||¡n|jr„d|_| d¡dS|r–| | ¡¡| d |¡¡dS) a Closes the current element (opened by the most recent call to `start`). Parameters ---------- tag : str Element name. If given, the tag must match the start tag. If omitted, the current element is closed. zunbalanced end({})r#zexpected end({}), got {}zunbalanced end()rz/> Nz ) rr8r%Úpoprr"rrr)rr,rr r r r r;ùs$   z XMLWriter.endcCs xt|jƒ|kr| ¡qWdS)zð Closes open elements, up to (and including) the element identified by the given identifier. Parameters ---------- id : int Element identifier, as returned by the `start` method. N)r+rr;)rÚidr r r Úcloses zXMLWriter.closecKs0|j||f|Ž|r| |¡|jd|ddS)z¤ Adds an entire element. This is the same as calling `start`, `data`, and `end` in sequence. The ``text`` argument can be omitted. F)rr N)r1r!r;)rr,r=r r-r.r r r Úelement%s zXMLWriter.elementcCsdS)Nr )rr r r r0szXMLWriter.flushcCs t|jƒS)z\ Returns the number of indentation levels the file is currently in. )r+r)rr r r Úget_indentation3szXMLWriter.get_indentationrcCs|jdt|jƒ|…S)z` Returns a string of spaces that matches the current indentation level. N)rr+r)rÚoffsetr r r r:sz XMLWriter.get_indentation_spacescCs>i}x4|D],}t||ƒdk r tt||ƒƒ|| dd¡<q W|S)a Converts an object with a bunch of attributes on an object into a dictionary for use by the `XMLWriter`. Parameters ---------- obj : object Any Python object attrs : sequence of str Attribute names to pull from the object Returns ------- attrs : dict Maps attribute names to the values retrieved from ``obj.attr``. If any of the attributes is `None`, it will not appear in the output dictionary. NÚ_ú-)ÚgetattrÚstrr)ÚobjZattrsÚdÚattrr r r Ú object_attrsAs  zXMLWriter.object_attrs)TF)r2)NTF)r)Ú__name__Ú __module__Ú __qualname__Ú__doc__rr"r1Ú contextlibÚcontextmanagerr:r,r<r!r;r@rArrBrÚ staticmethodrKr r r r r&s"  , ;     r) rOrPrrrr7r rZescape_xml_cdatar2rr r r r Ús